Update image_analyzer.py

image_analyzer.py

@@ -1,4 +1,5 @@
 import base64
+import requests
 import openai
 from smolagents import Tool
 
@@ -6,23 +7,27 @@ class ImageAnalysisTool(Tool):
     name = "image_analysis"
     description = "Analyze the content of an image and answer a specific question about it."
     inputs = {
-        "image_path": {
+        "url": {
             "type": "string",
-            "description": "Path to the image file (jpg, png, etc.)"
+            "description": "URL to the image"
         },
         "question": {
             "type": "string",
-            "description": "A question about the image content"
+            "description": "Question about the image content"
         }
     }
     output_type = "string"
 
-    def __init__(self):
-        super().__init__()
-
-    def forward(self, image_path: str, question: str) -> str:
-        base64_image = self.encode_image(image_path)
+    def forward(self, url: str, question: str) -> str:
         try:
+            # Download image
+            image_path = "/tmp/image_input.jpg"
+            r = requests.get(url)
+            with open(image_path, "wb") as f:
+                f.write(r.content)
+
+            # Encode & analyze
+            base64_image = self.encode_image(image_path)
             response = openai.ChatCompletion.create(
                 model="gpt-4-turbo",
                 messages=[
@@ -41,7 +46,7 @@ class ImageAnalysisTool(Tool):
                 ],
                 max_tokens=300
             )
-            return response["choices"][0]["message"]["content"]
+            return response["choices"][0]["message"]["content"].strip()
         except Exception as e:
             return f"Error analyzing image: {e}"
 
@@ -50,4 +55,5 @@ class ImageAnalysisTool(Tool):
             return base64.b64encode(image_file.read()).decode("utf-8")
 
 
+