Update app.py

updated the YouTube tool

app.py

@@ -14,7 +14,11 @@ from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 import base64
 from google.ai.generativelanguage_v1beta.types import Tool as GenAITool
-from google.ai.generativelanguage_v1beta.types import FileData
+from google import genai
+from google.genai import types
+import os
+
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -117,26 +121,32 @@ def open_image(task_id: str) -> str:
     
     return base64.b64encode(image_file.read()).decode("utf-8")
 
-def open_youtube_video(url: str) -> HumanMessage:
+def open_youtube_video(url: str, query:str) -> str:
     '''
-    Opens a video file from the given URL.
+    Answers a question about a video from the given URL.
     Args:
         url: The URL of the video file.
+        query: The question to be answered about the video.
     Returns:
-        HumanMessage instructions for the video file.
+        Answer to the question about the video.
     '''
-    video = FileData(url=url)
-
-    return HumanMessage(
-        content=[
-            {"type": "text", "text": "Watch the video and answer the question."},
-            {
-                "type": "media",
-                "data": video,
-                "mime_type": "video/mp4",
-            },
-        ]
-    )
+    
+    client = genai.Client(api_key=GOOGLE_API_KEY)
+
+    response = client.models.generate_content(
+    model='models/gemini-2.0-flash',
+    contents=types.Content(
+        parts=[
+            types.Part(
+                file_data=types.FileData(file_uri=url)
+            ),
+            types.Part(text=f'''{query} YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated 
+                       list of numbers and/or strings.''')
+            ]
+            )
+            )
+    
+    return response.text
 
 def google_search(query: str) -> str:
     '''
@@ -184,7 +194,7 @@ class BasicAgent:
         If you are asked to transcribe an audio file, use the 'transcribe_audio' tool.
         If you are asked to run a Python code, use the 'python_code' tool.
         If you are asked to open an image, use the 'open_image' tool.
-        If you are asked to open a YouTube video, use the 'open_video' tool.
+        If you were given a link with www.youtube.com, use the 'open_youtube_video' tool.
         If the question requires a web search because your internal knowledge doesn't have the information, use the 'google_search' tool.
                         ''')