Preparar para executar!!

.gitignore

@@ -3,3 +3,4 @@ get-pip.py
 *.m4a
 *.mp4
 audio_analysis_output/
+files/

agent.py

@@ -1,4 +1,6 @@
+import re
 from langgraph.prebuilt import create_react_agent
+from agent_util import Agent_Util
 from prompts import *
 from tools import *
 from langgraph_supervisor import create_supervisor
@@ -38,18 +40,33 @@ class Agent:
             output_mode="full_history",
         ).compile()
         
-        print("Agent initialized.")
+        print("Agent initialized.")     
         
-    def _exist_file_with_task_id(task_id: str) -> str:
-        padrao = os.path.join(AGENTS_FILES_PATH, f"{task_id}.*")
-        arquivos_encontrados = glob.glob(padrao)
-        return arquivos_encontrados[0] if arquivos_encontrados else None     
+    
+    
+    def __call__(self, question: str, task_id: str, task_file_name: str) -> str:
+        print(f"Agent received question({task_id}) (first 50 chars): {question[:50]}...")
         
+        file_prefix = ""
+        if task_file_name:
+            print(f"Task com arquivo {task_file_name}")
+            File_Util.baixa_arquivo_task(task_file_name)
+            file_prefix = f"File: {task_file_name}  . "
         
-    def __call__(self, question: str, task_id: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        file = self._exist_file_with_task_id(task_id)
-        
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        for chunk in self.supervisor.stream(
+            {
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": f"{file_prefix}{question}",
+                    }
+                ]
+            },
+        ):
+            Agent_Util.pretty_print_messages(chunk, last_message=True)
+
+        agent_answer = chunk["supervisor"]["messages"]
+                
+        final_answer = re.sub(r"^FINAL ANSWER:\s*", "", agent_answer, flags=re.IGNORECASE)
+        print(f"Agent returning answer for task {task_id}: {final_answer}")
+        return final_answer

app.py

@@ -59,6 +59,9 @@ import mimetypes
 from urllib.parse import urlparse, unquote
 import google.generativeai as genai
 
+from agent import Agent
+from constantes import LISTA_TASKS_PROCESSAR
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -95,7 +98,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = Agent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -131,11 +134,17 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        question_file_name = item.get("file_name")
+        
+        if not task_id in LISTA_TASKS_PROCESSAR:
+            print(f"Task {"task_id"} não está na lista de execução")
+            continue
+        
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, task_id, question_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:

constantes.py

@@ -27,4 +27,12 @@ GEMINI_MODEL = "gemini-2.0-flash"
 CHESSVISION_TO_FEN_URL = "http://app.chessvision.ai/predict"
 CHESS_MOVE_API = "https://chess-api.com/v1"
 
-AGENTS_FILES_PATH = "./files"
+AGENTS_FILES_PATH = "./tmp"
+HUGGINGFACE_DATASET_URL_TEMPLATE = (
+    "https://huggingface.co/datasets/gdms/gaia/resolve/main/{filename}"
+)
+
+LISTA_TASKS_PROCESSAR = [
+    "8e867cd7-cff9-4e6c-867a-ff5ddc2550be",
+    
+]

file_util.py

@@ -3,7 +3,9 @@ import re
 import shutil
 from typing import List
 
-from constantes import AGENTS_FILES_PATH
+import requests
+
+from constantes import AGENTS_FILES_PATH, HUGGINGFACE_DATASET_URL_TEMPLATE
 
 
 
@@ -56,25 +58,32 @@ class File_Util:
 
 
     @staticmethod
-    def tratar_arquivo_local(caminho_entrada: str) -> str:
+    def baixa_arquivo_task(task_file_name: str) -> str:
         """
-        Verifica se o arquivo existe no caminho informado.
-        Se não existir, adiciona o path padrão do sistema.
+        Baixa o arquivo da task para o diretório temporario local
 
         Parâmetros:
-        - caminho_entrada (str): Caminho completo ou nome do arquivo.
+        - task_file_name (str): Caminho completo ou nome do arquivo.
 
         Retorna:
         - str: Caminho válido para o arquivo, ou None se não encontrado.
         """
         # Verifica se o arquivo já existe no caminho informado
-        if os.path.isfile(caminho_entrada):
-            return caminho_entrada
+        task_file_path= os.path.join(AGENTS_FILES_PATH, task_file_name)
+        if os.path.isfile(task_file_path):
+            return task_file_path
+
+        #Tenta baixar do dataset
+        url = HUGGINGFACE_DATASET_URL_TEMPLATE.format(filename=task_file_name)
+        
+        try:
+            response = requests.get(url)
+            response.raise_for_status()
 
-        # Tenta procurar no diretório padrão, se fornecido
-        novo_caminho = os.path.join(AGENTS_FILES_PATH, os.path.basename(caminho_entrada))
-        if os.path.isfile(novo_caminho):
-            return novo_caminho
+            with open(task_file_path, "wb") as f:
+                f.write(response.content)
 
-        # Arquivo não encontrado
-        return None
+            return task_file_path
+        except Exception as e:
+            print(f"[ERRO] Problemas ao baixar arquivo {task_file_name}")
+            return None