Update tools.py

tools.py

@@ -14,4 +14,44 @@ def get_hub_stats(author: str) -> str:
         else:
             return f"No models found for author {author}."
     except Exception as e:
-        return f"Error fetching models for {author}: {str(e)}"
+        return f"Error fetching models for {author}: {str(e)}"
+
+
+def analyze_image(image_path: str, question: str = "What do you see in this image?") -> str:
+    """Analyze an image using LangChain's ChatOpenAI with vision"""
+    
+    def encode_image_to_base64(image_path: str) -> str:
+        """Convert image file to base64 string"""
+        try:
+            with open(image_path, "rb") as image_file:
+                return base64.b64encode(image_file.read()).decode('utf-8')
+        except Exception as e:
+            raise Exception(f"Error encoding image: {e}")
+    
+    try:
+        # Create vision-capable LLM
+        vision_llm = ChatOpenAI(model="gpt-4o", max_tokens=1000)
+        
+        # Encode the image
+        base64_image = encode_image_to_base64(image_path)
+        
+        # Create message with image
+        message = HumanMessage(
+            content=[
+                {"type": "text", "text": question},
+                {
+                    "type": "image_url",
+                    "image_url": {
+                        "url": f"data:image/jpeg;base64,{base64_image}",
+                        "detail": "high"
+                    }
+                }
+            ]
+        )
+        
+        # Get response
+        response = vision_llm.invoke([message])
+        return response.content
+        
+    except Exception as e:
+        return f"Error analyzing image: {e}"