Update app.py

app.py

@@ -259,7 +259,7 @@ def video_translation(video_path, target_language, speaker_model, hf_token, deep
 
   return video_with_dubbing
 
-def translate_video(_, video_path, __, youtube_link, ___, target_language, speaker_model):
+def translate_video(video_path, youtube_link, target_language, speaker_model):
     try:
       if not video_path and not youtube_link:
         gr.Warning("You should either upload video or input a YouTube link")
@@ -269,24 +269,43 @@ def translate_video(_, video_path, __, youtube_link, ___, target_language, speak
       dubbed_video = video_translation(video_path, target_language, speaker_model, HF_TOKEN, DEEPL_TOKEN)
     except Exception as e:
       print(f"An error occurred: {e}")
+      raise e
     return gr.components.Video(dubbed_video)
 
-
-inputs = [
-    gr.Markdown("## Currently supported languages are: English, Polish, Ukrainian and Russian"),
-    gr.Video(label="Upload a video file"),
-    gr.Markdown("**OR**"),
-    gr.Textbox(label="Paste YouTube link"),
-    gr.Markdown("---"),
-    gr.Dropdown(["en", "pl", "uk", "ru"], value="pl", label="Select translation target language"),
-    gr.Dropdown(["(Recommended) XTTS_V2", "VITs (will be default for Ukrainian)"], value="(Recommended) XTTS_V2", label="Select text-to-speech generation model")
-]
-
-outputs = gr.Video(label="Translated video")
-
-gr.Interface(fn=translate_video, 
-             inputs=inputs, 
-             outputs=outputs, 
-             title="🌐AI Video Translation",
-             theme=gr.themes.Base()
-             ).launch(show_error=True, debug=True, share=True)
+def clear_inputs():
+    return None, "", None, None
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("<h1 style='text-align: center;'>🌐AI Video Translation</h2>")
+    gr.Markdown("<h3 style='text-align: center;'>Currently supported languages are: English, Polish, Ukrainian, and Russian</h3>")
+    
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("<h2 style='text-align: center;'>Inputs</h3>")
+            video = gr.Video(label="Upload a video file")
+            gr.Markdown("<h3 style='text-align: center;'>OR</h3>")
+            youtube_link = gr.Textbox(label="Paste YouTube link")
+            gr.Markdown("---")
+            target_language = gr.Dropdown(["en", "pl", "uk", "ru"], value="pl", label="Select translation target language")
+            speaker_model = gr.Dropdown(["(Recommended) XTTS_V2", "VITs (will be default for Ukrainian)"], value="(Recommended) XTTS_V2", label="Select text-to-speech generation model")
+            with gr.Row():
+                clear_btn = gr.Button("Clear inputs")
+                translate_btn = gr.Button("Translate")    
+            
+        with gr.Column():
+            gr.Markdown("<h2 style='text-align: center;'>Translated Video</h3>")
+            output_video = gr.Video(label="Translated video")
+
+    translate_btn.click(
+        fn=translate_video,
+        inputs=[video, youtube_link, target_language, speaker_model],
+        outputs=output_video
+    )
+
+    clear_btn.click(
+        fn=clear_inputs,
+        inputs=[],
+        outputs=[video, youtube_link, target_language, speaker_model]
+    )
+
+demo.launch(show_error=True, debug=True, share=True)