Update app.py

app.py

@@ -54,15 +54,7 @@ class BasicAgent:
                 fn=extract_ingredients,
                 description="Extracts and returns a comma-separated, alphabetized list of ingredients for a pie filling from a transcription string."
             )
-
-            chess_tool = FunctionTool.from_defaults(
-                name="chess_move",
-                fn=chess_move,
-                description="Analyzes a chess image and returns the correct move to win"
-            )
-
             
-    
             # Registra il tool
             Settings.tools = [ingredient_tool, chess_tool]
 
@@ -73,7 +65,7 @@ class BasicAgent:
             )
     
             # Prepara l'agente
-            self.agent = OpenAIAgent.from_tools([ingredient_tool, chess_tool], llm=llm, verbose=True)
+            self.agent = OpenAIAgent.from_tools([ingredient_tool], llm=llm, verbose=True)
     
             # Client OpenAI per chiamate esterne (immagini/audio)
            
@@ -144,6 +136,16 @@ class BasicAgent:
             else:
                 risposta = "Error loading audio file"
 
+        elif file_info.endswith(".py"):
+            print_coso("Python code file detected")
+            code_content = get_or_download_code(file_info)
+            prompt = (
+                "The following Python code is attached. Please analyze it and provide only the final output:\n\n"
+                f"{code_content}\n\n"
+                f"Question: {question}"
+            )
+            risposta = self._ask_gpt4o(prompt)
+
         elif file_info.endswith(".txt"):
             print("coso Text file detected")
             text_content = self._load_text(file_info)
@@ -314,7 +316,25 @@ def get_or_download_audio(file_name: str) -> bytes:
         print(f"[ERRORE] Impossibile leggere il file audio {file_path}: {e}")
         return None
 
+def get_or_download_code(file_name: str) -> str:
+    import os
+    import requests
+
+    file_path = os.path.join("data", file_name)
+    hf_token = os.getenv("HF_TOKEN_READ")
 
+    if not os.path.exists(file_path):
+        print(f"[INFO] File {file_name} non trovato. Scarico...")
+        url = f"https://huggingface.co/datasets/gaia-benchmark/GAIA/resolve/main/2023/validation/{file_name}"
+        headers = {"Authorization": f"Bearer {hf_token}"}
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        with open(file_path, "wb") as f:
+            f.write(response.content)
+        print(f"[INFO] Scaricato in {file_path}")
+
+    with open(file_path, "r") as f:
+        return f.read()
     
     '''
     base_url = "https://huggingface.co/datasets/gaia-benchmark/GAIA/resolve"
@@ -358,13 +378,12 @@ def create_mock_questions():
     
     
     return [  
-        {    
-            "task_id":"cca530fc-4052-43b2-b130-b30968d8aa44",
-            "question":"Review the chess position provided in the image. It is black's turn. Provide the correct next move for black which guarantees a win. Please provide your response in algebraic notation.",
-            "Level":"1",
-            "file_name":"cca530fc-4052-43b2-b130-b30968d8aa44.png" 
-        }
-       
+       {
+          "task_id":"f918266a-b3e0-4914-865d-4faa564f1aef",
+          "question":"What is the final numeric output from the attached Python code?",
+          "Level":"1",
+          "file_name":"f918266a-b3e0-4914-865d-4faa564f1aef.py"
+       }
     ]
 
 
@@ -398,13 +417,7 @@ def extract_ingredients(transcription: str) -> str:
     # normalizza, rimuove duplicati e ordina
     unique_ingredients = sorted(set(match.strip() for match in matches))
     return ", ".join(unique_ingredients)
-    
-def chess_move() -> str:
-    """
-    Analyses a chess image and returns the correct move to win
-    """
-    print_coso("tool chess_move")
-    return 'Qxa5'
+
 
 def print_coso(scritta: str):
     print(f"coso {scritta}")
@@ -490,7 +503,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         try:
             file_name = item.get("file_name")
             print_coso(f"file_name riga in 3. Run your Agent: {file_name}")
-            submitted_answer = 'Qg2+'#agent(question_text, file_name)
+            submitted_answer = agent(question_text, file_name)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: