Update app.py

app.py

@@ -1,15 +1,25 @@
 import gradio as gr
 from pytube import YouTube
+from urllib.parse import urlparse, parse_qs
 
 def get_youtube_transcript(video_url):
     try:
-        yt = YouTube(video_url)
-        caption = yt.captions.get_by_language_code('en') # You can adjust the language code
-        if caption:
-            transcript = caption.generate_srt_captions() # Or .generate_vtt_captions() for VTT format
-            return transcript
+        parsed_url = urlparse(video_url)
+        if parsed_url.netloc == "www.youtube.com" and parsed_url.path == "/watch":
+            query_params = parse_qs(parsed_url.query)
+            if "v" in query_params:
+                video_id = query_params["v"][0]
+                yt = YouTube(f"https://www.youtube.com/watch?v={video_id}")
+                caption = yt.captions.get_by_language_code('en')
+                if caption:
+                    transcript = caption.generate_srt_captions()
+                    return transcript
+                else:
+                    return "No English transcript found for this video."
+            else:
+                return "Invalid YouTube video URL."
         else:
-            return "No English transcript found for this video."
+            return "Invalid YouTube video URL."
     except Exception as e:
         return f"An error occurred: {e}"