Update app.py

app.py

@@ -344,7 +344,7 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
     with gr.Row(elem_id="header") as header_row:
         with gr.Column(scale=8):
             gr.Markdown("## Mental Health Chatbot")
-        with gr.Column(scale=4) as profile_container:
+        with gr.Column(visible=False, elem_id="profile_container") as profile_container:
             profile_button = gr.Button("👤", elem_id="profile_button", variant="secondary")
             with gr.Column(visible=False, elem_id="profile_info_box") as profile_info_box:
                 profile_info = gr.HTML()
@@ -659,10 +659,10 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
 
     # Handle login: update previous patients dropdown
     def handle_login(username, password):
+        prev_choices = []
         if login_user(username, password):
             user_session.value = username
             prev_choices = get_previous_patients()
-            # Return 27 outputs now: existing 20 plus 7 more for sessions and booking tabs.
             return (
                 f"Welcome, {username}!",           # login_output
                 gr.update(visible=True),           # main_panel
@@ -690,10 +690,10 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
                 "",                                # patient_name_appt (Booking tab field)
                 "",                                # appointment_date (Booking tab field)
                 "",                                # appointment_time (Booking tab field)
-                ""                                 # reason (Booking tab field)
+                "",                                # reason (Booking tab field)
+                gr.update(visible=True)            # profile_container: show profile icon now
             )
         else:
-            # On failure, clear those fields as well.
             return (
                 "Invalid credentials.",            # login_output
                 gr.update(),                       # main_panel
@@ -721,7 +721,8 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
                 "",                                # patient_name_appt
                 "",                                # appointment_date
                 "",                                # appointment_time
-                ""                                 # reason
+                "",                                # reason
+                gr.update(visible=False)           # profile_container: hide profile icon on failure
             )
 
 
@@ -739,10 +740,12 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
         therapy_textbox, therapy_conf_html,
         summary_textbox, explanation_textbox,
         generate_report_btn, report_download, session_data_state,
-        search_name, search_date, booking_output, patient_name_appt, appointment_date, appointment_time, reason
+        search_name, search_date, booking_output, patient_name_appt, appointment_date, appointment_time, reason,
+        profile_container  # new output for profile container
     ]
     )
 
+
     def handle_logout():
         user_session.value = ""
         return (
@@ -759,9 +762,11 @@ with gr.Blocks(css=open("styles.css", "r").read(), theme="soft") as app:
             "",                        # Clear new_username textbox (register page)
             "",                        # Clear new_password textbox (register page)
             "",                        # Clear full_name textbox (register page)
-            ""                         # Clear email textbox (register page)
+            "",                        # Clear email textbox (register page)
+            gr.update(visible=False)   # profile_container: hide profile icon
         )
 
+
     logout_button.click(
     handle_logout,
     outputs=[