Update app.py

app.py

@@ -20,25 +20,22 @@ from smolagents import Tool
 from youtube_transcript_api import YouTubeTranscriptApi
 import re
 
+from smolagents import Tool
+
 class YouTubeTranscriptTool(Tool):
     name = "youtube_transcript"
-    description = "Use this to extract spoken content from a YouTube video given its URL."
+    description = "Extracts transcript text from a YouTube video URL"
+    inputs = {"url": str}
+    outputs = {"transcript": str}
+
+    def __call__(self, url: str) -> dict:
+        # Example logic (replace with real code)
+        from youtube_transcript_api import YouTubeTranscriptApi
+        video_id = url.split("v=")[-1].split("&")[0]
+        transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
+        transcript_text = " ".join([t["text"] for t in transcript_list])
+        return {"transcript": transcript_text}
 
-    def use(self, input_text: str) -> str:
-        try:
-            # Extract video ID from URL
-            video_id_match = re.search(r"(?:v=|youtu\.be/)([a-zA-Z0-9_-]{11})", input_text)
-            if not video_id_match:
-                return "Invalid YouTube URL provided."
-
-            video_id = video_id_match.group(1)
-            transcript = YouTubeTranscriptApi.get_transcript(video_id)
-            
-            # Combine into a single string
-            full_text = " ".join([entry['text'] for entry in transcript])
-            return full_text[:3000]  # limit to 3000 chars
-        except Exception as e:
-            return f"Error fetching transcript: {e}"
 transcript_tool = YouTubeTranscriptTool()
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):