Update app.py

app.py

@@ -99,48 +99,33 @@ def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_su
     ffmpeg.run(stream, overwrite_output=True)
     return output_video
 
-# 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)
-    # 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."
-)
-
-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")
-])
+# Initialisation de Gradio Blocks
+with gr.Blocks() as blocks_app:
+    with gr.Row():
+        video_file = gr.Video(label="Upload Video")
+        source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language", value="en")
+        target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language", value="fr")
+        transcribe_button = gr.Button("Transcribe Video")
+        translate_button = gr.Button("Translate Subtitles")
+    
+    output_video = gr.Video(label="Processed Video")
+    output_srt = gr.File(label="Subtitles File (.srt)")
+
+    def transcribe_and_add_subtitles(video_file):
+        transcription = transcribe(video_file.name, "tiny")
+        srt_path = text_to_srt(transcription)
+        output_video_path = add_subtitle_to_video(video_file.name, srt_path, subtitle_language="eng", soft_subtitle=False)
+        return output_video_path, srt_path
+    
+    def translate_subtitles_and_add_to_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_path = add_subtitle_to_video(video_file.name, translated_srt_path, target_language_code, soft_subtitle=False)
+        return output_video_path, translated_srt_path
+    
+    transcribe_button.click(transcribe_and_add_subtitles, inputs=video_file, outputs=[output_video, output_srt])
+    translate_button.click(translate_subtitles_and_add_to_video, inputs=[video_file, source_language_dropdown, target_language_dropdown], outputs=[output_video, output_srt])
 
 # Lancement de l'application
-tabs.launch()
+blocks_app.launch()