Update app.py

app.py

@@ -73,11 +73,8 @@ class BasicAgent:
             if idx == -1:
                 return full_response.strip()  # fallback: return whole response
             return full_response[idx + len(marker):].strip().rstrip(".")
-            
-        response = self.agent.invoke({"messages": messages})
-        model_answer = response["messages"][-1].content
-        fixed_answer = extract_model_answer(model_answer)
-        reasoning_trace = "\n".join([f"{m.type}: {m.content}" for m in messages + response["messages"]])
+        response = self.agent.run(question) 
+        fixed_answer = extract_model_answer(response)
         return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):