fixed bug

app.py

@@ -221,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["q_out"].splitlines()[-1]
+    last_line = res["question"]["output"]
     # Append the doctor's new line
     history.append(last_line)
     # Build the displayed conversation
@@ -234,7 +234,7 @@ demo = gr.Interface(
     fn=stateful_ui_fn,
     inputs=[
         gr.Textbox(label="Patient Response"),
-        gr.State([]),  # holds the conversation history
+        gr.State(),  # holds the conversation history
     ],
     outputs=[
         gr.Textbox(label="Doctor Simulation Output"),