Update app.py

app.py

@@ -110,7 +110,7 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
         with gr.TabItem("Find a Specialist 🧑‍⚕️"):
             render_doctor_list()
 
-        # --- UPDATED Prediction Tab ---
+        # --- Prediction Tab ---
         with gr.TabItem("Prediction Tool 🔬", id="prediction_tab"):
             
             with gr.Column(visible=False) as prediction_view:
@@ -128,15 +128,13 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
                         age = gr.Number(label="Age", info="Age in years")
                     
                     with gr.Column(scale=1):
-                        # --- New BMI UI Logic ---
                         bmi = gr.Number(label="BMI", info="Your Body Mass Index. We can calculate this for you.")
                         
                         bmi_choice = gr.Radio(
                             ["I know my BMI", "Calculate from Weight/Height"],
                             label="Provide BMI",
-                            value="I know my BMI" # Default selection
+                            value="I know my BMI"
                         )
-                        # This section is for inputs to calculate BMI
                         with gr.Column(visible=False) as bmi_calc_view:
                             weight = gr.Number(label="Your Weight (kg)")
                             height = gr.Number(label="Your Height (cm)")
@@ -173,25 +171,22 @@ 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), 
-            gr.update(visible=not is_logged_in)
-        )
+        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), gr.update(visible=not 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, 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])
+    # The logout handler must now also clear the weight and height fields.
     logout_btn.click(fn=logout, inputs=[user_state], outputs=[auth_message, user_state, main_tabs, email_login, pwd_login, preg, glucose, bp, insulin, bmi, age, result_output, pregnancies_row, weight, height, bmi_choice])
 
+    # --- CORRECTED BMI LOGIC ---
     def toggle_bmi_inputs(choice):
         if choice == "I know my BMI":
+            # Return 4 values: one for each output component
             return gr.update(interactive=True, value=None), None, None, gr.update(visible=False)
         else: # Calculate from Weight/Height
-            return gr.update(interactive=False, value=None), gr.update(visible=True)
+            # Also return 4 values here to match the outputs list
+            return gr.update(interactive=False, value=None), gr.update(), gr.update(), gr.update(visible=True)
     bmi_choice.change(fn=toggle_bmi_inputs, inputs=bmi_choice, outputs=[bmi, weight, height, bmi_calc_view])
 
     def calculate_bmi(w, h):
@@ -207,9 +202,12 @@ with gr.Blocks(title="DiaSpark: Your Health Management Platform", css=css) as de
     def handle_create_report(user, p, g, b, i, bmi_val, age_val, result):
         if not result:
             return gr.update(visible=False)
-        input_data = {"Pregnancies": p if p is not None else 0, "Glucose": g, "Blood Pressure": b, "Insulin": i, "BMI": bmi_val if bmi_val is not None else 0, "Age": age_val}
+        # Recalculate is no longer needed, just use the value from the bmi box
+        final_bmi_report = bmi_val if bmi_val is not None else 0
+        input_data = {"Pregnancies": p if p is not None else 0, "Glucose": g, "Blood Pressure": b, "Insulin": i, "BMI": final_bmi_report, "Age": age_val}
         file_path = create_report(user, input_data, result)
-        return gr.update(value=file_file, visible=True)
+        # CORRECTED TYPO: It should be 'file_path', not 'file_file'
+        return gr.update(value=file_path, visible=True)
     generate_report_btn.click(fn=handle_create_report, inputs=[user_state, preg, glucose, bp, insulin, bmi, age, result_output], outputs=[report_file_output])
     
     result_output.change(fn=lambda x: gr.update(visible=bool(x)), inputs=result_output, outputs=generate_report_btn)