Update app.py

app.py

@@ -48,21 +48,34 @@ class BasicAgent:
         return self.run(question)
         #return self.agent.run(question)
 
-def create_llamaindex_agent() -> AgentWorkflow:
-    api_key = os.getenv("HF_TOKEN")
+
+async def smart_agent(questions_data: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
     llm = HuggingFaceInferenceAPI(
-        model_name="deepseek-ai/DeepSeek-R1-0528",
+        model_name="deepseek-ai/DeepSeek-R1-0528", 
         token=api_key,
-        provider="auto",
-        max_iterations=10
-    )
-    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=llm
+        provider="auto"
         )
+    
+    llamaindex_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=llm,
+    )
 
-    return agent
+    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 llamaindex_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
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -84,7 +97,6 @@ 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)
-    agent = create_llamaindex_agent()
     # try:
     #     agent = BasicAgent()
     # except Exception as e:
@@ -139,24 +151,25 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     #3A
 
-    async def run_all_questions(questions_data):
-        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}")
-        return answers_payload
-
-    answers_payload = asyncio.run(run_all_questions(questions_data))
+    # async def run_all_questions(questions_data):
+    #     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}")
+    #     return answers_payload
+
+    #answers_payload = asyncio.run(run_all_questions(questions_data))
+    answers_payload = asyncio.run(smart_agent())
     
     #3B