Update app.py

app.py

@@ -9,7 +9,7 @@ import pysrt
 from gtts import gTTS
 from pydub import AudioSegment
 
-# Fix ImageMagick security policy
+# Configure ImageMagick policy
 change_settings({"IMAGEMAGICK_BINARY": "/usr/bin/convert"})
 
 # Configuration
@@ -65,34 +65,33 @@ def translate_subtitles(srt_path, target_langs, output_dir):
 
 def merge_video_subtitles(video_path, srt_path, output_path):
     try:
-        # Create a black background for text
         video = mp.VideoFileClip(video_path)
         txt_clips = []
         
-        # Add subtitles as simple text overlays
         for sub in pysrt.open(srt_path):
-            txt_clip = mp.TextClip(
+            txt_clip = (mp.TextClip(
                 sub.text,
                 fontsize=24,
                 color='white',
                 font='Arial',
                 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)
+                method='caption'
+            )
+            .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)
         
-        # 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']
+            ffmpeg_params=['-crf', '23'],
+            logger=None
         )
         return output_path
     except Exception as e:
@@ -122,12 +121,11 @@ def process_video(video_file, source_lang, target_langs, progress=gr.Progress())
                 output_files.append(output_path)
             
             progress(1.0, "Done!")
-            return output_files
+            return output_files[0]  # Return first video only for demo
             
         except Exception as e:
             raise gr.Error(f"Processing failed: {str(e)}")
 
-# Create interface
 with gr.Blocks() as demo:
     gr.Markdown("# Video Translation System")
     
@@ -147,12 +145,12 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Translate")
         
         with gr.Column():
-            output_videos = gr.Video(label="Translated Video")
+            output_video = gr.Video(label="Translated Video")
 
     submit_btn.click(
         process_video,
         inputs=[video_input, source_lang, target_langs],
-        outputs=output_videos
+        outputs=output_video
     )
 
 if __name__ == "__main__":