added app

app.py

@@ -88,7 +88,6 @@ class RoleAgent:
         response = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
 
         thinking = ""
-        print(f"[RESPONSE]: {response}")
         answer = response
         if "Output:" in response:
             # Split on the last occurrence of 'Output:' in case it's repeated
@@ -101,6 +100,15 @@ class RoleAgent:
                 block = response.split("THINKING:", 1)[1].split("END", 1)[0]
                 thinking = block.split("ANSWER:", 1)[0].strip()
                 answer = block.split("ANSWER:", 1)[1].strip()
+        print(
+            "[DEBUG] thinking/answer split:",
+            response,
+            "→",
+            "[THINKING] thinking:",
+            thinking,
+            "[ANSWER] answer:",
+            answer,
+        )
 
         return {"thinking": thinking, "output": answer}
 
@@ -164,7 +172,11 @@ def simulate_interaction(user_input, iterations=1):
         diagnosis = diag_out["output"]
 
         # Question
-        q_in = f"Vignette: {summary}\nCurrent Estimated Diagnosis: {diag_out['thinking']} {diagnosis}"
+        q_in = (
+            f"Vignette: {summary}\n"
+            f"Current Estimated Diagnosis:\n"
+            f"ANSWER: {diagnosis}"
+        )
         q_out = questioner.act(q_in)
         history.append(f"Doctor: {q_out['output']}")
         # Append actual user input to keep the conversation realistic