Upload app.py

app.py

@@ -95,33 +95,18 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         try:
-            # Process the question
             response = self.agent.query(question)
             answer_text = str(response)
-            
-            # Extract the FINAL ANSWER part if it exists
             if "FINAL ANSWER:" in answer_text:
-                reasoning_trace = answer_text.split("FINAL ANSWER:")[0].strip()
                 model_answer = answer_text.split("FINAL ANSWER:")[1].strip()
-                
-                # Include the reasoning trace in the response but formatted for JSON
-                result = {
-                    "model_answer": model_answer,
-                    "reasoning_trace": reasoning_trace
-                }
-                
-                # Return just the answer part for direct evaluation
                 print(f"Agent generated answer: {model_answer[:50]}..." if len(model_answer) > 50 else f"Agent generated answer: {model_answer}")
-                return json.dumps(result)
+                return model_answer
             else:
-                # If no FINAL ANSWER pattern, return the whole response
                 print(f"No 'FINAL ANSWER' found in response. Returning full response.")
-                return json.dumps({"model_answer": answer_text, "reasoning_trace": ""})
-                
+                return answer_text
         except Exception as e:
             print(f"Error generating answer: {e}")
-            error_msg = f"I encountered an error while answering your question: {str(e)}"
-            return json.dumps({"model_answer": error_msg, "reasoning_trace": ""})
+            return f"I encountered an error while answering your question: {str(e)}"
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """