Update app.py

app.py

@@ -15,6 +15,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 from smolagents import CodeAgent, LiteLLMModel 
 from my_tools import my_tool_list  
 
+def download_file(task_id, filename, save_dir="attachments"):
+    os.makedirs(save_dir, exist_ok=True)
+    url = f"https://agents-course-unit4-scoring.hf.space/files/{task_id}"
+    save_path = os.path.join(save_dir, filename)
+    try:
+        resp = requests.get(url, timeout=15)
+        resp.raise_for_status()
+        with open(save_path, "wb") as f:
+            f.write(resp.content)
+        print(f"Downloaded attachment for task {task_id} -> {save_path}")
+        return save_path
+    except Exception as e:
+        print(f"Attachment download failed for {task_id}: {e}")
+        return None
+
 class BasicAgent:
     def __init__(self):
         api_key = os.getenv("OPENAI_API_KEY")   # ← Read enviroment variables in space. 
@@ -121,19 +136,39 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for idx, item in enumerate(questions_data[:3]):
+    for idx, item in enumerate(questions_data[3:4]):
         task_id = item.get("task_id")
         question_text = item.get("question")
+        file_list = item.get("files", [])
+        
+        local_files = []
+        for fname in file_list:
+            fpath = download_file(task_id, fname)
+            if fpath:
+                local_files.append(fpath)
+        if local_files:
+            print(f"Downloaded attachments: {local_files}")
+        
         print(f"===== [Celum is answering No. {idx+1}/{len(questions_data)} ] =====")
         try:
             submitted_answer = safe_run_agent(agent, question_text, idx, len(questions_data))
             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})
+            results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": submitted_answer,
+                "Files": local_files
+            })
         except Exception as e:
             print(f"[Celum Error at Q{idx+1}]: {e}")
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+            results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": f"AGENT ERROR: {e}",
+                "Files": local_files
+            })
         time.sleep(7)
-
+    
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)