Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from moviepy.editor import VideoFileClip, AudioFileClip
+from moviepy.editor import VideoFileClip, AudioFileClip, concatenate_audioclips
 import whisper
 from translate import Translator
 from gtts import gTTS
@@ -90,17 +90,37 @@ if video_file:
             translated_text = translate_in_chunks(original_text, translator)
             st.write(f"Translated Text ({target_language}):", translated_text)
 
-            # Convert translated text to speech
-            tts = gTTS(text=translated_text, lang=LANGUAGES[target_language])
+            # Convert translated text to speech in chunks
+            tts_clips = []
+            words = translated_text.split()
+            chunk = ""
+            max_length = 200  # Adjust as needed
+
+            for word in words:
+                if len(chunk) + len(word) + 1 > max_length:
+                    tts = gTTS(text=chunk, lang=LANGUAGES[target_language])
+                    tts_audio_path = tempfile.mktemp(suffix=".mp3")
+                    tts.save(tts_audio_path)
+                    tts_clips.append(AudioFileClip(tts_audio_path))
+                    chunk = word
+                else:
+                    chunk += " " + word if chunk else word
+
+            if chunk:  # Process last chunk
+                tts = gTTS(text=chunk, lang=LANGUAGES[target_language])
+                tts_audio_path = tempfile.mktemp(suffix=".mp3")
+                tts.save(tts_audio_path)
+                tts_clips.append(AudioFileClip(tts_audio_path))
+
+            # Concatenate all TTS audio chunks
+            final_audio = concatenate_audioclips(tts_clips)
             translated_audio_path = tempfile.mktemp(suffix=".mp3")
-            tts.save(translated_audio_path)
+            final_audio.write_audiofile(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 = original_video.set_audio(AudioFileClip(translated_audio_path))
             final_video.write_videofile(final_video_path, codec='libx264', audio_codec='aac')
 
             # Display success message and provide download link
@@ -113,12 +133,11 @@ if video_file:
 
         except Exception as e:
             st.error(f"Error during transcription/translation: {e}")
-            translated_audio_path = None  # Ensure this variable is defined
 
         # Clean up temporary files
+        for clip in tts_clips:
+            os.remove(clip.filename)
         os.remove(temp_video_path)
         os.remove(audio_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)
+        os.remove(translated_audio_path)
+        os.remove(final_video_path)