Update app.py

app.py

@@ -103,14 +103,16 @@ def python_execution(code: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
+from typing import Optional
 from langchain_core.tools import BaseTool
 from youtube_transcript_api import YouTubeTranscriptApi
 
 class VideoTranscriptionTool(BaseTool):
-    name = "transcript_video"
-    description = "Fetch text transcript from YouTube videos using URL or ID. Optionally include timestamps."
+    name: str = "transcript_video"
+    description: str = "Fetch text transcript from YouTube videos using URL or ID. Optionally include timestamps."
 
-    def _run(self, url: str, include_timestamps: bool = False) -> str:
+    def _run(self, url: str, include_timestamps: Optional[bool] = False) -> str:
+        # Extract video ID
         video_id = None
         if "youtube.com/watch?v=" in url:
             video_id = url.split("v=")[1].split("&")[0]
@@ -118,7 +120,7 @@ class VideoTranscriptionTool(BaseTool):
             video_id = url.split("youtu.be/")[1].split("?")[0]
         elif len(url.strip()) == 11 and not ("http://" in url or "https://" in url):
             video_id = url.strip()
-        
+
         if not video_id:
             return f"Invalid or unsupported YouTube URL/ID: {url}"
 
@@ -138,7 +140,7 @@ class VideoTranscriptionTool(BaseTool):
             return f"Error fetching transcript: {str(e)}"
 
     def _arun(self, *args, **kwargs):
-        raise NotImplementedError("This tool does not support async yet.")
+        raise NotImplementedError("Async not supported for this tool.")