Update app.py

app.py

@@ -1,11 +1,12 @@
 import streamlit as st
-from moviepy.editor import VideoFileClip, AudioFileClip, concatenate_audioclips
+from moviepy.editor import VideoFileClip, AudioFileClip
 import whisper
 from translate import Translator
 from gtts import gTTS
 import tempfile
 import os
 import numpy as np
+import time
 
 # Initialize Whisper model
 try:
@@ -61,14 +62,24 @@ if video_file:
 
             return ' '.join(segments)
 
-        # Translate text function with debug info
-        def translate_text(original_text, translator):
-            translated_text = translator.translate(original_text)
-            # Debugging: Check translation results
-            st.write(f"Translated Text Debug: {translated_text}")
-            if translated_text.strip() == original_text.strip():
-                st.warning("The translated text is the same as the original. Check if the target language is appropriate.")
-            return translated_text
+        # Function to translate text in chunks
+        def translate_in_chunks(text, translator, max_length=500):
+            words = text.split()
+            chunks = []
+            current_chunk = ""
+
+            for word in words:
+                if len(current_chunk) + len(word) + 1 <= max_length:
+                    current_chunk += " " + word if current_chunk else word
+                else:
+                    chunks.append(current_chunk)
+                    current_chunk = word
+
+            if current_chunk:
+                chunks.append(current_chunk)
+
+            translated_chunks = [translator.translate(chunk) for chunk in chunks]
+            return ' '.join(translated_chunks)
 
         # Transcribe audio using Whisper
         try:
@@ -77,19 +88,20 @@ if video_file:
 
             # Translate text to the target language
             translator = Translator(to_lang=LANGUAGES[target_language])
-            translated_text = translate_text(original_text, translator)
+            translated_text = translate_in_chunks(original_text, translator)
             st.write(f"Translated Text ({target_language}):", translated_text)
 
             # Convert translated text to speech
-            tts_audio_path = tempfile.mktemp(suffix=".mp3")
             tts = gTTS(text=translated_text, lang=LANGUAGES[target_language])
-            tts.save(tts_audio_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)
-            final_audio = AudioFileClip(tts_audio_path)
-            final_video = original_video.set_audio(final_audio)
+            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
@@ -102,10 +114,13 @@ if video_file:
 
         except Exception as e:
             st.error(f"Error during transcription/translation: {e}")
+            translated_audio_path = None  # Ensure this variable is defined
+            final_video_path = None  # Ensure this variable is defined
 
         # Clean up temporary files
         os.remove(temp_video_path)
         os.remove(audio_path)
-        os.remove(tts_audio_path)
-        if 'final_video_path' in locals():  # Check if final_video_path exists
-            os.remove(final_video_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)