Update app.py

app.py

@@ -28,21 +28,62 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
         #color_clips.append(myclip.with_position(text_position))
     return subtitle_clips
 
+import subprocess
+import os
+
 def video_edit(srt, input_video, color, font, font_size, input_audio):
-    print(input_video)
-    input_video_name = input_video.split(".mp4")[0]
-    video = VideoFileClip(input_video)
-    audio = AudioFileClip(input_audio)
-    video = video.with_audio(audio)
-    print(video)
-    output_video_file = input_video_name + '_subtitled' + ".mp4"
-    subtitles = pysrt.open(srt, encoding="utf-8")
-    subtitle_clips = create_subtitle_clips(subtitles, video.size, int(font_size), f'{font}.ttf', color, False)
-    final_video = CompositeVideoClip([video]+ subtitle_clips)
-    final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac", logger=None, preset = "faster", fps=24)
-    print('final')
-    video.close()
-    audio.close()
+    """
+    Burns subtitles into a video using FFmpeg and replaces the audio.
+    
+    Args:
+        srt (str): Path to the SRT subtitle file.
+        input_video (str): Path to the input video file.
+        color (str): Subtitle color in hex format (e.g., '#FFFF00' for yellow).
+        font (str): Font name for subtitles.
+        font_size (int): Font size for subtitles.
+        input_audio (str): Path to the audio file to replace the video's audio.
+    
+    Returns:
+        str: Path to the output video file with burned subtitles and replaced audio.
+    """
+    # Derive output file name from input video
+    #input_video_base = os.path.splitext(input_video)[0]
+    output_video_file = f"{input_video_base}_subtitled.mp4"
+    
+    # Get font directory (assume fonts are in the same directory as the SRT file)
+    fonts_dir = os.path.dirname(srt)
+    
+    # Convert color to FFmpeg’s PrimaryColour format (e.g., '#FFFF00' -> '&HFFFF00&')
+   ''' if not color.startswith('#') or len(color) != 7:
+        raise ValueError("Color must be a hex code like '#RRGGBB'")
+    ffmpeg_color = '&H' + color[1:] + '&''''
+    
+    # Build subtitle style string for FFmpeg
+    subtitle_style = f"FontName={font},FontSize={font_size},PrimaryColour={&HFFFF00&}"
+    
+    # Construct FFmpeg command
+    cmd = [
+        'ffmpeg',
+        '-i', input_video,              # Input video file
+       # '-i', input_audio,              # Input audio file
+        '-vf', f"subtitles={srt}:fontsdir={fonts_dir}:force_style='{subtitle_style}'",  # Burn subtitles
+        '-c:v', 'libx264',              # Video codec (H.264)
+        '-c:a', 'aac',                  # Audio codec (AAC, matching MoviePy’s default)
+        '-r', '24',                     # Frame rate (adjust as needed)
+        '-preset', 'faster',            # Encoding speed vs. compression trade-off
+        '-map', '0:v:0',                # Use video from first input (after subtitle filter)
+        '-map', '1:a:0',                # Use audio from second input
+        '-y',                           # Overwrite output file if it exists
+        output_video_file               # Output file
+    ]
+    
+    # Execute FFmpeg command
+    try:
+        subprocess.run(cmd, check=True, stderr=subprocess.PIPE, universal_newlines=True)
+    except subprocess.CalledProcessError as e:
+        raise RuntimeError(f"FFmpeg failed: {e.stderr}")
+    
+    print(f"Video processed successfully: {output_video_file}")
     return output_video_file
 
 with gr.Blocks() as demo: