a

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from pathlib import Path
+from PIL import Image as PILImage
 
 from agent import agent
 
@@ -13,16 +15,43 @@ print("Gemini_Key in env:", "Gemini_Key" in os.environ)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# Helper function to check for images
+def is_image_file(filepath: Path) -> bool:
+    if not filepath or not filepath.is_file():
+        return False
+    try:
+        with PILImage.open(filepath) as img:
+            img.verify()  # Verify it's an image without loading all pixel data
+        return True
+    except Exception:  # Catches PIL.UnidentifiedImageError, IOError, SyntaxError, etc.
+        return False
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = agent
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = self.agent.run(question)
-        print(f"Agent returning  answer: {answer}")
+    def __call__(self, task_id: str, question: str, filename: Path | None = None) -> str:
+        print(f"Agent received task_id: {task_id}, question (first 50 chars): {question[:50]}...")
+        if filename:
+            print(f"Associated file: {filename.name}")
+
+        # Augment the question for the LLM
+        prompt = f"Task ID: {task_id}\\nQuestion: {question}"
+        if filename:
+            # Still inform about the file in the text prompt, as a fallback or for context
+            prompt += f"\\n\\nThis question may relate to an attached file named: '{filename.name}'. The file has been downloaded if it was provided with the question."
+
+        run_kwargs = {}
+        if filename and is_image_file(filename):
+            print(f"File {filename.name} is an image. Adding to agent.run arguments.")
+            # Assuming agent.run can take an 'images' kwarg with a list of Path objects or PIL.Image objects
+            # Passing Path object directly for simplicity, smolagents might handle it.
+            run_kwargs["images"] = [filename]
+        
+        answer = self.agent.run(prompt, **run_kwargs)
+        print(f"Agent returning answer: {answer}")
         return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
@@ -54,6 +83,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)
 
+    # Create a directory for downloaded files
+    files_dir = Path("downloaded_files")
+    files_dir.mkdir(exist_ok=True)
+    print(f"Files will be downloaded to: {files_dir.resolve()}")
+
     # 2. Fetch Questions
     print(f"Fetching questions from: {questions_url}")
     try:
@@ -82,16 +116,54 @@ 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")
+        file_name_str = item.get("file_name") # Check for a filename
+        file_path: Path | None = None # Initialize file_path
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+
+        if file_name_str:
+            print(f"Task {task_id} has an associated file: {file_name_str}")
+            # Construct download URL - using the structure suggested by GaiaClient in the prompt
+            # The API seems to be GET /files/{task_id} and filename is for local saving
+            file_url = f"{api_url}/files/{task_id}"
+            local_file_path = files_dir / file_name_str
+            
+            try:
+                print(f"Attempting to download file from: {file_url} to {local_file_path}")
+                response = requests.get(file_url, timeout=30) # Increased timeout for file download
+                response.raise_for_status() # Will raise an HTTPError for bad responses (4XX or 5XX)
+                with open(local_file_path, "wb") as f:
+                    f.write(response.content)
+                file_path = local_file_path
+                print(f"Successfully downloaded {file_name_str} for task {task_id}.")
+            except requests.exceptions.HTTPError as e_http:
+                print(f"HTTP error downloading file for task {task_id} ({file_name_str}): {e_http}")
+                print(f"Response status: {e_http.response.status_code}, Response text: {e_http.response.text[:200]}")
+            except requests.exceptions.RequestException as e_req:
+                print(f"Error downloading file for task {task_id} ({file_name_str}): {e_req}")
+            except Exception as e_file:
+                print(f"An unexpected error occurred downloading file for task {task_id} ({file_name_str}): {e_file}")
+        
         try:
-            submitted_answer = agent(question_text)
+            # Pass task_id, original question_text, and file_path to the agent
+            submitted_answer = agent(task_id, question_text, file_path)
             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,
+                "File": file_name_str if file_name_str else "N/A", # Log file name
+                "Submitted Answer": submitted_answer
+            })
         except Exception as e:
              print(f"Error running agent on task {task_id}: {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,
+                 "File": file_name_str if file_name_str else "N/A",
+                 "Submitted Answer": f"AGENT ERROR: {e}"
+            })
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")