Update app.py

app.py

@@ -226,24 +226,46 @@ class BasicAgent:
     def __call__(self, question: str, task_id: str = "unknown") -> str:
         # Parse question to get both text and file_url
         try:
-            question_data = json.loads(question)
-            state: AgentState = {
-                "question": question_data.get("question", ""),
-                "current_step": "answer",
-                "final_answer": "",
-                "history": [],
-                "needs_search": False,
-                "search_query": "",
-                "task_id": task_id,
-                "logs": {},
-                "file_url": question_data.get("file_url", ""),
-                "code_blocks": question_data.get("code_blocks", [])
-            }
+            # First try to parse as JSON
+            if isinstance(question, str) and question.strip().startswith('{'):
+                question_data = json.loads(question)
+                state: AgentState = {
+                    "question": question_data.get("question", ""),
+                    "current_step": "answer",
+                    "final_answer": "",
+                    "history": [],
+                    "needs_search": False,
+                    "search_query": "",
+                    "task_id": task_id,
+                    "logs": {},
+                    "file_url": question_data.get("file_url", ""),
+                    "code_blocks": question_data.get("code_blocks", [])
+                }
+            else:
+                # If not JSON, treat as plain text question
+                state: AgentState = {
+                    "question": question,
+                    "current_step": "answer",
+                    "final_answer": "",
+                    "history": [],
+                    "needs_search": False,
+                    "search_query": "",
+                    "task_id": task_id,
+                    "logs": {},
+                    "file_url": "",
+                    "code_blocks": []
+                }
+            
             print(f"\nProcessing task {task_id}")
             print(f"Question: {state['question']}")
             print(f"File URL: {state['file_url']}")
+            if state['file_url']:
+                print(f"Raw question data: {question}")
+            
         except (json.JSONDecodeError, KeyError) as e:
             print(f"Error parsing question data: {e}")
+            print(f"Raw question data: {question}")
+            # Fall back to treating as plain text
             state: AgentState = {
                 "question": question,
                 "current_step": "answer",