updated multilingual app

app.py

@@ -141,12 +141,10 @@ remove_files(7)
 
 
 
-#from pydub import AudioSegment
-
 #Download the model
 model = whisper.load_model("small")
 
-
+    
 def video2mp3(video_file, output_ext="mp3"):
     filename, ext = os.path.splitext(video_file)
     subprocess.call(["ffmpeg", "-y", "-i", video_file, f"{filename}.{output_ext}"],
@@ -162,24 +160,20 @@ def translate(input_video):
     translate_options = dict(task="translate", **options)
     result = model.transcribe(audio_file, **translate_options)
 
-    output_dir = "content"  # Specify the desired output directory
-    os.makedirs(output_dir, exist_ok=True)  # Create the directory if it doesn't exist
-    audio_path = os.path.splitext(os.path.basename(audio_file))[0]
-    subtitle_path = os.path.join(output_dir, audio_path + ".vtt")
+    output_dir = '/content/'
+    audio_path = audio_file.split(".")[0]
 
-    with open(subtitle_path, "w") as vtt:
+    with open(os.path.join(output_dir, audio_path + ".vtt"), "w") as vtt:
         write_vtt(result["segments"], file=vtt)
 
-    output_video = os.path.join(output_dir, audio_path + "_subtitled.mp4")
+    subtitle = audio_path + ".vtt"
+    output_video = audio_path + "_subtitled.mp4"
 
-    subprocess.call(["ffmpeg", "-i", input_video, "-vf", f"subtitles={subtitle_path}:scodec=srt,force_style='[email protected],FontSize=30'", output_video],
-                    stdout=subprocess.DEVNULL,
-                    stderr=subprocess.STDOUT)
+    os.system(f"ffmpeg -i {input_video} -vf subtitles={subtitle} {output_video}")
 
     return output_video
 
 
-
 st.title("MultiLingual AI: Add Caption to Videos")
 
 uploaded_file = st.file_uploader("Upload your video", type=["mp4"])
@@ -199,7 +193,6 @@ if uploaded_file is not None:
         # Remove temporary files
         os.remove("temp_video.mp4")
 
-
 st.markdown(
     '''
     <style>