Update app.py

app.py

@@ -25,10 +25,10 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
     with gr.Row(visible=False) as logged_in_header:
         welcome_message = gr.Markdown()
 
-    # --- Main Tabs  ---
+    # --- Main Tabs ---
     with gr.Tabs() as main_tabs:
 
-        # --- Tab 1: Homepage ---
+        # --- Tab 1: Homepage (Full Content Restored) ---
         with gr.TabItem("Home 🏠", id="home_tab"):
             gr.Markdown(
                 "<h2 class='centered-text'>A Smart First Step in Understanding Your Health</h2>",
@@ -70,8 +70,8 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
                     """
                 )
 
-        # --- Tab 2: Authentication (Moved Up) ---
-        with gr.TabItem("Login / Signup 🔑", id="auth_tab"):
+        # --- Tab 2: Authentication ---
+        with gr.TabItem("Login / Signup 🔑", id="auth_tab") as auth_tab_item:
             auth_message = gr.Markdown()
             
             with gr.Column(visible=True) as login_col:
@@ -144,7 +144,6 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
                 )
 
     # --- Component Logic (Event Handlers) ---
-    # NOTE: No changes are needed in the event handlers for this reordering.
     
     def switch_to_signup(): return gr.update(visible=False), gr.update(visible=True)
     def switch_to_login(): return gr.update(visible=True), gr.update(visible=False)
@@ -156,8 +155,25 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
         is_female = user_data.get("gender") == "Female"
         username = user_data.get("username", "User")
         welcome_text = f"### 👋 Welcome, {username}!"
-        return (gr.update(visible=is_logged_in), gr.update(visible=not is_logged_in), gr.update(visible=is_female), gr.update(value=welcome_text), gr.update(visible=is_logged_in))
-    user_state.change(fn=handle_user_state_change, inputs=user_state, outputs=[prediction_view, logged_out_view, pregnancies_row, welcome_message, logged_in_header])
+        
+        return (
+            gr.update(visible=is_logged_in),         # prediction_view
+            gr.update(visible=not is_logged_in),     # logged_out_view
+            gr.update(visible=is_female),            # pregnancies_row
+            gr.update(value=welcome_text),           # welcome_message
+            gr.update(visible=is_logged_in),         # logged_in_header
+            gr.update(visible=not is_logged_in)      # auth_tab_item (Show if NOT logged in)
+        )
+        
+    user_state.change(
+        fn=handle_user_state_change, 
+        inputs=user_state, 
+        outputs=[
+            prediction_view, logged_out_view, pregnancies_row, 
+            welcome_message, logged_in_header,
+            auth_tab_item
+        ]
+    )
 
     signup_btn.click(fn=register, inputs=[email_signup, pwd_signup, gender_signup, username_signup], outputs=[auth_message, email_signup, pwd_signup, gender_signup, username_signup])
     login_btn.click(fn=login, inputs=[email_login, pwd_login, user_state], outputs=[auth_message, user_state, main_tabs])