Update app.py

app.py

@@ -1,35 +1,25 @@
 import gradio as gr
 import os
 from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
-import ffmpeg  # Make sure to install ffmpeg-python
+import ffmpeg  # Ensure ffmpeg-python is installed
 
 def read_subtitle_file(subtitle_path):
     with open(subtitle_path, 'r', encoding='utf-8') as file:
         subtitle_content = file.read()
     return os.path.basename(subtitle_path), subtitle_content
 
-def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_subtitle):
-    subtitle_file_path = subtitle_file.replace(" ", "\\ ")
+def add_hard_subtitle_to_video(input_video, subtitle_file, subtitle_language):
     video_input_stream = ffmpeg.input(input_video)
-    subtitle_input_stream = ffmpeg.input(subtitle_file)
-    input_video_name = os.path.splitext(os.path.basename(input_video))[0]
-    output_video = f"/tmp/output-{input_video_name}.mp4"
-    subtitle_track_title = os.path.splitext(os.path.basename(subtitle_file))[0]
-
-        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}"}
-        )
+    output_video = f"/tmp/output-{os.path.splitext(os.path.basename(input_video))[0]}.mp4"
 
+    # Hard subtitle process
+    stream = ffmpeg.output(video_input_stream, output_video, vf=f"subtitles={subtitle_file}")
     ffmpeg.run(stream, overwrite_output=True)
     return output_video
 
-def video_demo(video, subtitle, subtitle_type, subtitle_language):
+def video_demo(video, subtitle, subtitle_language):
     if subtitle is not None:
-        soft_subtitle = "Hard"
-        processed_video_path = add_subtitle_to_video(video, subtitle, subtitle_language, soft_subtitle)
+        processed_video_path = add_hard_subtitle_to_video(video, subtitle, subtitle_language)
         return processed_video_path
     else:
         return video
@@ -71,6 +61,5 @@ with gr.Blocks() as demo:
         unsafe_allow_html=True
     )
 
-
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()