switched duplicates

app.py

@@ -176,8 +176,6 @@ def simulate_interaction(user_input, iterations=1):
         q_in = f"Vignette: {summary}\n" f"Current Estimated Diagnosis:\n" f"{diagnosis}"
         q_out = questioner.act(q_in)
         history.append(f"Doctor: {q_out['output']}")
-        # Append actual user input to keep the conversation realistic
-        history.append(f"Patient: {user_input}")
 
         # Treatment
         treatment_out = treatment_agent.act(f"Diagnosis: {diagnosis}\n")
@@ -213,6 +211,7 @@ ANSWER: {res['question']['output']}
 {res['conversation']}
 """
 
+
 # === Stateful Gradio UI ===
 def stateful_ui_fn(user_input, history):
     # Initialize or retrieve history list
@@ -222,7 +221,7 @@ def stateful_ui_fn(user_input, history):
     # Run one round of simulation
     res = simulate_interaction(user_input)
     # Extract last doctor line from the fresh conversation
-    last_line = res['conversation'].splitlines()[-1]
+    last_line = res["q_out"].splitlines()[-1]
     # Append the doctor's new line
     history.append(last_line)
     # Build the displayed conversation
@@ -230,6 +229,7 @@ def stateful_ui_fn(user_input, history):
     # Return both the display text and updated history
     return convo, history
 
+
 demo = gr.Interface(
     fn=stateful_ui_fn,
     inputs=[
@@ -238,7 +238,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Doctor Simulation Output"),
-        gr.State(),    # returns the updated history
+        gr.State(),  # returns the updated history
     ],
     title="🧠 AI Doctor Multi-Agent Reasoning",
 )