Update app.py

app.py

@@ -16,18 +16,26 @@ import ffmpeg
 import nltk
 nltk.download('averaged_perceptron_tagger_eng')
 
-def process_video(video_file, youtube_url, language_choice):
+def process_upload(video_file, language_choice):
     if language_choice == None:
         return None, "Language not selected."
-    elif video_file == None and youtube_url == "":
-        return None, "Upload either a video or a valid youtube URL."        
-    elif video_file != None and youtube_url != "":
-        return None, "Videos contradict. Delete either the uploaded video or youtube URL."
-    elif video_file == None and youtube_url != "":
+    elif video_file == None:
+        return None, "Video not uploaded."
+    else:
+        return process_video(video_file, language_choice)
+
+def process_youtube(youtube_url, language_choice):
+    if language_choice == None:
+        return None, "Language not selected."
+    elif youtube_url == None:
+        return None, "YouTube URL not entered."
+    else:
         yt = YouTube(youtube_url)
         yt.streams.filter(progressive=True, file_extension='mp4').first().download(filename="original.mp4")
         video_file = "original.mp4"
-    
+        return process_video(video_file, language_choice)
+
+def process_video(video_file, language_choice):
     # Initialize paths and devices
     ckpt_converter = 'checkpoints_v2/converter'
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -266,11 +274,10 @@ def process_video(video_file, youtube_url, language_choice):
 language_choices = ts.get_languages("google")["en"]
 language_choices.remove("auto")
 
-gr.Interface(
-    fn=process_video,
+uploaded_translator = gr.Interface(
+    fn=process_upload,
     inputs=[
         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=[
@@ -279,4 +286,20 @@ gr.Interface(
     ],
     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()
+)
+
+youtube_translator = gr.Interface(
+    fn=process_youtube,
+    inputs=[
+        gr.Textbox(label="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", format='mp4'),
+        gr.Textbox(show_label=False)
+    ],
+    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."
+)
+
+gr.TabbedInterface([uploaded_translator, youtube_translator], ["Upload video from device", "YouTube URL"]).launch()