Update app.py

app.py

@@ -87,48 +87,60 @@ def transcribe(audio_file_path, model_size="base"):
     return "\n".join(transcription_with_timestamps)
 
 # Fonction pour ajouter des sous-titres à une vidéo
-def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_subtitle):
+def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_subtitle=False):
     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]
-
+    output_video = f"/tmp/{input_video_name}_subtitled.mp4"
     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}"}
-        )
+        stream = ffmpeg.output(video_input_stream, subtitle_input_stream, output_video, **{"c": "copy", "c:s": "mov_text"})
     else:
-        stream = ffmpeg.output(
-            video_input_stream, output_video,
-            vf=f"subtitles={subtitle_file}"
-        )
-
+        stream = ffmpeg.output(video_input_stream, output_video, vf=f"subtitles={subtitle_file}")
     ffmpeg.run(stream, overwrite_output=True)
     return output_video
 
-# Définition des interfaces Gradio
-def transcribe_and_translate_video(video_file, source_language_code, target_language_code):
+# Fonction de transcription pour Gradio avec ajout de sous-titres à la vidéo
+def transcribe_video(video_file):
+    transcription = transcribe(video_file.name, "tiny")
+    srt_path = text_to_srt(transcription)
+    # Ajouter les sous-titres à la vidéo
+    output_video_path = add_subtitle_to_video(video_file.name, srt_path, subtitle_language="eng", soft_subtitle=False)
+    return output_video_path, srt_path
+
+# Fonction de traduction pour Gradio qui utilise maintenant add_subtitle_to_video
+def translate_video(video_file, source_language_code, target_language_code):
     transcription = transcribe(video_file.name, "tiny")
     srt_path = text_to_srt(transcription)
     translated_srt_path = translate_srt(srt_path, source_language_code, target_language_code)
-    output_video = add_subtitle_to_video(video_file.name, translated_srt_path, target_language_code, False)
-    return output_video, translated_srt_path
-
-# Inputs et outputs Gradio
-video_input = gr.inputs.Video(label="Video File")
-source_language_dropdown = gr.inputs.Dropdown(choices=language_options, label="Source Language")
-target_language_dropdown = gr.inputs.Dropdown(choices=language_options, label="Target Language")
-
-transcribe_translate_interface = gr.Interface(
-    fn=transcribe_and_translate_video,
-    inputs=[video_input, source_language_dropdown, target_language_dropdown],
-    outputs=[gr.outputs.Video(label="Video with Translated Subtitles"), gr.outputs.File(label="Translated Subtitles (.srt)")],
-    title="Video Transcribe & Translate",
-    description="Transcribe and translate the subtitles of your video into another language."
+    # Ajoutez les sous-titres traduits à la vidéo
+    output_video_path = add_subtitle_to_video(video_file.name, translated_srt_path, target_language_code, soft_subtitle=False)
+    return output_video_path, translated_srt_path
+
+# Interface Gradio
+file_input = gr.Video(label="Video File")
+source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language")
+target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language")
+
+transcription_interface = gr.Interface(
+    fn=transcribe_video,
+    inputs=file_input,
+    outputs=[gr.Video(label="Video with Subtitles"), gr.File(label="Transcription File (.srt)")],
+    title="Video Transcription",
+    description="Automatically transcribe the audio from your video into subtitles and add them to the video."
 )
 
-# Lancement de l'application Gradio
-transcribe_translate_interface.launch()
+translation_interface = gr.Interface(
+    fn=translate_video,
+    inputs=[file_input, source_language_dropdown, target_language_dropdown],
+    outputs=[gr.Video(label="Video with Translated Subtitles"), gr.File(label="Translated Subtitles (.srt)")],
+    title="Video Translation",
+    description="Translate the subtitles of your video into another language and add them to the video."
+)
+
+tabs = gr.Tabs([
+    gr.Tab(transcription_interface, label="Transcribe"),
+    gr.Tab(translation_interface, label="Translate")
+])
+
+# Lancement de l'application
+tabs.launch()