Update app.py

app.py

@@ -1,14 +1,14 @@
 import streamlit as st
-from moviepy.editor import VideoFileClip, AudioFileClip, TextClip, CompositeVideoClip, concatenate_audioclips
+from moviepy.editor import VideoFileClip, AudioFileClip, TextClip, CompositeVideoClip
 import whisper
 from translate import Translator
 from gtts import gTTS
 import tempfile
 import os
 import numpy as np
-from datetime import timedelta
 import shutil
 from pathlib import Path
+import time
 
 # Set page configuration
 st.set_page_config(
@@ -86,7 +86,7 @@ class VideoProcessor:
                 # Extract audio to temporary file
                 audio_path = self.create_temp_path(".wav")
                 video.audio.write_audiofile(audio_path, fps=16000, verbose=False, logger=None)
-
+                
                 # Check if audio file is not empty
                 if os.path.getsize(audio_path) == 0:
                     raise ValueError("Extracted audio file is empty")
@@ -128,11 +128,12 @@ class VideoProcessor:
         return translated_segments
 
     def generate_tamil_audio(self, text):
-        """Generate Tamil audio using gTTS"""
+        """Generate Tamil audio using gTTS with rate limiting"""
         try:
             audio_path = self.create_temp_path(".mp3")
             tts = gTTS(text=text, lang='ta', slow=False)
             tts.save(audio_path)
+            time.sleep(1)  # Adding delay to avoid rate limit issues
             return audio_path
         except Exception as e:
             raise Exception(f"Audio generation error: {str(e)}")
@@ -266,27 +267,33 @@ def main():
     with col2:
         generate_subtitles = st.checkbox("Generate Subtitles", value=True)
         
-    # Ensure subtitle size and color are always defined
-    subtitle_size = 24  # Default subtitle size
-    subtitle_color = 'white'  # Default subtitle color
-    
     if generate_subtitles:
-        subtitle_size = st.slider("Subtitle Size", 10, 40, 24)
-        subtitle_color = st.color_picker("Subtitle Color", "#FFFFFF")
-
+        col3, col4 = st.columns(2)
+        with col3:
+            subtitle_size = st.slider("Subtitle Size", 16, 32, 24)
+        with col4:
+            subtitle_color = st.color_picker("Subtitle Color", "#FFFFFF")
+    
+    # Process video
     if st.button("Process Video"):
-        try:
-            output_video_path = process_video(
-                video_file.read(),
-                voice_type,
-                generate_subtitles=generate_subtitles,
-                subtitle_size=subtitle_size,
-                subtitle_color=subtitle_color
-            )
-            st.success("Video processing complete!")
-            st.video(output_video_path)
-        except Exception as e:
-            st.error(f"Error: {e}")
-
+        with st.spinner("Processing video..."):
+            try:
+                output_video_path = process_video(
+                    video_file.read(),
+                    voice_type,
+                    generate_subtitles,
+                    subtitle_size,
+                    subtitle_color
+                )
+                
+                st.video(output_video_path)
+                st.success("Video processed successfully!")
+                
+                with open(output_video_path, "rb") as f:
+                    st.download_button("Download Processed Video", f, file_name="processed_video.mp4")
+                    
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
+    
 if __name__ == "__main__":
     main()