Update app.py

app.py

@@ -1,5 +1,5 @@
 '''
-Copyright 2023-2024 LangBridge Inc.
+Copyright 2023-2025 LangBridge Inc.
 All Rights Reserved.
 LangBridge Restricted
 '''
@@ -17,7 +17,7 @@ import torch
 import whisperx
 from moviepy import afx
 from moviepy.audio.AudioClip import AudioArrayClip
-from pytube import YouTube
+import yt_dlp
 from TTS.api import TTS
 
 
@@ -282,13 +282,28 @@ def check_video_duration(video_path):
     return duration > 180
 
 def download_youtube_video(url):
-  yt = YouTube(url)
-  if yt.age_restricted:
-    gr.Warning("The Youtube video you are trying to translate is age restricted. Manually download it using the following link(https://en.savefrom.net/) and use file upload, as pytube library doesn't support restricted videos download.")
-    return None
-  stream = yt.streams.filter(file_extension='mp4').first()
-  output_path = stream.download()
-  return output_path
+    ydl_opts = {
+        "format": "mp4/best[ext=mp4]",  # Prefer mp4 format
+        "outtmpl": "%(title)s.%(ext)s",  # Output filename template
+    }
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        try:
+            info = ydl.extract_info(url, download=False)
+
+            if info.get("age_limit", 0) > 0:
+                gr.Warning(
+                    "The Youtube video you are trying to translate is age restricted. Manually download it using the following link(https://en.savefrom.net/) and use file upload, as yt-dlp may not support restricted videos download."
+                )
+                return None
+
+            ydl.download([url])
+
+            return ydl.prepare_filename(info)
+
+        except Exception as e:
+            gr.Error(f"An error occurred while downloading the video: {e}")
+            return None
 
 def translation_limit():
     translator = deepl.Translator(DEEPL_TOKEN)