Update app.py

app.py

@@ -28,62 +28,21 @@ 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):
-    """
-    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}_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:] + '& 
-    ffmpeg_color='&HFFFF00&'
-    # Build subtitle style string for FFmpeg
-    subtitle_style = f"FontName={font},FontSize={font_size},PrimaryColour={ffmpeg_color}"
-    
-    # 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}")
+    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()
     return output_video_file
 
 with gr.Blocks() as demo: