Update app.py

app.py

@@ -11,16 +11,49 @@ from openai import OpenAI
 openai_api_key = os.getenv("OPENAI_API_KEY")
 openai = OpenAI(api_key=openai_api_key)
 
+# def download_audio(youtube_url):
+#     try:
+#         output_template = "/tmp/downloaded_audio.%(ext)s"
+
+#         # Remove any old files
+#         for f in glob.glob("/tmp/downloaded_audio.*"):
+#             os.remove(f)
+
+#         command = [
+#             "yt-dlp", "-f", "bestaudio",
+#             "--extract-audio", "--audio-format", "mp3",
+#             "--audio-quality", "0",
+#             "-o", output_template,
+#             youtube_url
+#         ]
+
+#         result = subprocess.run(command, capture_output=True, text=True)
+#         print("stdout:\n", result.stdout)
+#         print("stderr:\n", result.stderr)
+
+#         if result.returncode != 0:
+#             raise RuntimeError(f"yt-dlp failed: {result.stderr}")
+
+#         files = glob.glob("/tmp/downloaded_audio.*")
+#         if not files:
+#             raise FileNotFoundError("No audio file downloaded.")
+
+#         return files[0]
+#     except Exception as e:
+#         raise RuntimeError(f"Download error: {e}")
+
+from pytube import YouTube
+
 def download_audio(youtube_url):
     try:
+        # First try yt-dlp
         output_template = "/tmp/downloaded_audio.%(ext)s"
-
-        # Remove any old files
         for f in glob.glob("/tmp/downloaded_audio.*"):
             os.remove(f)
 
         command = [
-            "yt-dlp", "-f", "bestaudio",
+            "yt-dlp",
+            "-f", "bestaudio",
             "--extract-audio", "--audio-format", "mp3",
             "--audio-quality", "0",
             "-o", output_template,
@@ -28,17 +61,21 @@ def download_audio(youtube_url):
         ]
 
         result = subprocess.run(command, capture_output=True, text=True)
-        print("stdout:\n", result.stdout)
-        print("stderr:\n", result.stderr)
 
-        if result.returncode != 0:
-            raise RuntimeError(f"yt-dlp failed: {result.stderr}")
+        if result.returncode == 0:
+            files = glob.glob("/tmp/downloaded_audio.*")
+            if files:
+                return files[0]
+
+        # If yt-dlp fails, fall back to pytube
+        print("🔁 yt-dlp failed, trying pytube fallback...")
 
-        files = glob.glob("/tmp/downloaded_audio.*")
-        if not files:
-            raise FileNotFoundError("No audio file downloaded.")
+        yt = YouTube(youtube_url)
+        stream = yt.streams.filter(only_audio=True).first()
+        output_path = "/tmp/fallback_audio.mp4"
+        stream.download(filename=output_path)
+        return output_path
 
-        return files[0]
     except Exception as e:
         raise RuntimeError(f"Download error: {e}")