Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr 
 import os
 import sys
@@ -29,12 +28,13 @@ def translate(input_video):
     result = model.transcribe(audio_file,**translate_options)
 
     output_dir = ''
-    
-    with open(os.path.join(output_dir, 'audio' + ".vtt"), "w") as vtt:
+    audio_path = audio_file.split(".")[0]
+
+    with open(os.path.join(output_dir, audio_path + ".vtt"), "w") as vtt:
       write_vtt(result["segments"], file=vtt)
 
-    subtitle     = "subtitles.vtt"
-    output_video = "subtitled.mp4"
+    subtitle = audio_path + ".vtt"
+    output_video = audio_path + "_subtitled.mp4"
 
     os.system(f"ffmpeg -i {input_video} -vf subtitles={subtitle} {output_video}")
 
@@ -60,5 +60,6 @@ with block:
 
         
         btn.click(translate, inputs=[inp_video], outputs=[op_video])
-
+        
+        
 block.launch(enable_queue = True)