added multiple iterations feature

app.py

@@ -151,6 +151,7 @@ diagnosis = ""
 def simulate_interaction(user_input, iterations=1):
     history = [f"Doctor: What brings you in today?", f"Patient: {user_input}"]
     summary, diagnosis = "", ""
+    treatment_out = None
 
     for i in range(iterations):
         # Summarize
@@ -166,20 +167,21 @@ def simulate_interaction(user_input, iterations=1):
         q_in = f"Vignette: {summary}\nCurrent Estimated Diagnosis: {diag_out['thinking']} {diagnosis}"
         q_out = questioner.act(q_in)
         history.append(f"Doctor: {q_out['output']}")
-        history.append("Patient: (awaiting response)")
+        # Append actual user input to keep the conversation realistic
+        history.append(f"Patient: {user_input}")
 
         # Treatment
         treatment_out = treatment_agent.act(
             f"Diagnosis: {diagnosis}\nVignette: {summary}"
         )
 
-        return {
-            "summary": sum_out,
-            "diagnosis": diag_out,
-            "question": q_out,
-            "treatment": treatment_out,
-            "conversation": "\n".join(history),
-        }
+    return {
+        "summary": sum_out,
+        "diagnosis": diag_out,
+        "question": q_out,
+        "treatment": treatment_out,
+        "conversation": "\n".join(history),
+    }
 
 
 # === Gradio UI ===