Update app.py

app.py

@@ -445,22 +445,22 @@ def process_video(video, url, type):
     os.remove("google_translate.srt")
     return srt, video, input_audio
 
-def video_edit(srt, input_video, input_audio= "audio.mp3"):
+def video_edit(srt, input_video, input_audio= 'audio.mp3'):
 
-    input_video_name = input_video.replace(".mp4", "")
+"""    input_video_name = input_video.replace(''.mp4', '')
     video = VideoFileClip(input_video)
     audio = AudioFileClip(input_audio)
     video = video.with_audio(audio)
     print(video)
-    output_video_file = input_video_name + '_subtitled' + ".mp4"
+    output_video_file = input_video_name + '_subtitled' + '.mp4'
     write_srt(srt)
-    subtitles = pysrt.open("edited_srt.srt", encoding="utf-8")
+    subtitles = pysrt.open('edited_srt.srt', encoding='utf-8')
     subtitle_clips = create_subtitle_clips(subtitles, video.size, 32, 'arial.ttf', 'white', False)
     final_video = CompositeVideoClip([video] + subtitle_clips)
-    final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac", logger=None)
-    os.remove("google_translate.srt")
-    print('final')
-    return output_video_file
+    final_video.write_videofile(output_video_file, codec='libx264', audio_codec='aac', logger=None)
+    os.remove('google_translate.srt')
+    print('final')"""
+    return input_video, 'edited_srt.srt'
 
     
 with gr.Blocks() as demo:
@@ -476,12 +476,13 @@ with gr.Blocks() as demo:
         btn.click(fn=process_video, inputs=[video_file_input, inp, drp], outputs=[out, video_path_output, audio_path_output])
     with gr.Row():
         vid_out = gr.Video()
+        srt_file = gr.File()
         btn2 = gr.Button("transcribe")
         gr.on(
             triggers=[btn2.click],
             fn=write_google,
             inputs=out,
-        ).then(video_edit, [out, video_path_output, audio_path_output], outputs=[vid_out])
+        ).then(video_edit, [out, video_path_output, audio_path_output], outputs=[vid_out, srt_file])
 
 
 demo.launch(debug=True)