Update app.py

app.py

@@ -61,24 +61,14 @@ if video_file:
 
             return ' '.join(segments)
 
-        # 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)
+        # 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
 
         # Transcribe audio using Whisper
         try:
@@ -87,40 +77,19 @@ if video_file:
 
             # Translate text to the target language
             translator = Translator(to_lang=LANGUAGES[target_language])
-            translated_text = translate_in_chunks(original_text, translator)
+            translated_text = translate_text(original_text, translator)
             st.write(f"Translated Text ({target_language}):", translated_text)
 
-            # 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")
-            final_audio.write_audiofile(translated_audio_path)
+            # 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)
 
             # Merge translated audio with the original video
             final_video_path = tempfile.mktemp(suffix=".mp4")
             original_video = VideoFileClip(temp_video_path)
-            final_video = original_video.set_audio(AudioFileClip(translated_audio_path))
+            final_audio = AudioFileClip(tts_audio_path)
+            final_video = original_video.set_audio(final_audio)
             final_video.write_videofile(final_video_path, codec='libx264', audio_codec='aac')
 
             # Display success message and provide download link
@@ -135,9 +104,8 @@ if video_file:
             st.error(f"Error during transcription/translation: {e}")
 
         # Clean up temporary files
-        for clip in tts_clips:
-            os.remove(clip.filename)
         os.remove(temp_video_path)
         os.remove(audio_path)
-        os.remove(translated_audio_path)
-        os.remove(final_video_path)
+        os.remove(tts_audio_path)
+        if 'final_video_path' in locals():  # Check if final_video_path exists
+            os.remove(final_video_path)