Update app.py

app.py

@@ -2,14 +2,18 @@ import gradio as gr
 import os
 import tempfile
 import moviepy.editor as mp
+from moviepy.config import change_settings
 import assemblyai as aai
 from deep_translator import GoogleTranslator
 import pysrt
 from gtts import gTTS
 from pydub import AudioSegment
 
+# Fix ImageMagick security policy
+change_settings({"IMAGEMAGICK_BINARY": "/usr/bin/convert"})
+
 # Configuration
-aai.settings.api_key = os.getenv("ASSEMBLYAI_API_KEY")  # Set in HF Secrets
+aai.settings.api_key = os.getenv("ASSEMBLYAI_API_KEY")
 LANGUAGES = {
     "English": "en",
     "Spanish": "es", 
@@ -20,17 +24,15 @@ LANGUAGES = {
 }
 
 def extract_audio(video_path, output_dir):
-    """Extract audio from video file"""
     try:
         video = mp.VideoFileClip(video_path)
-        audio_path = os.path.join(output_dir, "extracted_audio.wav")
+        audio_path = os.path.join(output_dir, "audio.wav")
         video.audio.write_audiofile(audio_path)
         return audio_path
     except Exception as e:
         raise Exception(f"Audio extraction failed: {str(e)}")
 
 def generate_subtitles(audio_path, output_dir):
-    """Generate subtitles from audio file"""
     try:
         transcriber = aai.Transcriber()
         transcript = transcriber.transcribe(audio_path)
@@ -42,7 +44,6 @@ def generate_subtitles(audio_path, output_dir):
         raise Exception(f"Subtitle generation failed: {str(e)}")
 
 def translate_subtitles(srt_path, target_langs, output_dir):
-    """Translate subtitles to multiple languages"""
     try:
         subs = pysrt.open(srt_path)
         results = {}
@@ -63,32 +64,41 @@ def translate_subtitles(srt_path, target_langs, output_dir):
         raise Exception(f"Translation failed: {str(e)}")
 
 def merge_video_subtitles(video_path, srt_path, output_path):
-    """Merge video with subtitles"""
     try:
+        # Create a black background for text
         video = mp.VideoFileClip(video_path)
-        subs = pysrt.open(srt_path)
+        txt_clips = []
         
-        subtitle_clips = []
-        for sub in subs:
+        # Add subtitles as simple text overlays
+        for sub in pysrt.open(srt_path):
             txt_clip = mp.TextClip(
                 sub.text,
                 fontsize=24,
                 color='white',
                 font='Arial',
-                size=(video.w * 0.8, None)
-            ).set_start(sub.start.ordinal / 1000.0
-            ).set_duration((sub.end.ordinal - sub.start.ordinal) / 1000.0
-            ).set_pos(('center', 'bottom'))
-            subtitle_clips.append(txt_clip)
+                bg_color='black',
+                size=(video.w*0.9, None),
+                method='caption'  # Simpler rendering method
+            ).set_position(('center', 'bottom')
+             .set_duration((sub.end.ordinal - sub.start.ordinal)/1000.0)
+             .set_start(sub.start.ordinal/1000.0)
+            
+            txt_clips.append(txt_clip)
         
-        final_video = mp.CompositeVideoClip([video] + subtitle_clips)
-        final_video.write_videofile(output_path, codec='libx264', audio_codec='aac')
+        # Composite video with subtitles
+        final = mp.CompositeVideoClip([video] + txt_clips)
+        final.write_videofile(
+            output_path,
+            codec='libx264',
+            audio_codec='aac',
+            threads=4,
+            ffmpeg_params=['-crf', '23']
+        )
         return output_path
     except Exception as e:
         raise Exception(f"Video merging failed: {str(e)}")
 
 def process_video(video_file, source_lang, target_langs, progress=gr.Progress()):
-    """Main processing function"""
     with tempfile.TemporaryDirectory() as tmpdir:
         try:
             progress(0.1, "Extracting audio...")
@@ -117,13 +127,13 @@ def process_video(video_file, source_lang, target_langs, progress=gr.Progress())
         except Exception as e:
             raise gr.Error(f"Processing failed: {str(e)}")
 
-# Gradio Interface
-with gr.Blocks(title="Video Translator") as demo:
-    gr.Markdown("## 🎥 Video Translation System")
+# Create interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Video Translation System")
     
     with gr.Row():
         with gr.Column():
-            video_input = gr.File(label="Upload Video", file_types=["video"])
+            video_input = gr.Video(label="Upload Video")
             source_lang = gr.Dropdown(
                 label="Source Language",
                 choices=list(LANGUAGES.keys()),
@@ -132,13 +142,13 @@ with gr.Blocks(title="Video Translator") as demo:
             target_langs = gr.CheckboxGroup(
                 label="Target Languages",
                 choices=list(LANGUAGES.keys()),
-                value=["Spanish", "French"]
+                value=["Spanish"]
             )
             submit_btn = gr.Button("Translate")
         
         with gr.Column():
-            output_videos = gr.Files(label="Translated Videos")
-    
+            output_videos = gr.Video(label="Translated Video")
+
     submit_btn.click(
         process_video,
         inputs=[video_input, source_lang, target_langs],