Update app.py

app.py

@@ -246,6 +246,8 @@ questions = [
     ("Have these difficulties been present since childhood and continue to affect your work or relationships?", "क्या ये कठिनाइयाँ बचपन से रही हैं और अब भी आपके काम या रिश्तों को प्रभावित कर रही हैं?"),
 ]
 
+
+
 # ---- STATE ----
 state = {
     "index": 0,
@@ -253,6 +255,7 @@ state = {
 }
 
 # ---- FUNCTIONS ----
+
 def render_question():
     idx = state["index"]
     total = len(questions)
@@ -267,15 +270,6 @@ def render_question():
         gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
     )
 
-def format_responses(questions, responses):
-    report = []
-    for idx, ((en, hi), answer) in enumerate(zip(questions, responses), 1):
-        if answer.lower() == "yes":
-            report.append(f"{idx}. {en}\nAnswer: Yes\n")
-    if not report:
-        return "No significant symptoms reported."
-    return "\n".join(report)
-
 def next_step(response):
     state["responses"][state["index"]] = response
     state["index"] += 1
@@ -284,7 +278,7 @@ def next_step(response):
         return render_question()
     else:
         return (
-            gr.update(value="✅ All questions completed. Click 'Submit for AI Analysis'."),
+            gr.update(value="✅ All questions completed. Click 'Submit for AI Analysis'."), 
             gr.update(value=""),
             gr.update(visible=False), gr.update(visible=False),
             gr.update(visible=True),
@@ -292,37 +286,43 @@ def next_step(response):
             gr.update(visible=False)
         )
 
+def format_yes_responses():
+    yes_topics = []
+    for (en, _), ans in zip(questions, state["responses"]):
+        if ans.lower() == "yes":
+            yes_topics.append(en)
+    if not yes_topics:
+        return "No significant symptoms reported."
+    return "\n".join(yes_topics)
+
 def run_final_analysis():
     yield (
         gr.update(value="🧠 AI is analyzing your responses... Please wait. ⏳"),
         gr.update(value=""),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False)
+        *[gr.update(visible=False) for _ in range(5)]
     )
 
-    time.sleep(2)
+    time.sleep(1)
+
+    yes_summary = format_yes_responses()
 
-    llm_input = format_responses(questions, state["responses"])
     prompt = (
-        "You are a clinical assistant AI evaluating a user's mental health screening responses.\n"
-        "Analyze only the 'Yes' answers provided below. Group the symptoms into clusters like Anxiety, Depression, Psychosis, Sleep Disorders, or Substance Use.\n"
-        "Summarize the symptoms clearly, give a preliminary impression, and suggest recommended action.\n"
-        "Respond in clear English and Hindi.\n\n"
-        f"{llm_input}"
+        "You are a psychiatric specialist clinical assistant AI evaluating a user's mental health screening.\n"
+        "Summarize the user's symptom clusters (like Anxiety, Depression, Psychosis, etc.) based ONLY on these symptoms:\n"
+        f"{yes_summary}\n\n"
+        "Do NOT list individual questions.\n"
+        "Give a brief, professional clinical impression.\n"
+        "Suggest recommended next steps.\n"
+        "Finally, provide a Hindi translation of your full response.\n"
     )
-    output = llm(prompt, max_new_tokens=500, temperature=0.3)
+
+    output = llm(prompt, max_new_tokens=1500, temperature=0.1, do_sample=False)
     ai_result = output[0]["generated_text"]
 
     yield (
         gr.update(value="✅ AI Analysis Completed."),
         gr.update(value=""),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
+        *[gr.update(visible=False) for _ in range(5)],
         gr.update(value=ai_result, visible=True)
     )
 
@@ -340,7 +340,9 @@ def start_app():
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("## 🧠 MindScreen: Mental Health Self-Screening")
     gr.Markdown("### मानसिक स्वास्थ्य आत्म-स्क्रीनिंग प्रश्नावली")
-    gr.Markdown("**Note: Please choose 'Yes' only if symptoms interfere with your daily life.**")
+    gr.Markdown("**Note: Please choose 'Yes' only if the symptoms cause distress to you or your family, or if they interfere with your day-to-day functioning.**\n\n"
+            "**नोट: कृपया 'हाँ' तभी चुनें जब ये लक्षण आपके या आपके परिवार के लिए परेशानी का कारण बन रहे हों, या आपकी दिनचर्या को प्रभावित करते हों।**")
+
 
     progress_bar = gr.Markdown("")
     question_display = gr.Markdown("", elem_id="question-box")
@@ -357,36 +359,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         result_box = gr.Textbox(
             label="🧠 AI Interpretation (English + Hindi)",
             visible=False,
-            lines=18,
+            lines=20,
             max_lines=25,
             show_copy_button=True,
             interactive=False
         )
 
-    app.load(start_app, outputs=[
-        question_display, progress_bar, yes_btn, no_btn, back_btn,
-        result_btn, result_box
-    ])
-
-    yes_btn.click(lambda: next_step("Yes"), outputs=[
-        question_display, progress_bar, yes_btn, no_btn, back_btn,
-        result_btn, result_box
-    ])
-    no_btn.click(lambda: next_step("No"), outputs=[
-        question_display, progress_bar, yes_btn, no_btn, back_btn,
-        result_btn, result_box
-    ])
-    back_btn.click(go_back, outputs=[
-        question_display, progress_bar, yes_btn, no_btn, back_btn,
-        result_btn, result_box
-    ])
-    result_btn.click(
-        run_final_analysis,
-        outputs=[
-            question_display, progress_bar, yes_btn, no_btn, back_btn,
-            result_btn, result_box
-        ],
-        concurrency_limit=1
-    )
+    app.load(start_app, outputs=[question_display, progress_bar, yes_btn, no_btn, back_btn, result_btn, result_box])
+
+    yes_btn.click(lambda: next_step("Yes"), outputs=[question_display, progress_bar, yes_btn, no_btn, back_btn, result_btn, result_box])
+    no_btn.click(lambda: next_step("No"), outputs=[question_display, progress_bar, yes_btn, no_btn, back_btn, result_btn, result_box])
+    back_btn.click(go_back, outputs=[question_display, progress_bar, yes_btn, no_btn, back_btn, result_btn, result_box])
+    result_btn.click(run_final_analysis, outputs=[question_display, progress_bar, yes_btn, no_btn, back_btn, result_btn, result_box], concurrency_limit=1)
 
 app.launch()