Fixing lost tool code

tools.py

@@ -1,20 +1,12 @@
-import base64
 import os
 from typing import ClassVar
-from urllib.parse import urlparse
-import requests
 from smolagents import Tool
-import math
 import datetime
 from PIL import Image
 import pandas as pd
-import litellm
-import yt_dlp
 from prompts import SYSTEM_PROMPT
-from pytube import YouTube
 from PIL import Image
-import pytesseract
-from smolagents import DuckDuckGoSearchTool, WikipediaSearchTool
+from smolagents import WikipediaSearchTool
 from dotenv import load_dotenv
 import time
 from langchain_google_community import GoogleSearchAPIWrapper
@@ -152,7 +144,7 @@ class ExcelAndCSVTableInspectorTool(Tool):
 class YouTubeVideoAnalyzerTool(Tool):
     name: ClassVar[str] = "youtube_video_analyzer"
     description: ClassVar[str] = (
-        "Given a YouTube URL, extracts metadata, analyzes the video content, and answers user queries about it."
+        "Given a YouTube URL, analyzes the video content using Gemini (google-genai SDK) and answers user queries about it."
     )
     inputs: ClassVar[dict] = {
         "url": {"type": "string", "description": "Full YouTube video URL"},
@@ -164,51 +156,31 @@ class YouTubeVideoAnalyzerTool(Tool):
     output_type: ClassVar[str] = "string"
 
     def forward(self, url: str, user_prompt: str):
+        api_key = os.getenv("GEMINI_API_KEY")
+        if not api_key:
+            return "Error: GEMINI_API_KEY environment variable is not set."
         try:
-            parsed_url = urlparse(url)
-            if (
-                parsed_url.scheme not in ["http", "https"]
-                or "youtube.com" not in parsed_url.netloc
-            ):
-                return "Invalid YouTube URL. Please provide a valid URL."
-
-            try:
-
-                user_final_prompt = """
-                You are an AI video analyzer. A user wants to analyze the following YouTube video. 
-                Use your tools to extract information and analyze the video content.
-                ### User Request
-                {user_prompt}
-                ### Video URL
-                {url}
-                ### Instructions:
-                - Analyze the video content based on the user's request.
-                - Identify the main key thing needed to be analyzed.
-                - Provide the answer as per system prompt.
-                """
-
-                response = litellm.completion(
-                    api_key=os.getenv("GEMINI_API_KEY"),
-                    model=os.getenv("GEMINI_MODEL"),
-                    messages=[
-                        {"role": "system", "content": SYSTEM_PROMPT},
-                        {
-                            "role": "user",
-                            "content": user_final_prompt.format(
-                                user_prompt=user_prompt, url=url
-                            ),
-                        },
-                    ],
-                )
-
-                return response["choices"][0]["message"]["content"]
-            except Exception as e:
-                if "Sign in" in str(e):
-                    return "This video requires age verification or sign-in. Please provide a different video URL."
-                return f"Error accessing video: {str(e)}"
-
+            genai.configure(api_key=api_key)
+            model_name = os.getenv("GENAI_MODEL", "gemini-1.5-pro")
+            model = genai.GenerativeModel(model_name)
+            prompt = f"""
+You are an AI video analyzer. A user wants to analyze the following YouTube video.
+
+### Video URL
+{url}
+
+### User Request
+{user_prompt}
+
+### Instructions:
+- Analyze the video content based on the user's request.
+- Identify the main key thing needed to be analyzed.
+- Provide the answer as per system prompt.
+"""
+            response = model.generate_content([prompt])
+            return response.text if hasattr(response, "text") else str(response)
         except Exception as e:
-            return f"Error analyzing video: {e}"
+            return f"Error analyzing video with Gemini: {e}"
 
 
 # --- Math Tools ---
@@ -500,7 +472,7 @@ class ImageInfoTool(Tool):
             return f"Error loading image: {e}"
 
 
-class WikipediaTool(Tool):
+class WikipediaCustomTool(Tool):
     name: ClassVar[str] = "wikipedia_search_summary"
     description: ClassVar[str] = "Search Wikipedia and return max 3 results"
     inputs: ClassVar[dict] = {
@@ -637,6 +609,7 @@ agent_tools = [
     SortListTool(),
     UniqueListTool(),
     GoogleSearchTool(),
-    WikipediaSearchTool(),
+    WikipediaCustomTool(),
+    ArvixSearchTool(),
     AdvanceGoogleAISearchTool(),
 ]