Update app.py

app.py

@@ -16,18 +16,11 @@ def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_su
     output_video = f"/tmp/output-{input_video_name}.mp4"
     subtitle_track_title = os.path.splitext(os.path.basename(subtitle_file))[0]
 
-    if soft_subtitle:
         stream = ffmpeg.output(
             video_input_stream, subtitle_input_stream, output_video,
             **{"c": "copy", "c:s": "mov_text"},
             **{"metadata:s:s:0": f"language={subtitle_language}",
                "metadata:s:s:0": f"title={subtitle_track_title}"}
-        )
-    else:
-        stream = ffmpeg.output(
-            video_input_stream, output_video,
-            vf=f"subtitles={subtitle_file}"
-        )
 
     ffmpeg.run(stream, overwrite_output=True)
     return output_video
@@ -41,18 +34,42 @@ def video_demo(video, subtitle, subtitle_type, subtitle_language):
         return video
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Text to SRT Converter")
-    gr.Markdown("### ⚠️ For information, the process can be really long.")
+    # Header and information
+    gr.Markdown("<h1 style='text-align: center;'>Text to SRT Converter</h1>", unsafe_allow_html=True)
+    gr.Markdown("<h3 style='text-align: center; color: #FF5733;'>⚠️ Note: The processing can take some time depending on the video length and size.</h3>", unsafe_allow_html=True)
     
-    with gr.Group():
-        video_input = gr.Video(label="Video", interactive=True)
-        subtitle_input = gr.File(label="Subtitle", file_types=[".srt", ".vtt"])
-        subtitle_language_input = gr.Textbox(label="Subtitle Language (ISO 639-1 code, 'en' for English)")
-        submit_button = gr.Button("Submit")
+    # Inputs section
+    with gr.Row():
+        with gr.Column(scale=1):
+            video_input = gr.Video(label="Upload Video")
+        with gr.Column(scale=1):
+            subtitle_input = gr.File(label="Upload Subtitle File", file_types=[".srt", ".vtt"])
+        with gr.Column(scale=1):
+            subtitle_language_input = gr.Textbox(label="Subtitle Language (ISO 639-1, e.g., 'en')")
     
-    output_video = gr.Video(label="Processed Video")
+    # Submit button
+    with gr.Row():
+        submit_button = gr.Button("Process Video", elem_id="process_button")
+
+    # Output video
+    output_video = gr.Video(label="Processed Video", elem_id="output_video")
 
+    # Button click action
     submit_button.click(fn=video_demo, inputs=[video_input, subtitle_input, subtitle_language_input], outputs=output_video)
 
+    # Custom CSS to enhance visual appearance
+    demo.style(
+        '''
+        <style>
+            #process_button { background-color: #4CAF50; color: white; border: none; padding: 15px 32px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 8px; }
+            #output_video { border: 2px solid #4CAF50; border-radius: 8px; }
+            .gr-column { padding: 10px; }
+            .gr-row { justify-content: center; margin-top: 20px; }
+        </style>
+        ''', 
+        unsafe_allow_html=True
+    )
+
+
 if __name__ == "__main__":
     demo.launch()