Update app.py

app.py

@@ -46,7 +46,30 @@ def read_audio(file_path):
         return f"Audio length: {len(audio)} ms"
     except:
         return "Audio read error"
+        
+def execute_tool(self, tool, input_data, file_url):
 
+    if tool == "read_excel":
+
+        return self.read_excel(self.download_file(file_url))
+
+    if tool == "read_image":
+
+        return self.read_image(self.download_file(file_url))
+
+    if tool == "scrape_page":
+
+        return self.scrape_page(input_data)
+
+    if tool == "calculator":
+
+        return str(eval(input_data, {"__builtins__": {}}))
+
+    if tool == "wiki_search":
+
+        return self.wiki_search(input_data)
+
+    return "Unknown tool"
 
 def read_youtube(url):
     try:
@@ -61,7 +84,68 @@ def read_youtube(url):
 from openai import OpenAI
 
 class BasicAgent:
-
+    def agent_loop(self, question, file_url):
+    
+        memory = ""
+    
+        for step in range(5):
+    
+            prompt = f"""
+    You are a GAIA solving agent.
+    
+    Available tools:
+    
+    1. read_excel
+    2. read_image
+    3. scrape_page
+    4. calculator
+    5. wiki_search
+    
+    Question:
+    {question}
+    
+    Previous steps:
+    {memory}
+    
+    Decide next action.
+    
+    Format:
+    
+    TOOL: tool_name
+    INPUT: tool_input
+    
+    OR
+    
+    FINAL: answer
+    """
+    
+            response = self.client.chat.completions.create(
+                model="gpt-4o-mini",
+                temperature=0,
+                messages=[{"role": "user", "content": prompt}]
+            )
+    
+            action = response.choices[0].message.content
+    
+            print("Agent step:", action)
+    
+            # FINAL ANSWER
+            if "FINAL:" in action:
+    
+                return action.split("FINAL:")[-1].strip()
+    
+            # TOOL CALL
+            if "TOOL:" in action:
+    
+                tool = action.split("TOOL:")[1].split("\n")[0].strip()
+    
+                input_data = action.split("INPUT:")[-1].strip()
+    
+                result = self.execute_tool(tool, input_data, file_url)
+    
+                memory += f"\nTool {tool} result:\n{result}\n"
+    
+        return "No answer found"
     def __init__(self):
         print("🚀 Super GAIA Agent initialized")
         self.client = OpenAI()
@@ -325,27 +409,13 @@ Return only the answer.
     # ------------------------------------------------
 
     def __call__(self, question, file_url=None):
-
+    
         print("Question:", question)
-
-        math = self.calculator(question)
-
-        if math:
-            print("Math used:", math)
-            return math
-
-        file_content = self.load_file(file_url)
-
-        reasoning = self.reason(question, file_content)
-
-        print("Reasoning:", reasoning)
-
-        answer = self.extract(reasoning)
-
-        answer = self.clean(answer)
-
-        print("Final Answer:", answer)
-
+    
+        answer = self.agent_loop(question, file_url)
+    
+        print("Final:", answer)
+    
         return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):