Update app.py

app.py

@@ -12,8 +12,6 @@ import torchaudio
 import gradio as gr
 from speechbrain.inference.classifiers import EncoderClassifier
 from yt_dlp import YoutubeDL
-from yt_dlp.utils import DownloadError
-from pytubefix import YouTube as PyTubeFixYT
 
 # ─────────────── Model setup ───────────────
 ACCENT_MODEL_ID = "Jzuluaga/accent-id-commonaccent_ecapa"
@@ -40,34 +38,19 @@ def sec_to_hms(sec: int) -> str:
 
 def download_audio(url: str, out_path: Path) -> Path:
     """
-    1) Try yt-dlp Python API for all URLs except YouTube.
-    2) If yt-dlp raises DownloadError on YouTube, fall back to pytubefix.
-    Returns the actual downloaded file path (.m4a, .webm, etc.).
+    Download best audio only via yt_dlp Python API.
+    Returns the actual file saved (could be .m4a, .webm, etc.).
     """
-    ydl_opts = {
+    opts = {
         "format": "bestaudio/best",
         "outtmpl": str(out_path.with_suffix(".%(ext)s")),
-        "postprocessors": [],
+        "postprocessors": [],  
         "quiet": True,
     }
-    try:
-        with YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=True)
-            return Path(ydl.prepare_filename(info))
-    except DownloadError:
-        if "youtube.com" in url or "youtu.be" in url:
-            # fallback to pytubefix
-            try:
-                yt = PyTubeFixYT(url,use_po_token=True)
-                audio_stream = yt.streams.get_audio_only()
-                out_file = audio_stream.download(
-                    output_path=str(out_path.parent),
-                    filename=out_path.stem + ".mp4"
-                )
-                return Path(out_file)
-            except HTTPError as e:
-                raise RuntimeError(f"pytubefix download failed: {e}") from e
-        raise  # re-raise for non-YouTube errors
+    with YoutubeDL(opts) as ydl:
+        info = ydl.extract_info(url, download=True)
+        filename = ydl.prepare_filename(info)
+    return Path(filename)
 
 def extract_wav(src: Path, dst: Path, start: int, dur: int = 8) -> None:
     target_sr = 16000
@@ -189,4 +172,4 @@ if __name__ == "__main__":
     parser.add_argument("--share", action="store_true",
                         help="Enable public share link")
     args = parser.parse_args()
-    app().launch(share=args.share)
+    app().launch(share=args.share)