Update tools.py

tools.py

@@ -8,7 +8,7 @@ from urllib.parse import urlparse
 from openai import OpenAI
 from youtube_transcript_api import YouTubeTranscriptApi
 from pytube import extract
-
+from openai import OpenAI
 
 @tool
 def add(a: float, b: float) -> float:
@@ -95,7 +95,6 @@ def transcribe_audio(audio_file: str, file_extension: str) -> str:
     Returns:
         str: The transcribed text from the audio.
     """
-    
     try:
         response = requests.get(audio_file)  # download the audio_file
         response.raise_for_status()  # check if the http request was successful
@@ -137,4 +136,32 @@ def transcribe_youtube(youtube_url: str) -> str:
         return text
         
     except Exception as e:
-        return f"transcribe_youtube failed: {e}"
+        return f"transcribe_youtube failed: {e}"
+
+
+# TESTING
+@tool
+def query_image(query: str, image_url: str) -> str:
+    """ Ask anything about an image using a Vision Language Model
+    Args:
+        query (str): the query about the image, e.g. how many animals are on the image?
+        image_url (str): the image's URL
+    """
+    try:
+        client = OpenAI()
+        response = client.responses.create(
+            model="gpt-4.1-nano",
+            input=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "input_text", "text": query},
+                        {"type": "input_image","image_url": image_url},
+                    ],
+                }
+            ],
+        )
+        return response.output_text
+
+    except Exception as e:
+        return f"query_image failed: {e}"