Upload 2 files

Dockerfile

@@ -1,25 +1,27 @@
-FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
-
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3 python3-pip wget curl ca-certificates xz-utils \
-    && rm -rf /var/lib/apt/lists/*
-
-RUN wget -q https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-linux64-gpl-shared.tar.xz \
-        -O /tmp/ffmpeg.tar.xz \
-    && tar -xJf /tmp/ffmpeg.tar.xz --strip-components=1 -C /usr/local \
-        --wildcards '*/bin/ffmpeg' '*/bin/ffprobe' '*/lib/*.so*' \
-    && ldconfig \
-    && rm /tmp/ffmpeg.tar.xz \
-    && ffmpeg -version | head -3
-
-WORKDIR /app
-COPY app.py .
-
-RUN pip3 install --no-cache-dir gradio
-
-EXPOSE 7860
-ENV PYTHONUNBUFFERED=1
-
-CMD ["python3", "app.py"]
+FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3 python3-pip wget curl ca-certificates xz-utils \
+    && rm -rf /var/lib/apt/lists/*
+
+# nonfree-shared = inclui libfdk_aac
+RUN wget -q https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-linux64-nonfree-shared.tar.xz \
+        -O /tmp/ffmpeg.tar.xz \
+    && tar -xJf /tmp/ffmpeg.tar.xz --strip-components=1 -C /usr/local \
+        --wildcards '*/bin/ffmpeg' '*/bin/ffprobe' '*/lib/*.so*' \
+    && ldconfig \
+    && rm /tmp/ffmpeg.tar.xz \
+    && ffmpeg -version | head -3 \
+    && ffmpeg -codecs 2>/dev/null | grep fdk
+
+WORKDIR /app
+COPY app.py .
+
+RUN pip3 install --no-cache-dir gradio
+
+EXPOSE 7860
+ENV PYTHONUNBUFFERED=1
+
+CMD ["python3", "app.py"]

app.py

@@ -1,105 +1,163 @@
-import gradio as gr
-import subprocess
-import os
-import json
-
-def reencode_video(video_file, modo, resolucao):
-    if video_file is None:
-        return None, "❌ Nenhum vídeo enviado!"
-
-    input_path = "input.mp4"
-    with open(input_path, "wb") as f:
-        f.write(video_file)
-
-    output_path = "output_reencoded.mp4"
-    stats_path = "loudnorm_stats.json"
-
-    # PASSO 1: loudnorm two-pass
-    subprocess.run([
-        "ffmpeg", "-hide_banner", "-loglevel", "error", "-y",
-        "-i", input_path, "-af", "loudnorm=print_format=json",
-        "-f", "null", "-"
-    ], stderr=open(stats_path, "w"))
-
-    with open(stats_path, "r") as f:
-        stats = json.load(f)
-    m_I = stats["input_i"]
-    m_TP = stats["input_tp"]
-    m_LRA = stats["input_lra"]
-    m_thresh = stats["input_thresh"]
-    offset = stats["target_offset"]
-
-    # Filtro de vídeo
-    if resolucao == "Original":
-        vf = "mpdecimate=hi=1"
-    else:
-        w, h = resolucao.split("x")
-        vf = f"scale={w}:{h}:force_original_aspect_ratio=decrease,pad={w}:{h}:(ow-iw)/2:(oh-ih)/2,mpdecimate=hi=1"
-
-    cmd = ["ffmpeg", "-hide_banner", "-loglevel", "error", "-stats", "-y"]
-    if "GPU" in modo:
-        cmd += ["-hwaccel", "cuda", "-hwaccel_output_format", "cuda"]
-    cmd += ["-i", input_path, "-vf", vf, "-vsync", "vfr", "-r", "24"]
-
-    # Codec
-    if "x264" in modo:
-        if "GPU" in modo:
-            cmd += ["-c:v", "h264_nvenc", "-preset", "p7", "-tune", "hq", "-rc", "vbr", "-cq", "24", "-b:v", "0", "-profile:v", "high", "-pix_fmt", "yuv420p"]
-        else:
-            cmd += ["-c:v", "libx264", "-profile:v", "high", "-preset", "slow", "-crf", "24", "-pix_fmt", "yuv420p"]
-    else:  # x265
-        if "GPU" in modo:
-            cmd += ["-c:v", "hevc_nvenc", "-preset", "p7", "-tune", "uhq", "-rc", "vbr", "-cq", "24", "-b:v", "0", "-profile:v", "main", "-pix_fmt", "yuv420p"]
-        else:
-            cmd += ["-c:v", "libx265", "-preset", "slow", "-crf", "24", "-pix_fmt", "yuv420p", "-x265-params", "sao=0:rd=6:psy-rd=1.0:psy-rdoq=2.0:rskip=1"]
-
-    cmd += [
-        "-c:a", "libfdk_aac", "-profile:a", "aac_he_v2", "-b:a", "32k", "-ar", "22050", "-ac", "2",
-        "-af", f"loudnorm=I=-23:TP=-2:LRA=7:linear=true:measured_I={m_I}:measured_tp={m_TP}:measured_LRA={m_LRA}:measured_thresh={m_thresh}:offset={offset}",
-        "-movflags", "+faststart",
-        "-metadata:s:a:0", "language=por",
-        "-metadata:s:a:0", "comment=Re-encoded by Super Re-Encoder - https://t.me/virumaniaa",
-        output_path
-    ]
-
-    result = subprocess.run(cmd, capture_output=True, text=True)
-    if result.returncode != 0:
-        return None, f"❌ Erro no FFmpeg:\n{result.stderr[-800:]}"
-
-    orig_mb = os.path.getsize(input_path) / (1024*1024)
-    final_mb = os.path.getsize(output_path) / (1024*1024)
-    reducao = round((1 - final_mb / orig_mb) * 100, 1)
-
-    return output_path, f"✅ Concluído!\nOriginal: {orig_mb:.1f} MB\nFinal: {final_mb:.1f} MB\nRedução: {reducao}%"
-
-# Interface limpa
-with gr.Blocks(title="Super Re-Encoder") as demo:
-    gr.Markdown("# 🎥 **Super Re-Encoder**")
-    gr.Markdown("Suba o vídeo → escolha modo e resolução → re-encode automático.")
-
-    with gr.Row():
-        video = gr.Video(label="📤 Seu vídeo", sources=["upload"], height=300)
-        with gr.Column():
-            modo = gr.Radio(
-                choices=["x264 GPU (máx T4)", "x265 GPU (máx T4)", "x264 CPU only", "x265 CPU only"],
-                value="x264 GPU (máx T4)",
-                label="🔥 Modo"
-            )
-            res = gr.Dropdown(
-                choices=["Original", "480p (854x480)", "720p (1280x720)", "1080p (1920x1080)"],
-                value="Original",
-                label="📐 Resolução"
-            )
-
-    btn = gr.Button("🚀 RE-ENCODE AGORA", variant="primary", size="large")
-    out_video = gr.Video(label="📥 Vídeo final")
-    status = gr.Textbox(label="📋 Relatório", lines=5)
-
-    btn.click(reencode_video, inputs=[video, modo, res], outputs=[out_video, status])
-
-demo.queue(max_size=5)
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    theme=gr.themes.Soft()
-)
+import gradio as gr
+import subprocess
+import os
+import json
+import shutil
+
+def reencode_video(video_file, modo, resolucao, fps, normalizar_audio, remover_duplicados):
+    if video_file is None:
+        return None, "❌ Nenhum vídeo enviado!"
+
+    input_path = "input.mp4"
+    shutil.copy(video_file, input_path)
+
+    output_path = "output_reencoded.mp4"
+
+    # --- Filtro de vídeo ---
+    vf_parts = []
+
+    if resolucao != "Original":
+        w, h = resolucao.split("x")
+        vf_parts.append(f"scale={w}:{h}:force_original_aspect_ratio=decrease,pad={w}:{h}:(ow-iw)/2:(oh-ih)/2")
+
+    if remover_duplicados:
+        vf_parts.append("mpdecimate=hi=1")
+
+    vf = ",".join(vf_parts) if vf_parts else "null"
+
+    # --- Comando base ---
+    cmd = ["ffmpeg", "-hide_banner", "-loglevel", "error", "-stats", "-y"]
+    if "GPU" in modo:
+        cmd += ["-hwaccel", "cuda", "-hwaccel_output_format", "cuda"]
+    cmd += ["-i", input_path, "-vf", vf, "-vsync", "vfr", "-r", str(fps)]
+
+    # --- Codec de vídeo ---
+    if "x264" in modo:
+        if "GPU" in modo:
+            cmd += ["-c:v", "h264_nvenc", "-preset", "p7", "-tune", "hq",
+                    "-rc", "vbr", "-cq", "24", "-b:v", "0", "-profile:v", "high", "-pix_fmt", "yuv420p"]
+        else:
+            cmd += ["-c:v", "libx264", "-profile:v", "high", "-preset", "slow",
+                    "-crf", "24", "-pix_fmt", "yuv420p"]
+    else:  # x265
+        if "GPU" in modo:
+            cmd += ["-c:v", "hevc_nvenc", "-preset", "p7", "-tune", "uhq",
+                    "-rc", "vbr", "-cq", "24", "-b:v", "0", "-profile:v", "main", "-pix_fmt", "yuv420p"]
+        else:
+            cmd += ["-c:v", "libx265", "-preset", "slow", "-crf", "24", "-pix_fmt", "yuv420p",
+                    "-x265-params", "sao=0:rd=6:psy-rd=1.0:psy-rdoq=2.0:rskip=1"]
+
+    # --- Áudio: libfdk_aac HE-AACv2 32k stereo (padrão) ---
+    if normalizar_audio:
+        stats_path = "loudnorm_stats.json"
+        subprocess.run([
+            "ffmpeg", "-hide_banner", "-loglevel", "error", "-y",
+            "-i", input_path, "-af", "loudnorm=print_format=json",
+            "-f", "null", "-"
+        ], stderr=open(stats_path, "w"))
+
+        with open(stats_path, "r") as f:
+            raw = f.read()
+        json_start = raw.find("{")
+        stats = json.loads(raw[json_start:])
+        m_I      = stats["input_i"]
+        m_TP     = stats["input_tp"]
+        m_LRA    = stats["input_lra"]
+        m_thresh = stats["input_thresh"]
+        offset   = stats["target_offset"]
+
+        af = (f"loudnorm=I=-23:TP=-2:LRA=7:linear=true"
+              f":measured_I={m_I}:measured_tp={m_TP}"
+              f":measured_LRA={m_LRA}:measured_thresh={m_thresh}:offset={offset}")
+        cmd += [
+            "-c:a", "libfdk_aac", "-profile:a", "aac_he_v2",
+            "-b:a", "32k", "-ar", "22050", "-ac", "2",
+            "-af", af
+        ]
+    else:
+        cmd += [
+            "-c:a", "libfdk_aac", "-profile:a", "aac_he_v2",
+            "-b:a", "32k", "-ar", "22050", "-ac", "2"
+        ]
+
+    cmd += [
+        "-movflags", "+faststart",
+        "-metadata:s:a:0", "language=por",
+        "-metadata:s:a:0", "comment=Re-encoded by Super Re-Encoder",
+        output_path
+    ]
+
+    result = subprocess.run(cmd, capture_output=True, text=True)
+    if result.returncode != 0:
+        return None, f"❌ Erro no FFmpeg:\n{result.stderr[-1000:]}"
+
+    orig_mb  = os.path.getsize(input_path)  / (1024 * 1024)
+    final_mb = os.path.getsize(output_path) / (1024 * 1024)
+    reducao  = round((1 - final_mb / orig_mb) * 100, 1)
+
+    extras = []
+    if normalizar_audio:   extras.append("áudio normalizado")
+    if remover_duplicados: extras.append("frames duplicados removidos")
+    extras_str = " | ".join(extras) if extras else "nenhum extra"
+
+    return output_path, (
+        f"✅ Concluído!\n"
+        f"Original : {orig_mb:.1f} MB\n"
+        f"Final    : {final_mb:.1f} MB\n"
+        f"Redução  : {reducao}%\n"
+        f"FPS      : {fps}\n"
+        f"Extras   : {extras_str}"
+    )
+
+# --- Interface ---
+with gr.Blocks(title="Super Re-Encoder") as demo:
+    gr.Markdown("# 🎥 **Super Re-Encoder**")
+    gr.Markdown("Suba o vídeo, configure as opções e clique em Re-Encode.")
+
+    with gr.Row():
+        video = gr.Video(label="📤 Seu vídeo", sources=["upload"], height=300)
+
+        with gr.Column():
+            modo = gr.Radio(
+                choices=["x264 GPU (máx T4)", "x265 GPU (máx T4)", "x264 CPU only", "x265 CPU only"],
+                value="x264 CPU only",
+                label="🔥 Codec / Modo"
+            )
+            resolucao = gr.Dropdown(
+                choices=[
+                    "Original",
+                    "360p (640x360)",
+                    "480p (854x480)",
+                    "540p (960x540)",
+                    "720p (1280x720)",
+                    "1080p (1920x1080)",
+                    "1440p (2560x1440)",
+                    "4K (3840x2160)",
+                ],
+                value="Original",
+                label="📐 Resolução"
+            )
+            fps = gr.Dropdown(
+                choices=[15, 20, 24, 25, 30, 48, 60],
+                value=24,
+                label="🎞️ FPS alvo"
+            )
+
+    with gr.Row():
+        normalizar_audio   = gr.Checkbox(value=True, label="🔊 Normalizar áudio (loudnorm -23 LUFS)")
+        remover_duplicados = gr.Checkbox(value=True, label="🗑️ Remover frames duplicados (mpdecimate)")
+
+    btn = gr.Button("🚀 RE-ENCODE AGORA", variant="primary", size="large")
+
+    with gr.Row():
+        out_video = gr.Video(label="📥 Vídeo final")
+        status    = gr.Textbox(label="📋 Relatório", lines=7)
+
+    btn.click(
+        reencode_video,
+        inputs=[video, modo, resolucao, fps, normalizar_audio, remover_duplicados],
+        outputs=[out_video, status]
+    )
+
+demo.queue(max_size=5)
+demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft())