Update image_search_tool.py

image_search_tool.py

@@ -5,17 +5,12 @@ from io import BytesIO
 import json
 import os
 
-from crewai_tools import tool
-
-@tool
 def image_search_tool(query: str, num_images: int = 5) -> str:
     """
     A tool to search for images based on the given query and download them.
-
     Parameters:
     - query (str): The search query for finding images.
     - num_images (int): The number of images to retrieve (default is 5).
-
     Returns:
     - str: The paths where the downloaded images are saved.
     """
@@ -34,10 +29,15 @@ def image_search_tool(query: str, num_images: int = 5) -> str:
     results = []
     for img_tag in soup.find_all('a', class_='iusc')[:num_images]:
         m = img_tag.get('m')
-        m_json = json.loads(m)
-        img_url = m_json.get('murl')
-        img_title = m_json.get('turl')
-        results.append({'title': img_title, 'url': img_url})
+        if m:
+            try:
+                m_json = json.loads(m)
+                img_url = m_json.get('murl')
+                img_title = m_json.get('turl')
+                if img_url:
+                    results.append({'title': img_title, 'url': img_url})
+            except (json.JSONDecodeError, TypeError) as e:
+                continue
 
     image_paths = []
     folder = 'images'
@@ -54,4 +54,7 @@ def image_search_tool(query: str, num_images: int = 5) -> str:
         except Exception as e:
             return f"Failed to download {result['title']}: {e}"
 
-    return f"Downloaded images to: {', '.join(image_paths)}"
+    return f"Downloaded images to: {', '.join(image_paths)}"
+
+# Example usage:
+print(image_search_tool("artificial intelligence", 3))