Update tools/tools.py

tools/tools.py

@@ -74,24 +74,20 @@ def ask_youtube_video(url: str, question: str) -> str:
     Sends a YouTube video to a multimodal model and asks a question about it.
 
     Args:
-        url (str): The URI of the video file (already uploaded and hosted).
+        url (str): The URL of the YouTube video.
         question (str): The natural language question to ask about the video.
 
     Returns:
         str: The model's answer to the question.
     """
-
     try:
-        client = genai.Client(api_key=os.getenv('GEMINI_API'))
-        response = client.generate_content(
-            model=MODEL_ID,
-            contents=[
-                {"role": "user", "parts": [
-                    {"text": question},
-                    {"file_data": {"file_uri": url}}
-                ]}
-            ]
-        )
+        genai.configure(api_key=os.getenv('GEMINI_API'))
+        model = genai.GenerativeModel(MODEL_ID)
+        
+        # Gemini natively supports YouTube links passed directly in the prompt!
+        prompt = f"Watch this video carefully: {url}\n\nQuestion: {question}"
+        response = model.generate_content(prompt)
+        
         return response.text
     except Exception as e:
         return f"Error asking {MODEL_ID} about video: {str(e)}"