Update tools/web_search_tools.py

tools/web_search_tools.py

@@ -3,9 +3,6 @@ from bs4 import BeautifulSoup
 import json
 import os
 
-import requests
-from bs4 import BeautifulSoup
-
 class WebSearchTools:
 
     @staticmethod
@@ -40,11 +37,10 @@ class WebSearchTools:
         return text
 
 # Example usage
-image_urls = WebSearchTools.search_images("random forest")
-for idx, img_url in enumerate(image_urls):
-    WebSearchTools.download_image(img_url, f'image_{idx}.png')
-
-        return results
+if __name__ == "__main__":
+    image_urls = WebSearchTools.search_images("random forest")
+    for idx, img_url in enumerate(image_urls):
+        WebSearchTools.download_image(img_url, f'image_{idx}.png')