Update tools.py

tools.py

@@ -1,16 +1,20 @@
 import tempfile
 import requests
 import os
-
 from urllib.parse import urlparse
 from typing import Optional, List
 import yt_dlp
 import imageio
-
 from PIL import Image
-from dotenv import load_dotenv
-from gaia_benchmark.tools import tool
 import whisper
+from dotenv import load_dotenv
+
+# Fallback tool decorator if gaia_benchmark.tools is not available
+try:
+    from gaia_benchmark.tools import tool
+except ImportError:
+    def tool(func):
+        return func
 
 load_dotenv()
 
@@ -34,11 +38,11 @@ def youtube_frames_to_images(url: str, sample_interval_seconds: int = 5) -> List
             'force_ipv4': True,
         }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=True)
+            ydl.download([url])
 
         video_path = next((os.path.join(tmpdir, f) for f in os.listdir(tmpdir) if f.endswith('.mp4')), None)
         reader = imageio.get_reader(video_path)
-        fps = reader.get_meta_data().get('fps')
+        fps = reader.get_meta_data().get('fps', 25)
         frame_interval = int(fps * sample_interval_seconds)
         images = [Image.fromarray(frame) for idx, frame in enumerate(reader) if idx % frame_interval == 0]
         reader.close()
@@ -107,28 +111,11 @@ def youtube_transcribe(url: str) -> str:
             'force_ipv4': True,
         }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.extract_info(url, download=True)
+            ydl.download([url])
+
         audio_path = next((os.path.join(tmpdir, f) for f in os.listdir(tmpdir) if f.endswith('.wav')), None)
         return model.transcribe(audio_path)['text']
 
 @tool
 def transcribe_audio(audio_file_path: str) -> str:
-    return whisper.load_model("small").transcribe(audio_file_path)['text']
-
-# 🔧 Stub Tools for Mistral Compatibility
-
-@tool
-def web_search(query: str) -> str:
-    return f"(Stub) Web search results for: {query}"
-
-@tool
-def wikipedia_search(topic: str) -> str:
-    return f"(Stub) Wikipedia summary for: {topic}"
-
-@tool
-def visit_webpage(url: str) -> str:
-    return f"(Stub) Visited webpage at: {url}"
-
-@tool
-def final_answer(response: str) -> str:
-    return f"(Stub) Final answer: {response}"
+    return whisper.load_model("small").transcribe(audio_file_path)['text']