Update app.py

app.py

@@ -44,9 +44,15 @@ openai = OpenAI(api_key=openai_api_key)
 
 from pytube import YouTube
 
+def clean_url(url):
+    # Extract just the video ID
+    match = re.search(r"(?:v=|shorts/)([a-zA-Z0-9_-]{11})", url)
+    video_id = match.group(1) if match else None
+    return f"https://www.youtube.com/watch?v={video_id}" if video_id else url
+
 def download_audio(youtube_url):
     try:
-        # First try yt-dlp
+        # Try yt-dlp first
         output_template = "/tmp/downloaded_audio.%(ext)s"
         for f in glob.glob("/tmp/downloaded_audio.*"):
             os.remove(f)
@@ -67,10 +73,11 @@ def download_audio(youtube_url):
             if files:
                 return files[0]
 
-        # If yt-dlp fails, fall back to pytube
-        print("🔁 yt-dlp failed, trying pytube fallback...")
+        # 🔁 Fallback to pytube
+        print("🔁 yt-dlp failed, trying pytube...")
 
-        yt = YouTube(youtube_url)
+        safe_url = clean_url(youtube_url)
+        yt = YouTube(safe_url)
         stream = yt.streams.filter(only_audio=True).first()
         output_path = "/tmp/fallback_audio.mp4"
         stream.download(filename=output_path)