update

app.py

@@ -26,33 +26,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 @tool
-def get_image_content(file_name: str) -> Image.Image | None:
+def get_file_location(file_name: str) -> str:
     """
-    A tool that load the content of an image for a question.
+    A tool that resolve the file URL for a question.
     Args:
-        file_name: File name to load.
+        file_name: File name.
     """
-    from io import BytesIO
-
-    import requests
-    from PIL import Image
-
-    try:
-        # Construct the API endpoint URL for file retrieval
-        file_url = f"{DEFAULT_API_URL}/files/{file_name}"
-
-        # Make GET request to retrieve file contents
-        response = requests.get(file_url, timeout=30)
-        response.raise_for_status()
-
-        image = Image.open(BytesIO(response.content)).convert("RGB")
-        # Return the file contents as text
-        return image
-
-    except requests.exceptions.RequestException as e:
-        print(f"Error retrieving file '{file_name}': {e}")
-    except Exception as e:
-        print(f"Unexpected error retrieving file '{file_name}': {e}")
+    return f"{DEFAULT_API_URL}/files/{file_name}"
 
 
 # --- Basic Agent Definition ---
@@ -61,7 +41,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.multimodal_agent = CodeAgent(
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image_content],
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
             model=InferenceClientModel(
                 model_id="meta-llama/Llama-3.2-11B-Vision-Instruct",
             ),
@@ -83,7 +63,7 @@ class BasicAgent:
         )
 
         self.code_agent = CodeAgent(
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image_content],
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
             model=InferenceClientModel(
                 model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             ),