Update app.py

app.py

@@ -20,6 +20,10 @@ from smolagents import Tool
 from youtube_transcript_api import YouTubeTranscriptApi
 from urllib.parse import urlparse, parse_qs
 
+from smolagents import Tool
+from youtube_transcript_api import YouTubeTranscriptApi
+from urllib.parse import urlparse, parse_qs
+
 class YouTubeDialogueTool(Tool):
     name = "youtube_dialogue_qa"
     description = "Extracts the transcript and finds what Teal'c says in response to a specific question."
@@ -35,7 +39,8 @@ class YouTubeDialogueTool(Tool):
         },
         "speaker": {
             "type": "string",
-            "description": "Name of the character who responds (e.g., 'Teal'c')"
+            "description": "Name of the character who responds (e.g., 'Teal'c')",
+            "nullable": True
         }
     }
 
@@ -43,26 +48,17 @@ class YouTubeDialogueTool(Tool):
 
     def forward(self, url: str, question: str, speaker: str = "Teal'c") -> str:
         try:
-            # Extract video ID
             parsed = urlparse(url)
             video_id = parse_qs(parsed.query).get("v", [None])[0]
             if not video_id:
                 return "ERROR: Invalid YouTube URL or missing video ID."
 
-            # Get transcript
             transcript = YouTubeTranscriptApi.get_transcript(video_id)
 
-            # Build list of lines
-            lines = []
-            for entry in transcript:
-                text = entry["text"].replace("\n", " ").strip()
-                if text:
-                    lines.append(text)
+            lines = [entry["text"].replace("\n", " ").strip() for entry in transcript if entry["text"].strip()]
 
-            # Find the question and capture the next line(s)
             for idx, line in enumerate(lines):
                 if question.lower().rstrip("?") in line.lower():
-                    # Look ahead for a line that matches speaker response style
                     if idx + 1 < len(lines):
                         return lines[idx + 1].strip()
                     break
@@ -74,6 +70,7 @@ class YouTubeDialogueTool(Tool):
 
 
 
+
 ##Tool 2
 import wikipedia
 from smolagents import Tool