Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from moviepy.editor import VideoFileClip
+from moviepy.editor import VideoFileClip, AudioFileClip
 import whisper
 from translate import Translator
 from gtts import gTTS
@@ -92,18 +92,33 @@ if video_file:
 
             # Convert translated text to speech
             tts = gTTS(text=translated_text, lang=LANGUAGES[target_language])
-            audio_output_path = tempfile.mktemp(suffix=".mp3")
-            tts.save(audio_output_path)
+            translated_audio_path = tempfile.mktemp(suffix=".mp3")
+            tts.save(translated_audio_path)
+
+            # Merge translated audio with the original video
+            final_video_path = tempfile.mktemp(suffix=".mp4")
+            original_video = VideoFileClip(temp_video_path)
+            translated_audio = AudioFileClip(translated_audio_path)
+
+            final_video = original_video.set_audio(translated_audio)
+            final_video.write_videofile(final_video_path, codec='libx264', audio_codec='aac')
+
+            # Display success message and provide download link
+            st.success("Translation successful! Download your translated video below:")
+            st.video(final_video_path)
+
+            # Provide download link
+            with open(final_video_path, "rb") as f:
+                st.download_button("Download Translated Video", f, file_name="translated_video.mp4")
 
-            # Display translated text and audio
-            st.success("Translation successful!")
-            st.audio(audio_output_path, format="audio/mp3")
         except Exception as e:
             st.error(f"Error during transcription/translation: {e}")
-            audio_output_path = None  # Ensure this variable is defined
+            translated_audio_path = None  # Ensure this variable is defined
 
         # Clean up temporary files
         os.remove(temp_video_path)
         os.remove(audio_path)
-        if audio_output_path:  # Only remove if it was created
-            os.remove(audio_output_path)
+        if translated_audio_path:  # Only remove if it was created
+            os.remove(translated_audio_path)
+        if final_video_path:  # Only remove if it was created
+            os.remove(final_video_path)