Update tools.py

tools.py

@@ -4,8 +4,8 @@ from langchain_core.messages import HumanMessage
 from langchain.tools import tool
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
-from langchain_core.video import VideoFile
 import yt_dlp
+import ffmpeg
 
 
 @tool
@@ -208,6 +208,7 @@ class TranscribeAudio:
             return ""
 
 
+@tool
 def download_youtube_video(youtube_url: str, output_path: str) -> str:
     """
     Download a YouTube video as an MP4 file.
@@ -230,44 +231,30 @@ def download_youtube_video(youtube_url: str, output_path: str) -> str:
     return output_path
 
 
-class AnalyzeVideo:
-    def __init__(self, multimodal_model):
-        self.multimodal_model = multimodal_model
-
-    def __call__(self, video_path: str, query: str) -> str:
-        try:
-            with open(video_path, "rb") as video_file:
-                video_bytes = video_file.read()
-
-            video_data = VideoFile(
-                mime_type="video/mp4",
-                data=video_bytes
-            )
-
-            message = [
-                HumanMessage(
-                    content=[
-                        {
-                            "type": "text",
-                            "text": (
-                                f"In relation to this video, answer the following request: {query} "
-                            ),
-                        },
-                        {
-                            "type": "video",
-                            "video": video_data,
-                        },
-                    ]
-                )
-            ]
-
-            response = self.multimodal_model.invoke(message)
-            return response.content.strip()
+@tool
+def extract_audio_from_video(video_path: str, audio_output: str) -> str:
+    """
+    Extracts audio from an MP4 video file and saves it as MP3.
 
-        except Exception as e:
-            print(f"Error analyzing video: {str(e)}")
-            return ""
+    Args:
+        video_path: Path to the input MP4 video file.
+        audio_output: Path for the output MP3 file.
 
+    Returns:
+        Path to the audio file.
+    """
+    try:
+        (
+            ffmpeg
+            .input(video_path)
+            .output(audio_output, format='mp3', acodec='libmp3lame', t=60)  # limit to 60 sec
+            .overwrite_output()
+            .run(quiet=True)
+        )
+        return audio_output
+    except ffmpeg.Error as e:
+        raise RuntimeError(f"FFmpeg error: {e.stderr.decode()}") from e
+        
     
 @tool
 def wiki_search(query: str) -> str: