Ffmpeg added

app.py

@@ -28,22 +28,102 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
         #color_clips.append(myclip.with_position(text_position))
     return subtitle_clips
 
-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()
-    return output_video_file
+import subprocess
+import os
+
+# Mapping of color names to FFmpeg-compatible hex codes
+COLOR_MAP = {
+    'white': '#FFFFFF',
+    'yellow': '#FFFF00',
+    'red': '#FF0000',
+    'green': '#00FF00',
+    'blue': '#0000FF',
+    'black': '#000000'
+}
+
+def hex_to_ffmpeg_color(hex_code):
+    """
+    Convert standard hex color (#RRGGBB) to FFmpeg's PrimaryColour format (&HBBGGRR&).
+    
+    Args:
+        hex_code (str): Standard hex color code (e.g., '#FFFFFF')
+    Returns:
+        str: FFmpeg color code (e.g., '&HFFFFFF&')
+    """
+    hex_code = hex_code.lstrip('#')
+    r, g, b = hex_code[0:2], hex_code[2:4], hex_code[4:6]
+    return f'&H{b}{g}{r}&'
+
+def video_edit(srt, input_video, font_color, font_type, font_size, input_audio=None):
+    """
+    Burns subtitles into a video using FFmpeg with customizable styling.
+    
+    Args:
+        srt (str): Path to SRT subtitle file
+        input_video (dict or str): Input video path or dict with 'video' key
+        font_color (str): Color name (e.g., 'white', 'yellow')
+        font_type (str): Font name for subtitles
+        font_size (int): Font size for subtitles
+        input_audio (str, optional): Path to audio file (if provided)
+    
+    Returns:
+        str: Path to output video file
+    """
+    # Handle input_video as dict or string
+    if isinstance(input_video, dict):
+        video_path = input_video.get('video', '')
+    else:
+        video_path = input_video
+
+    # Validate input
+    if not video_path or not os.path.exists(video_path):
+        raise ValueError("Invalid or missing video file")
+    if not os.path.exists(srt):
+        raise ValueError("Subtitle file not found")
+
+    # Generate output filename
+    input_base = os.path.splitext(video_path)[0]
+    output_video = f"{input_base}_subtitled.mp4"
+
+    # Convert color name to FFmpeg format
+    hex_color = COLOR_MAP.get(font_color.lower())
+    if not hex_color:
+        raise ValueError(f"Unsupported color: {font_color}. Supported colors: {', '.join(COLOR_MAP.keys())}")
+    ffmpeg_color = hex_to_ffmpeg_color(hex_color)
+
+    # Build subtitle style
+    subtitle_style = f"FontName={font_type},FontSize={font_size},PrimaryColour={ffmpeg_color}"
+    fonts_dir = os.path.dirname(srt)
+
+    # Construct FFmpeg command
+    cmd = [
+        'ffmpeg',
+        '-i', video_path,  # Input video
+    ]
+
+    # Add audio input if provided (though we'll still keep original audio)
+    if input_audio and os.path.exists(input_audio):
+        cmd.extend(['-i', input_audio])
+
+    cmd.extend([
+        '-vf', f"subtitles={srt}:fontsdir={fonts_dir}:force_style='{subtitle_style}'",  # Burn subtitles
+        '-c:v', 'libx264',  # Video codec
+        '-c:a', 'copy',     # Always copy original audio
+        '-r', '24',         # Frame rate
+        '-preset', 'ultrafast',# Encoding preset
+        '-y',               # Overwrite output
+        output_video
+    ])
+
+    # 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}")
+    return output_video
+
 
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the progress and final output.")