intrgrating suggestions after betta 1

transportmanager/client/pages/two_factor_page.py

@@ -1,6 +1,7 @@
 import flet as ft
 import requests
 from config import API_BASE_URL
+from pages.temporary_password_page import TemporaryPassword
 
 class TwoFactorAuth:
     def __init__(self, page: ft.Page, email: str, login, auth):
@@ -35,10 +36,30 @@ class TwoFactorAuth:
                     print('Admin Logged In')
                     self.page.go("/admin")
                 else:
-                    self.success_text.value = "Verification successful. You are now logged in."
-                    self.error_text.value = ""
-                    self.page.update()
-                    self.page.go("/dashboard")  # Change this to your main page
+                    ui = requests.get(
+                        f"{API_BASE_URL}/auth/me",
+                        headers={"Authorization": f"Bearer {token}"},
+                        timeout=10,
+                    )
+
+                    if ui.status_code == 200:
+                        uj = ui.json()
+                        try:
+                            logo_filename = uj.get("logo_filename", "")
+                            if logo_filename:
+                                self.page.client_storage.set("custom_logo", logo_filename)
+                        except Exception:
+                            pass
+                    if uj['temporary_password'] == 1:
+                        self.auth.placeholder.content.clean()
+                        temp_passwd_page = TemporaryPassword(self.page)
+                        self.auth.placeholder.content = temp_passwd_page.build()
+                        self.auth.placeholder.update()
+                    else:
+                        self.success_text.value = "Verification successful. You are now logged in."
+                        self.error_text.value = ""
+                        self.page.update()
+                        self.page.go("/dashboard")  
             else:
                 self.error_text.value = "Invalid or expired code."
                 self.success_text.value = ""