update

app.py

@@ -25,19 +25,24 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 @tool
-def get_image(file_name: str) -> Image.Image:
+def get_image(file_name: str) -> str:
     """
-    A tool that load image from file name.
+    A tool that downloads an image from file name and saves it locally.
     Args:
         file_name: File name.
+    Returns:
+        str: Local file path where the image was saved.
     """
     import requests
-    from PIL import Image
-    from io import BytesIO
+    import os
     
     url = f"{DEFAULT_API_URL}/files/{file_name}"
     print(f"Fetching image from URL: {url}")
     
+    # Create downloads directory if it doesn't exist
+    downloads_dir = "downloaded_images"
+    os.makedirs(downloads_dir, exist_ok=True)
+    
     response = None
     try:
         response = requests.get(url, timeout=30)
@@ -52,18 +57,23 @@ def get_image(file_name: str) -> Image.Image:
         print(f"Response content-type: {content_type}")
         print(f"Response content length: {len(response.content)} bytes")
         
-        # Try to open the image
-        image_bytes = BytesIO(response.content)
-        image = Image.open(image_bytes)
-        return image
+        # Use original filename directly
+        local_path = os.path.join(downloads_dir, file_name)
+        
+        # Save the image to local file
+        with open(local_path, 'wb') as f:
+            f.write(response.content)
+        
+        print(f"Image saved to: {local_path}")
+        return local_path
         
     except requests.exceptions.RequestException as e:
         raise ValueError(f"Failed to fetch image from {url}: {e}")
     except Exception as e:
         # Print first 200 characters of response content for debugging
         content_preview = response.content[:200] if response and hasattr(response, 'content') else b"No response"
-        print(f"Error opening image. Content preview: {content_preview}")
-        raise ValueError(f"Failed to open image from {url}: {e}")
+        print(f"Error downloading image. Content preview: {content_preview}")
+        raise ValueError(f"Failed to download image from {url}: {e}")
 
 
 # --- Basic Agent Definition ---