Update app.py

app.py

@@ -29,10 +29,25 @@ class BasicAgent:
             provider="auto",
             max_iterations=10
         )
-        self.agent = AgentWorkflow.from_tools_or_functions(
-            [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, 
-             round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
-            llm=self.llm
+        # self.agent = AgentWorkflow.from_tools_or_functions(
+        #     [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, 
+        #      round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
+        #     llm=self.llm
+        # )
+
+        self.tools = [
+            wikipedia_search_tool,
+            youtube_transcript_tool,
+            youtube_transcript_snippet_tool,
+            round_to_two_decimals_tool,
+            text_inverter_tool,
+            google_web_search_tool
+        ]
+
+        self.agent = ReActAgent.from_tools(
+            tools=self.tools,
+            llm=self.llm,
+            verbose=True
         )
 
     async def run(self, question: str) -> str:
@@ -66,11 +81,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     submit_url = f"{api_url}/submit"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
-    # try:
-    #     agent = BasicAgent()
-    # except Exception as e:
-    #     print(f"Error instantiating agent: {e}")
-    #     return f"Error initializing agent: {e}", None
+    try:
+        agent = BasicAgent()
+    except Exception as e:
+        print(f"Error instantiating agent: {e}")
+        return f"Error initializing agent: {e}", None
     #In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)
@@ -121,32 +136,20 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     #3A
 
     async def run_all_questions(questions_data):
-        try:
-            agent = BasicAgent()
-        except Exception as e:
-            print(f"Error instantiating agent: {e}")
-            return f"Error initializing agent: {e}", None
-        question_text = questions_data[0].get("question")
-        task_id = questions_data[0].get("task_id")
-        print(f"Processing my question 0 : : : : : : :")
-        answer = await agent.run(question_text)
-        print(f"Answered Task {task_id}: {answer[:50]}...")
         answers_payload = []
-        # answers_payload.append({"task_id": task_id, "submitted_answer": answer})
-        # answers_payload = []
-        # for item in questions_data:
-        #     task_id = item.get("task_id")
-        #     question_text = item.get("question")
-        #     if not task_id or question_text is None:
-        #         print(f"Skipping item with missing task_id or question: {item}")
-        #         continue
-        #     try:
-        #         answer = await agent.run(question_text)  # await coroutine
-        #         answers_payload.append({"task_id": task_id, "submitted_answer": answer})
-        #         print(f"Answered Task {task_id}: {answer[:50]}...")
-        #     except Exception as e:
-        #         answers_payload.append({"task_id": task_id, "submitted_answer": f"AGENT ERROR: {e}"})
-        #         print(f"Error on Task {task_id}: {e}")
+        for item in questions_data:
+            task_id = item.get("task_id")
+            question_text = item.get("question")
+            if not task_id or question_text is None:
+                print(f"Skipping item with missing task_id or question: {item}")
+                continue
+            try:
+                answer = await agent.run(question_text)  # await coroutine
+                answers_payload.append({"task_id": task_id, "submitted_answer": answer})
+                print(f"Answered Task {task_id}: {answer[:50]}...")
+            except Exception as e:
+                answers_payload.append({"task_id": task_id, "submitted_answer": f"AGENT ERROR: {e}"})
+                print(f"Error on Task {task_id}: {e}")
         return answers_payload
 
     answers_payload = asyncio.run(run_all_questions(questions_data))