import json
import os
import shutil
import subprocess

import gradio as gr


def probe_video(video_path):
    if not video_path:
        return "Please upload a video.", None

    if shutil.which("ffprobe") is None:
        return (
            "ffprobe not found. Add `ffmpeg` to packages.txt at the repo root.",
            None,
        )

    try:
        result = subprocess.run(
            [
                "ffprobe",
                "-v", "quiet",
                "-print_format", "json",
                "-show_format",
                "-show_streams",
                video_path,
            ],
            capture_output=True,
            text=True,
            check=True,
        )
    except subprocess.CalledProcessError as e:
        return f"ffprobe failed:\n{e.stderr}", None

    info = json.loads(result.stdout)
    fmt = info.get("format", {}) or {}
    streams = info.get("streams", []) or []
    v = next((s for s in streams if s.get("codec_type") == "video"), {})
    a = next((s for s in streams if s.get("codec_type") == "audio"), {})

    summary = {
        "filename": os.path.basename(fmt.get("filename", "")),
        "format": fmt.get("format_long_name") or fmt.get("format_name"),
        "duration_sec": fmt.get("duration"),
        "size_bytes": fmt.get("size"),
        "overall_bitrate_bps": fmt.get("bit_rate"),
        "video": {
            "codec": v.get("codec_name"),
            "profile": v.get("profile"),
            "width": v.get("width"),
            "height": v.get("height"),
            "pix_fmt": v.get("pix_fmt"),
            "frame_rate": v.get("r_frame_rate"),
            "bitrate_bps": v.get("bit_rate"),
        },
        "audio": {
            "codec": a.get("codec_name"),
            "sample_rate": a.get("sample_rate"),
            "channels": a.get("channels"),
            "bitrate_bps": a.get("bit_rate"),
        },
    }
    return json.dumps(summary, indent=2, ensure_ascii=False), video_path


with gr.Blocks(title="OneVision Encoder Codec View") as demo:
    gr.Markdown(
        "# OneVision Encoder Codec View\n"
        "Upload a video to inspect its container / codec metadata via `ffprobe`."
    )
    with gr.Row():
        with gr.Column():
            video_in = gr.Video(label="Input video", sources=["upload"])
            run_btn = gr.Button("Probe", variant="primary")
        with gr.Column():
            video_out = gr.Video(label="Preview")
            info_out = gr.Code(label="Metadata (JSON)", language="json")
    run_btn.click(probe_video, inputs=video_in, outputs=[info_out, video_out])


if __name__ == "__main__":
    demo.launch()