Update app.py

app.py

@@ -269,13 +269,8 @@ def translate_transcriptions(df, selected_translation_lang_2):
         
     print("translations done")
 
-    return df
+    print("Starting SRT-file creation")
 
-def create_srt_and_burn(df, video_in):
-    
-    print("Starting creation of video wit srt")
-    
-       
     with open('subtitles.srt','w', encoding="utf-8") as file:
         for i in range(len(df)):
             file.write(str(i+1))
@@ -284,32 +279,32 @@ def create_srt_and_burn(df, video_in):
            
     
             milliseconds = round(start * 1000.0)
-
+    
             hours = milliseconds // 3_600_000
             milliseconds -= hours * 3_600_000
-
+    
             minutes = milliseconds // 60_000
             milliseconds -= minutes * 60_000
-
+    
             seconds = milliseconds // 1_000
             milliseconds -= seconds * 1_000
-
+    
             file.write(f"{hours}:{minutes:02d}:{seconds:02d}.{milliseconds:03d}")
             
             stop = df.iloc[i]['end']
             
             
             milliseconds = round(stop * 1000.0)
-
+    
             hours = milliseconds // 3_600_000
             milliseconds -= hours * 3_600_000
-
+    
             minutes = milliseconds // 60_000
             milliseconds -= minutes * 60_000
-
+    
             seconds = milliseconds // 1_000
             milliseconds -= seconds * 1_000
-
+    
             
             file.write(' --> ')
             file.write(f"{hours}:{minutes:02d}:{seconds:02d}.{milliseconds:03d}")
@@ -318,7 +313,14 @@ def create_srt_and_burn(df, video_in):
             if int(i) != len(df)-1:
                 file.write('\n\n')
         
-    print("SRT DONE")        
+    print("SRT DONE") 
+
+    return df, 'subtitles.srt'
+
+def create_srt_and_burn(srt_file, video_in):
+    
+    print("Starting creation of video wit srt")
+    
     try:
         file1 = open('./subtitles.srt', 'r', encoding="utf-8")
         Lines = file1.readlines()
@@ -357,7 +359,7 @@ selected_whisper_model = gr.Dropdown(choices=whisper_models, type="value", value
 transcription_df = gr.DataFrame(value=df_init,label="Transcription dataframe", row_count=(0, "dynamic"), max_rows = 10, wrap=True, overflow_row_behaviour='paginate')
 transcription_and_translation_df = gr.DataFrame(value=df_init,label="Transcription and translation dataframe", max_rows = 10, wrap=True, overflow_row_behaviour='paginate')
 
-text_file = gr.File(
+srt_file = gr.File(
                 label="Download srt-file",
                 file_count="single",
                 type="file",
@@ -432,8 +434,13 @@ with demo:
             ##### ''')
             selected_translation_lang_2.render()
             translate_transcriptions_button = gr.Button("Step 3. Translate transcription")
-            translate_transcriptions_button.click(translate_transcriptions, [transcription_df, selected_translation_lang_2], transcription_and_translation_df)
+            translate_transcriptions_button.click(translate_transcriptions, [transcription_df, selected_translation_lang_2], transcription_and_translation_df, srt_file)
             transcription_and_translation_df.render()
+    
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown('''##### From here you can download the srt-file ''')
+            srt_file.render()
             
     with gr.Row():
         with gr.Column():
@@ -442,14 +449,10 @@ with demo:
             ##### ''')
             translate_and_make_srt_btn = gr.Button("Step 4. Create and burn srt to video")
             print(video_in)
-            translate_and_make_srt_btn.click(create_srt_and_burn, [transcription_and_translation_df,video_in], [
-                video_out, text_file])
+            translate_and_make_srt_btn.click(create_srt_and_burn, [srt_file,video_in], [
+                video_out])
             video_out.render()
 
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown('''##### From here you can download the srt-file ''')
-            text_file.render()
 
                 
 demo.launch()