update

app.py

@@ -26,13 +26,18 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 @tool
-def get_file_location(file_name: str) -> str:
+def get_image(file_name: str) -> Image.Image:
     """
-    A tool that resolve the file URL for a question.
+    A tool that load image from file name.
     Args:
         file_name: File name.
     """
-    return f"{DEFAULT_API_URL}/files/{file_name}"
+    import requests
+    from PIL import Image
+    from io import BytesIO
+    url = f"{DEFAULT_API_URL}/files/{file_name}"
+    response = requests.get(url)
+    return Image.open(BytesIO(response.content)).convert("RGB")
 
 
 # --- Basic Agent Definition ---
@@ -41,7 +46,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.multimodal_agent = CodeAgent(
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool()],
             model=InferenceClientModel(
                 model_id="meta-llama/Llama-3.2-11B-Vision-Instruct",
             ),
@@ -70,7 +75,7 @@ class BasicAgent:
         ) 
         
         self.code_agent = CodeAgent(
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image],
             model=InferenceClientModel(
                 model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             ),
@@ -92,7 +97,11 @@ class BasicAgent:
                 "chess.engine",
             ],
             name="code_agent",
-            description="I am strong at writing code to solve problem.",
+            description="""
+                I am strong at: 
+                    - Writing code to solve problem. 
+                    - Browse the web to find information.
+            """,
             verbosity_level=0,
             max_steps=10,
         )
@@ -101,7 +110,7 @@ class BasicAgent:
             model=InferenceClientModel(
                 "Qwen/Qwen2.5-32B-Instruct"
             ),
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
+            tools=[get_image],
             managed_agents=[
                 self.multimodal_agent, 
                 self.code_agent],