Update app.py

app.py

@@ -269,9 +269,14 @@ def translate_transcriptions(df, selected_translation_lang_2):
         
     print("translations done")
 
-    print("Starting SRT-file creation")
+    return df
 
-    with open('subtitles.srt','w', encoding="utf-8") as file:
+def create_srt_and_burn(df, video_in):
+    
+    print("Starting creation of video wit srt")
+    
+       
+    with open('testi.srt','w', encoding="utf-8") as file:
         for i in range(len(df)):
             file.write(str(i+1))
             file.write('\n')
@@ -279,32 +284,32 @@ def translate_transcriptions(df, selected_translation_lang_2):
            
     
             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}")
@@ -313,16 +318,9 @@ def translate_transcriptions(df, selected_translation_lang_2):
             if int(i) != len(df)-1:
                 file.write('\n\n')
         
-    print("SRT DONE") 
-
-    return df, 'subtitles.srt'
-
-def create_srt_and_burn(srt_file, video_in):
-    
-    print("Starting creation of video wit srt")
-    
+    print("SRT DONE")        
     try:
-        file1 = open('./subtitles.srt', 'r', encoding="utf-8")
+        file1 = open('./testi.srt', 'r', encoding="utf-8")
         Lines = file1.readlines()
         
         count = 0
@@ -336,10 +334,10 @@ def create_srt_and_burn(srt_file, video_in):
         
         video_out = video_in.replace('.mp4', '_out.mp4')
         print(video_out)
-        command = 'ffmpeg -i "{}" -y -vf subtitles=./subtitles.srt "{}"'.format(video_in, video_out)
+        command = 'ffmpeg -i "{}" -y -vf subtitles=./testi.srt "{}"'.format(video_in, video_out)
         print(command)
         os.system(command)
-        return video_out, './subtitles.srt'
+        return video_out
     except Exception as e:
         print(e)
         return video_out
@@ -359,13 +357,6 @@ 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')
 
-srt_file = gr.File(
-                label="Download srt-file",
-                file_count="single",
-                type="file",
-                interactive=False,
-            )
-
 demo = gr.Blocks(css='''
 #cut_btn, #reset_btn { align-self:stretch; }
 #\\31 3 { max-width: 540px; }
@@ -434,13 +425,8 @@ 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, srt_file)
+            translate_transcriptions_button.click(translate_transcriptions, [transcription_df, selected_translation_lang_2], transcription_and_translation_df)
             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():
@@ -449,10 +435,9 @@ 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, [srt_file,video_in], [
+            translate_and_make_srt_btn.click(create_srt_and_burn, [transcription_and_translation_df,video_in], [
                 video_out])
             video_out.render()
 
-
                 
 demo.launch()