usability fixes

lib/screens/accounts/widgets/account_balance_widget.dart

@@ -1,6 +1,7 @@
 import 'package:flutter/material.dart';
 import 'package:go_router/go_router.dart';
 import 'package:intl/intl.dart';
+import 'package:provider/provider.dart';
 import 'package:qpay/screens/accounts/account_provider.dart';
 import 'package:qpay/screens/accounts/models/account_model.dart';
 import 'package:shared_preferences/shared_preferences.dart';
@@ -20,8 +21,6 @@ class AccountBalanceWidget extends StatefulWidget {
 }
 
 class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
-  late AccountProvider _accountProvider;
-
   bool _isLoggedIn = false;
   String? _workspaceId;
 
@@ -36,20 +35,12 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
   @override
   void initState() {
     super.initState();
-    _accountProvider = AccountProvider();
     _loadAuthStateAndBalances();
   }
 
-  @override
-  void dispose() {
-    _accountProvider.dispose();
-    super.dispose();
-  }
-
   Future<void> _loadAuthStateAndBalances() async {
     final prefs = await SharedPreferences.getInstance();
 
-    // Determine login state from persisted token
     if (prefs.getString("token") != null) {
       setState(() {
         _isLoggedIn = true;
@@ -62,18 +53,37 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
       return;
     }
 
+    if (!_isLoggedIn) {
+      return;
+    }
+
+    if (!mounted) return;
+
+    final accountProvider = context.read<AccountProvider>();
+
     setState(() {
       _isLoadingUsd = true;
       _isLoadingZwG = true;
     });
 
-    // no need to proceed if not logged in
-    if (!_isLoggedIn) {
+    // Try the cached balances first
+    final cachedUsd = accountProvider.balanceForCurrency('USD');
+    final cachedZwg = accountProvider.balanceForCurrency('ZWG');
+
+    if (cachedUsd != null && cachedZwg != null) {
+      if (mounted) {
+        setState(() {
+          _usdAccount = cachedUsd;
+          _zwgAccount = cachedZwg;
+          _isLoadingUsd = false;
+          _isLoadingZwG = false;
+        });
+      }
       return;
     }
 
     // Fetch USD balance
-    final usdResult = await _accountProvider.fetchAccount(_workspaceId, 'USD');
+    final usdResult = await accountProvider.fetchAccount(_workspaceId, 'USD');
     if (mounted) {
       setState(() {
         _isLoadingUsd = false;
@@ -87,7 +97,7 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
     }
 
     // Fetch ZWG balance
-    final zwgResult = await _accountProvider.fetchAccount(_workspaceId, 'ZWG');
+    final zwgResult = await accountProvider.fetchAccount(_workspaceId, 'ZWG');
     if (mounted) {
       setState(() {
         _isLoadingZwG = false;
@@ -117,12 +127,16 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
 
     if (_workspaceId == null || _workspaceId!.isEmpty) return;
 
+    if (!mounted) return;
+
+    final accountProvider = context.read<AccountProvider>();
+
     setState(() {
       _isLoadingUsd = true;
       _isLoadingZwG = true;
     });
 
-    final usdResult = await _accountProvider.fetchAccount(_workspaceId, 'USD');
+    final usdResult = await accountProvider.fetchAccount(_workspaceId, 'USD');
     if (mounted) {
       setState(() {
         _isLoadingUsd = false;
@@ -135,7 +149,7 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
       });
     }
 
-    final zwgResult = await _accountProvider.fetchAccount(_workspaceId, 'ZWG');
+    final zwgResult = await accountProvider.fetchAccount(_workspaceId, 'ZWG');
     if (mounted) {
       setState(() {
         _isLoadingZwG = false;
@@ -509,4 +523,4 @@ class _AccountBalanceWidgetState extends State<AccountBalanceWidget> {
     final formatter = NumberFormat('#,##0.00', 'en_US');
     return '\$${formatter.format(balance)}';
   }
-}
+}