Update app.py

app.py

@@ -126,11 +126,11 @@ def format_yes_responses():
 
 def run_final_analysis():
     yield (
-        gr.update(value="⏳ Please wait... analyzing your responses 🧠"),
-        gr.update(value=""),
-        *[gr.update(visible=False) for _ in range(5)],
-        gr.update(visible=False),
-        gr.update(visible=False)
+        gr.update(value="⏳ Please wait... analyzing your responses 🧠"),   # question_display
+        gr.update(value=""),                                               # progress_bar
+        *[gr.update(visible=False) for _ in range(4)],                     # yes_btn, no_btn, back_btn, result_btn
+        gr.update(visible=False),                                          # result_box
+        gr.update(visible=False)                                           # restart_btn
     )
 
     time.sleep(1)
@@ -138,29 +138,28 @@ def run_final_analysis():
     yes_summary = format_yes_responses()
 
     prompt = (
-    "You are a psychiatric clinical assistant. Based on the following symptoms reported by the user, "
-    "please provide:\n\n"
-    "1. A brief clinical impression (mention relevant symptom clusters like Anxiety, Depression, Psychosis, etc.).\n"
-    "2. Suggested next steps.\n"
-    "3. Hindi translation of your entire response.\n\n"
-    "Symptoms reported:\n"
-    f"{yes_summary}\n\n"
-    "Do not repeat this instruction. Directly provide your analysis."
-)
-
+        "You are a psychiatric clinical assistant. Based on the following symptoms reported by the user, "
+        "please provide:\n\n"
+        "1. A brief clinical impression (mention relevant symptom clusters like Anxiety, Depression, Psychosis, etc.).\n"
+        "2. Suggested next steps.\n"
+        "3. Hindi translation of your entire response.\n\n"
+        "Symptoms reported:\n"
+        f"{yes_summary}\n\n"
+        "Do not repeat this instruction. Directly provide your analysis."
+    )
 
     output = llm(prompt, max_new_tokens=400, temperature=0.3, do_sample=False)
     ai_result = output[0]["generated_text"]
 
     yield (
-        gr.update(value="✅ AI Analysis Completed."),
-        gr.update(value=""),
-        *[gr.update(visible=False) for _ in range(4)],
-        gr.update(visible=True),
-        gr.update(value=ai_result, visible=True),
-        gr.update(visible=True)
+        gr.update(value="✅ AI Analysis Completed."),       # question_display
+        gr.update(value=""),                                # progress_bar
+        *[gr.update(visible=False) for _ in range(4)],       # yes_btn, no_btn, back_btn, result_btn
+        gr.update(value=ai_result, visible=True),           # result_box (important: both value+visible)
+        gr.update(visible=True)                             # restart_btn
     )
 
+
 def go_back():
     if state["index"] > 0:
         state["index"] -= 1