fixed local running bug

app.py

@@ -78,15 +78,20 @@ class RoleAgent:
 
 # === Agents ===
 summarizer = RoleAgent(
-    "You are a clinical summarizer trained to extract structured vignettes from doctor–patient dialogues."
+    "You are a clinical summarizer trained to extract structured vignettes from doctor–patient dialogues.",
+    tokenizer,
+    model,
 )
 diagnoser = RoleAgent(
-    "You are a board-certified diagnostician that diagnoses patients."
+    "You are a board-certified diagnostician that diagnoses patients.", tokenizer, model
+)
+questioner = RoleAgent(
+    "You are a physician asking questions to diagnose a patient.", tokenizer, model
 )
-questioner = RoleAgent("You are a physician asking questions to diagnose a patient.")
-
 treatment_agent = RoleAgent(
-    "You are a board-certified clinician. Based on the diagnosis and patient vignette provided below, suggest a concise treatment plan that could realistically be initiated by a primary care physician or psychiatrist."
+    "You are a board-certified clinician. Based on the diagnosis and patient vignette provided below, suggest a concise treatment plan that could realistically be initiated by a primary care physician or psychiatrist.",
+    tokenizer,
+    model,
 )
 
 
@@ -104,16 +109,16 @@ def simulate_interaction(user_input, iterations=1):
     for i in range(iterations):
         # Summarize
         sum_in = "\n".join(history) + f"\nPrevious Vignette: {summary}"
-        sum_out = summarizer.act(sum_in, tokenizer, model)
+        sum_out = summarizer.act(sum_in)
         summary = sum_out["output"]
 
         # Diagnose
-        diag_out = diagnoser.act(summary, tokenizer, model)
+        diag_out = diagnoser.act(summary)
         diagnosis = diag_out["output"]
 
         # Question
         q_in = f"Vignette: {summary}\nCurrent Estimated Diagnosis: {diag_out['thinking']} {diagnosis}"
-        q_out = questioner.act(q_in, tokenizer, model)
+        q_out = questioner.act(q_in)
         history.append(f"Doctor: {q_out['output']}")
         history.append("Patient: (awaiting response)")