adjust

adjust.py

@@ -101,7 +101,15 @@ def get_speakers_previous(to_name):
     return get_current()
 
 
-def start_adjust(input_file, to_name, start_time, end_time, progress=gr.Progress()):
+def start_adjust_subtitle(input_file, start_time, end_time, progress=gr.Progress()):
+    # Adding subtitle to video
+    progress(0.8, desc=ui_lang["progress_add_subtitle"])
+    add_subtitle_to_video(
+        input_file, base_subtitle_adjusted_file, video_subtitle_file, start_time, end_time)
+    return video_subtitle_file
+
+
+def start_adjust(input_file, to_name, progress=gr.Progress()):
     change_name(to_name)
 
     # Replacing texts
@@ -113,18 +121,12 @@ def start_adjust(input_file, to_name, start_time, end_time, progress=gr.Progress
     utils.write_transcribe_subtitle_file(
         input_file, modified_transcribe, modified_subtitle, True)
 
-    # Adding subtitle to video
-    progress(0.8, desc=ui_lang["progress_add_subtitle"])
-    add_subtitle_to_video(
-        input_file, base_subtitle_adjusted_file, video_subtitle_file, start_time, end_time)
-
     # Return video file link, transcribe string, transcribe.txt, subtitle.txt
     transcribe_txt_list, _ = utils.read_transcribe_subtitle_file(
         input_file, True)
     print(line for line in transcribe_txt_list)
     transcribe_txt = "\n".join(transcribe_txt_list)
     return [
-        video_subtitle_file,
         transcribe_txt,
         [transcribe_adjusted_file, subtitle_adjusted_file]
     ]

app.py

@@ -38,9 +38,8 @@ with gr.Blocks() as demo:
             output_video.render()
             output_file.render()
             output_file.change(prepare_output, inputs=input_video, outputs=[
-                adjust_speaker, adjust_audio, prev_button, next_button, adjust_button])
-            output_file.change(prepare_video_subtitle, inputs=[
-                               input_video, start_time, end_time], outputs=output_video)
+                adjust_speaker, adjust_audio, prev_button, next_button, adjust_button]).then(prepare_video_subtitle, inputs=[
+                    input_video, start_time, end_time], outputs=output_video)
 
         with gr.Column():
             output_transcribe.render()
@@ -59,8 +58,8 @@ with gr.Blocks() as demo:
                           adjust_speaker, adjust_audio])
 
     adjust_button.render()
-    adjust_button.click(start_adjust, inputs=[input_video, adjust_speaker, start_time, end_time,], outputs=[
-                        output_video, output_transcribe, output_file])
+    adjust_button.click(start_adjust, inputs=[input_video, adjust_speaker], outputs=[output_transcribe, output_file]).then(
+        start_adjust_subtitle, inputs=[input_video, start_time, end_time], outputs=output_video)
 
     with gr.Accordion("Copyright"):
         gr.Markdown("Created with OpenAI Whisper and Huggingface")