Update dashboard.py

dashboard.py

@@ -102,10 +102,9 @@ ABOUT_TAB_NAME = "ℹ️ About"
 COMMUNITY_TAB_NAME = "🌐 Community"
 GUIDE_TAB_NAME = "📘 User Guide"
 
-# ... [imports and other code remain unchanged above]
-
 with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
+    selected_tab = gr.State(HOME_TAB_NAME)
 
     with gr.Tabs() as tabs:
         with gr.Tab(HOME_TAB_NAME, visible=True) as home_tab:
@@ -167,29 +166,29 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
     def update_ui_on_auth_change(logged_in_status):
         if logged_in_status:
             return (
-                gr.update(visible=False),  # login_tab
-                gr.update(visible=True),   # detect_tab
-                gr.update(visible=False),  # home_tab
+                gr.update(visible=False),
+                gr.update(visible=True),
+                gr.update(visible=False),
                 gr.update(value="✅ Login successful!", visible=True),
-                gr.update(value=DETECT_TAB_NAME)  # auto-switch to Detect tab
+                DETECT_TAB_NAME
             )
         else:
             return (
-                gr.update(visible=True),   # login_tab
-                gr.update(visible=False),  # detect_tab
-                gr.update(visible=True),   # home_tab
+                gr.update(visible=True),
+                gr.update(visible=False),
+                gr.update(visible=True),
                 gr.update(value="", visible=False),
-                gr.update(value=LOGIN_TAB_NAME)  # auto-switch to Login tab on logout
+                LOGIN_TAB_NAME
             )
 
     def handle_login(email, password):
         if login_user(email, password):
-            return True, gr.update(value="✅ Login successful!", visible=True)
+            return True, gr.update(value="✅ Login successful!", visible=True), DETECT_TAB_NAME
         else:
-            return False, gr.update(value="❌ Invalid email or password.", visible=True)
+            return False, gr.update(value="❌ Invalid email or password.", visible=True), LOGIN_TAB_NAME
 
     def handle_logout():
-        return False, "", "", gr.update(selected=LOGIN_TAB_NAME)
+        return False, "", "", LOGIN_TAB_NAME
 
     def handle_signup(name, phone, email, gender, password):
         msg = register_user(name, phone, email, gender, password)
@@ -198,19 +197,16 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
         else:
             return gr.update(value=msg, visible=True), name, phone, email, gender, password, gr.update(open=True)
 
-    login_btn.click(fn=handle_login, inputs=[email_login, password_login], outputs=[is_logged_in, message_output])
-    logout_btn.click(fn=handle_logout, inputs=[], outputs=[is_logged_in, email_login, password_login, tabs])
-
-    is_logged_in.change(
-        fn=update_ui_on_auth_change,
-        inputs=is_logged_in,
-        outputs=[login_tab, detect_tab, home_tab, message_output, tabs]
-    )
+    login_btn.click(fn=handle_login, inputs=[email_login, password_login], outputs=[is_logged_in, message_output, selected_tab])
+    logout_btn.click(fn=handle_logout, inputs=[], outputs=[is_logged_in, email_login, password_login, selected_tab])
+    is_logged_in.change(fn=update_ui_on_auth_change, inputs=is_logged_in,
+                        outputs=[login_tab, detect_tab, home_tab, message_output, selected_tab])
     signup_btn.click(fn=handle_signup, inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],
                      outputs=[message_output, name_signup, phone_signup, email_signup, gender_signup, password_signup, signup_accordion])
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
 
-    demo.load(lambda: False, None, [is_logged_in])
+    demo.load(lambda: (False, HOME_TAB_NAME), None, [is_logged_in, selected_tab])
+    selected_tab.change(fn=None, inputs=selected_tab, outputs=None, _js="(tab) => window.gradioApp().getElementById('tabs').querySelector(`[aria-label='${tab}']`).click()")
 
 if __name__ == "__main__":
     demo.launch()