Update app.py

app.py

@@ -374,22 +374,36 @@ def process_video(url, type):
     srt_string = read_srt_file(subtitle_file)
     google_translate = translate_text(api_key, source_language, target_language, srt_string)
     write_google(google_translate)
+    srt = read_srt_file("google_translate.srt")
+    return srt, input_video
+
+def video_edit(srt, input_video, input_audio= "audio.mp3"):
+
+    input_video_name = input_video.replace(".mp4", "")
     video = VideoFileClip(input_video)
     audio = AudioFileClip(input_audio)
     video = video.with_audio(audio)
     print(video)
     subtitles = pysrt.open("google_translate.srt", encoding="utf-8")
     output_video_file = input_video_name + '_subtitled' + ".mp4"
-    subtitle_clips = create_subtitle_clips(subtitles, video.size, 24, 'arial.ttf', 'yellow', False)
+    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)
     print('final')
-    google = read_srt_file("google_translate.srt")
-    return output_video_file, google
-
-#def download_file(file_path):
- #   return gr.File.update(file_path)
 
-iface = gr.Interface(fn=process_video, inputs=["text" ,gr.Dropdown(["insta", "youtube"])], outputs=["file",gr.Textbox(interactive=True)])
-
-iface.launch(debug=True)
+    
+with gr.Blocks() as demo:
+    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    with gr.Row():
+        inp = gr.Textbox(placeholder="Enter URL or upload")
+        drp = gr.Dropdown(["insta", "youtube"])
+        btn = gr.Button("transcribe")
+        out = gr.Textbox(interactive=True)
+        video_path_output = gr.Textbox(label="Video Path", visible=False)
+        btn.click(fn=process_video, inputs=[inp, drp], outputs=[out, video_path_output])
+    with gr.Row():
+        vid_out = gr.Video()
+        btn2 = gr.Button("transcribe")
+        btn2.click(video_edit, [out, video_path_output], vid_out)
+
+demo.launch(debug=True)