import subprocess
import os
import sys
import textwrap
import random

# Validate input arguments
# Usage: python make_video.py image.jpg audio.mp3 arial.ttf "Hello World" output.mp4 '' ''
if len(sys.argv) < 6:
    print("Usage: python make_video.py <image> <audio> <font.ttf> <text> <output.mp4> [logo.png] [watermark text]")
    print("Optional: pass '' for logo or watermark if not needed.")
    sys.exit(1)

image = sys.argv[1]
audio = sys.argv[2]
font = sys.argv[3]
raw_text = sys.argv[4]
output = os.path.abspath(sys.argv[5])

logo = sys.argv[6] if len(sys.argv) > 6 and sys.argv[6] != "''" else None
watermark_text = sys.argv[7] if len(sys.argv) > 7 and sys.argv[7] != "''" else None

# Helper to get audio duration
def get_duration(file):
    cmd = [
        "ffprobe", "-v", "error",
        "-show_entries", "format=duration",
        "-of", "default=noprint_wrappers=1:nokey=1", file
    ]
    try:
        out = subprocess.check_output(cmd).decode().strip()
        return float(out)
    except subprocess.CalledProcessError as e:
        print("Error getting audio duration:")
        print(e.output.decode())
        sys.exit(1)

# Helper to wrap multiline quote text
def split_lines(text, width=22):
    wrapped = textwrap.wrap(text, width=width)
    padded = [f"    {line}    " for line in wrapped]
    escaped = [line.replace(":", "\\:").replace(",", "\\,") for line in padded]
    return escaped

# Step 1: Trim audio if longer than 60 seconds
original_audio = audio
duration = get_duration(audio)
if duration > 60:
    trimmed_audio = "trimmed_audio.mp3"
    print(f"Trimming audio to 60s: {trimmed_audio}")
    subprocess.run(["ffmpeg", "-y", "-i", audio, "-t", "60", "-c", "copy", trimmed_audio])
    audio = trimmed_audio
    duration = 60  # force duration to 60
else:
    trimmed_audio = None  # no temp file used

# Step 2: Prepare filters
lines = split_lines(raw_text)
framecount = int(duration * 30)
z_expr = f"1.0+0.3*sin(PI*on/{framecount})"

fontsize = 50
spacing = 80
last_label = "v0"
inputs = ["-loop", "1", "-i", image, "-i", audio]
if logo:
    inputs += ["-i", logo]

filter_complex = (
    f"[0:v]scale=720:1280,format=rgba,"
    f"drawbox=x=0:y=0:w=iw:h=ih:color=black@0.4:t=fill,"
    f"zoompan=z='{z_expr}':d=1:x='iw/2-(iw/zoom/2)':y='ih/2-(ih/zoom/2)',"
    f"scale=720:1280[{last_label}]"
)

# Add logo overlay
if logo:
    filter_complex += (
        f";[2:v]scale=80:80,format=rgba,colorchannelmixer=aa=0.4[logo];"
        f"[{last_label}][logo]overlay=W-w-40:20[v1]"
    )
    last_label = "v1"

# Add watermark text
if watermark_text:
    rand_x = random.randint(10, 640)
    rand_y = random.randint(10, 1200)
    safe_wm = watermark_text.replace(":", "\\:").replace("'", "\\'")
    filter_complex += (
        f";[{last_label}]drawtext=fontfile='{font}':"
        f"text='{safe_wm}':x={rand_x}:y={rand_y}:fontsize=36:"
        f"fontcolor=white@0.2:box=1:boxcolor=black@0.2[v2]"
    )
    last_label = "v2"

# Add main quote text
for i, line in enumerate(lines):
    y_pos = f"(h-text_h)/2+{(i - len(lines)//2) * spacing}"
    safe_line = line.replace(":", "\\:").replace("'", "\\'")
    draw = (
        f"drawtext=fontfile='{font}':text='{safe_line}':"
        f"fontcolor=yellow:fontsize={fontsize}:"
        f"borderw=4:bordercolor=black@0.7:"
        f"x=(w-text_w)/2:y={y_pos}"
    )
    filter_complex += f";[{last_label}]{draw}[v{i}]"
    last_label = f"v{i}"

final_map = last_label

# Final FFmpeg command
cmd = ["ffmpeg", "-y"] + inputs + [
    "-filter_complex", filter_complex,
    "-map", f"[{final_map}]",
    "-map", "1:a",
    "-t", "60",  # Enforce max length
    "-shortest",
    "-c:v", "libx264", "-preset", "veryfast", "-crf", "23",
    "-c:a", "aac", "-b:a", "128k",
    output
]

print("\nGenerating video with:")
print(f"- Output: {output}")
if logo: print("- Logo overlay")
if watermark_text: print("- Watermark text")
print("- Max duration: 60 seconds")

result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)

# Check result
if result.returncode != 0:
    print("FFmpeg failed:\n", result.stderr)
    sys.exit(1)
elif not os.path.exists(output):
    print("Output file not found.")
    sys.exit(1)
else:
    size = os.path.getsize(output)
    if size > 1000:
        print(f" Video created: {output} ({size / 1024:.1f} KB)")
    else:
        print(" Output file is too small or empty.")

# Cleanup temp audio file
if trimmed_audio and os.path.exists(trimmed_audio):
    os.remove(trimmed_audio)