Update app.py

app.py

@@ -4,21 +4,16 @@ import requests
 import inspect
 import pandas as pd
 
+from config import set_environment
+set_environment()
+
+from tools import upload_file_to_bucket
+from graph import generate_graph
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -40,7 +35,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = generate_graph()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -76,11 +71,40 @@ 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")
+        filename = item.get('file_name')
         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)
+            path_filename = None
+            gcp_uri_file = None
+            if filename != '':
+                url = f"{api_url}/files/{task_id}"
+        
+                with requests.get(url, stream=True, timeout=15) as r:
+                    r.raise_for_status() 
+                    # ── Obtener nombre de archivo ──────────────────────────────
+                    cd = r.headers.get("content-disposition", "")
+                    m  = re.search(r'filename="?([^";]+)', cd)    # RFC 6266
+                    filename = m.group(1) if m else f"{task_id}.bin"
+                    # ── Guardar en disco en chunks ─────────────────────────────
+                    path_filename = "data/" + filename
+                    with open(path_filename, "wb") as f:
+                        for chunk in r.iter_content(chunk_size=8192):
+                            if chunk:                             # evita keep-alives vacíos
+                                f.write(chunk)
+        
+                gcp_uri_file = upload_file_to_bucket(path_filename)
+                
+            #submitted_answer = agent(question_text)
+            result = agent.invoke({"messages": [("user",
+                                             task)],
+                               "check_final_answer": False,
+                               "path_filename": path_filename,
+                               "gcp_path": gcp_uri_file},
+                              {"recursion_limit": 20})
+            
+            submitted_answer = result['final_answer']
             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: