Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import os
 import tempfile
 import moviepy.editor as mp
-from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 import assemblyai as aai
 from deep_translator import GoogleTranslator
 import pysrt
@@ -20,9 +19,13 @@ LANGUAGES = {
     "Hindi": "hi"
 }
 
+# Add this at the beginning to disable ImageMagick usage
+import os
+os.environ["IMAGEMAGICK_BINARY"] = ""
+
 def extract_audio(video_path, output_dir):
     try:
-        video = VideoFileClip(video_path)
+        video = mp.VideoFileClip(video_path)
         audio_path = os.path.join(output_dir, "audio.wav")
         video.audio.write_audiofile(audio_path)
         return audio_path
@@ -60,45 +63,88 @@ def translate_subtitles(srt_path, target_langs, output_dir):
     except Exception as e:
         raise Exception(f"Translation failed: {str(e)}")
 
-def merge_video_subtitles(video_path, srt_path, output_path):
-    try:
-        video = VideoFileClip(video_path)
-        subs = pysrt.open(srt_path)
+def create_subtitle_clips(subtitles, videosize, fontsize=24):
+    """Create subtitle clips based on subtitles file"""
+    subtitle_clips = []
+    
+    for subtitle in subtitles:
+        start_time = subtitle.start.ordinal / 1000.0
+        end_time = subtitle.end.ordinal / 1000.0
+        duration = end_time - start_time
         
-        subtitles = []
-        for sub in subs:
-            start_time = sub.start.ordinal / 1000.0
-            end_time = sub.end.ordinal / 1000.0
-            duration = end_time - start_time
-            
-            text_clip = TextClip(
-                sub.text,
-                fontsize=24,
-                color='white',
-                stroke_color='black',
-                stroke_width=1,
+        # Create a solid color clip with the dimensions of the video
+        # and overlay the text on it
+        import numpy as np
+        from moviepy.editor import ColorClip, TextClip, CompositeVideoClip
+        
+        txt = subtitle.text.replace('\n', ' ')
+        
+        # Create text overlay
+        try:
+            text_clip = mp.TextClip(
+                txt, 
                 font='Arial',
-                size=(video.w*0.9, None),
-                method='label'  # Changed from 'caption' to 'label'
+                fontsize=fontsize,
+                color='white',
+                bg_color='black',
+                size=(videosize[0] * 0.9, None),
+                method='pango'  # Try to use pango instead of ImageMagick
             )
-            
-            text_clip = text_clip.set_position(('center', 'bottom'))
-            text_clip = text_clip.set_start(start_time)
-            text_clip = text_clip.set_duration(duration)
-            subtitles.append(text_clip)
+        except:
+            # If pango fails, try without method specification
+            try:
+                text_clip = mp.TextClip(
+                    txt, 
+                    font='Arial',
+                    fontsize=fontsize,
+                    color='white', 
+                    bg_color='black',
+                    size=(videosize[0] * 0.9, None)
+                )
+            except:
+                # Last resort - create a simple black background with text overlay
+                w, h = videosize[0], 60
+                txtclip = ColorClip((w, h), col=(0, 0, 0))
+                txt_col = ColorClip((w, h), col=(0, 0, 0))
+                text_clip = txtclip
+
+        # Position the text
+        text_clip = text_clip.set_position(('center', 'bottom'))
+        text_clip = text_clip.set_start(start_time)
+        text_clip = text_clip.set_duration(duration)
+        
+        subtitle_clips.append(text_clip)
+    
+    return subtitle_clips
+
+def merge_video_subtitles(video_path, srt_path, output_path):
+    try:
+        # Load video and subtitles
+        video = mp.VideoFileClip(video_path)
+        subtitles = pysrt.open(srt_path)
+        
+        # Create subtitle clips
+        subtitle_clips = create_subtitle_clips(subtitles, video.size)
+        
+        # Overlay subtitles on video
+        final_video = mp.CompositeVideoClip([video] + subtitle_clips)
         
-        final = CompositeVideoClip([video] + subtitles)
-        final.write_videofile(
+        # Write output file
+        final_video.write_videofile(
             output_path,
             codec='libx264',
-            audio_codec='aac',
+            fps=video.fps,
+            preset='ultrafast',  # Use ultrafast preset to speed up encoding
             threads=4,
-            ffmpeg_params=['-crf', '23'],
-            logger=None
+            audio_codec='aac',
+            logger=None  # Disable verbose logging
         )
+        
         return output_path
     except Exception as e:
-        raise Exception(f"Video merging failed: {str(e)}")
+        import traceback
+        error_details = traceback.format_exc()
+        raise Exception(f"Video merging failed: {str(e)}\nDetails: {error_details}")
 
 def process_video(video_file, source_lang, target_langs, progress=gr.Progress()):
     with tempfile.TemporaryDirectory() as tmpdir: