Update app.py

app.py

@@ -13,9 +13,14 @@ from melo.api import TTS
 from concurrent.futures import ThreadPoolExecutor
 import ffmpeg
 
-def process_video(video_file, language_choice):
+def process_video(video_file, youtube_url, language_choice):
     if video_file == None or language_choice == None:
         return None
+    if video_file != None and youtube_url != None:
+        return None
+    else if youtube_url != None:
+        youtube_command = "youtube-dl " + youtube_url
+        os.system(youtube_command)
     
     # Initialize paths and devices
     ckpt_converter = 'checkpoints_v2/converter'
@@ -258,10 +263,11 @@ language_choices.remove("auto")
 gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Video(label="Upload Video", sources=['upload']),
+        gr.Video(label="Upload a video from your device storage", sources=['upload']),
+        gr.Textbox(label="OR enter a YouTube video URL"),
         gr.Dropdown(choices=language_choices, label="Choose Language for Translation (Expressed in ISO 639-1 code)")
     ],
-    outputs=gr.Video(label="Translated Video"),
+    outputs=gr.Video(label="Translated Video", format='mp4'),
     title="Video Translation and Voice Cloning",
     description="Upload a video, choose a language to translate the audio, and download the processed video with translated audio."
 ).launch()