Upload 2 files

Dockerfile

@@ -8,12 +8,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libfdk-aac-dev libx264-dev libx265-dev libnuma-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# nv-codec-headers (necessário para h264_nvenc / hevc_nvenc)
+# nv-codec-headers (necessário para h264_nvenc / hevc_nvenc / cuvid)
 RUN git clone --depth 1 https://git.videolan.org/git/ffmpeg/nv-codec-headers.git /tmp/nv-codec-headers \
     && make -C /tmp/nv-codec-headers install \
     && rm -rf /tmp/nv-codec-headers
 
-# Compila ffmpeg 7.1.1 com libfdk_aac + NVENC
+# Compila ffmpeg 7.1.1 com libfdk_aac + NVENC + scale_cuda
 RUN wget -q -O /tmp/ffmpeg.tar.bz2 https://ffmpeg.org/releases/ffmpeg-7.1.1.tar.bz2 \
     && tar xf /tmp/ffmpeg.tar.bz2 -C /tmp \
     && rm /tmp/ffmpeg.tar.bz2 \
@@ -25,14 +25,16 @@ RUN wget -q -O /tmp/ffmpeg.tar.bz2 https://ffmpeg.org/releases/ffmpeg-7.1.1.tar.
         --disable-everything \
         --enable-encoder=libx264,libx265,libfdk_aac,aac,h264_nvenc,hevc_nvenc \
         --enable-decoder=h264,hevc,aac,mp3,ac3,eac3,opus,vorbis,flac,pcm_s16le,pcm_s16be \
+        --enable-hwaccel=h264_nvdec,hevc_nvdec \
         --enable-demuxer=mov,mp4,matroska,avi,flv,mpegts,wav,mp3,aac,flac \
         --enable-muxer=mp4,matroska,null \
-        --enable-filter=loudnorm,volume,aresample,scale,pad,mpdecimate,fps,null \
+        --enable-filter=loudnorm,volume,aresample,scale,scale_cuda,hwupload_cuda,hwdownload,pad,mpdecimate,fps,null \
         --enable-protocol=file,pipe \
         --enable-libx264 \
         --enable-libx265 \
         --enable-libfdk-aac \
         --enable-nvenc \
+        --enable-nvdec \
         --enable-cuda \
         --enable-cuvid \
         --extra-cflags="-O2 -pipe -I/usr/local/cuda/include" \
@@ -63,6 +65,7 @@ COPY --from=builder /usr/local/bin/ffprobe /usr/local/bin/ffprobe
 
 RUN ffmpeg -version | head -3 \
     && ffmpeg -encoders 2>/dev/null | grep -E "fdk|nvenc" \
+    && ffmpeg -filters 2>/dev/null | grep -E "scale_cuda|hwupload" \
     && echo "✅ ffmpeg OK"
 
 WORKDIR /app

app.py

@@ -56,6 +56,11 @@ def reencode_video(video_file, modo, resolucao, fps, crf_valor, normalizar_audio
     use_gpu  = quer_gpu and gpu_disponivel()
     gpu_fallback = quer_gpu and not use_gpu
 
+    # Pipeline 100% GPU só é possível sem mpdecimate.
+    # O mpdecimate não tem equivalente CUDA no FFmpeg, então quando está
+    # ativo o decode e os filtros são forçados a rodar na CPU.
+    full_gpu_pipeline = use_gpu and not remover_duplicados
+
     # ── Filtro de vídeo ─────────────────────────────────────────
     vf_parts = []
 
@@ -64,26 +69,40 @@ def reencode_video(video_file, modo, resolucao, fps, crf_valor, normalizar_audio
         tw, th = m.group(1), m.group(2)
         if is_vertical:
             tw, th = th, tw
-        vf_parts.append(
-            f"scale={tw}:{th}:force_original_aspect_ratio=increase,"
-            f"crop={tw}:{th}"
-        )
-
-    vf_parts.append(f"fps={fps}")
+        if full_gpu_pipeline:
+            # scale_cuda mantém o frame na VRAM — zero transferência CPU↔GPU
+            vf_parts.append(f"scale_cuda=w={tw}:h={th}:interp_algo=lanczos")
+        else:
+            vf_parts.append(
+                f"scale={tw}:{th}:force_original_aspect_ratio=increase,"
+                f"crop={tw}:{th}"
+            )
 
-    if remover_duplicados:
-        vf_parts.append("mpdecimate=hi=1")
+    if not full_gpu_pipeline:
+        # No pipeline de CPU, fps e mpdecimate ficam no filtro de vídeo
+        vf_parts.append(f"fps={fps}")
+        if remover_duplicados:
+            vf_parts.append("mpdecimate=hi=1")
 
-    vf = ",".join(vf_parts) if vf_parts else "null"
+    vf = ",".join(vf_parts) if vf_parts else None
 
     # ── Monta comando base ──────────────────────────────────────
-    # NOTA: Removido o -hwaccel cuda. Como usamos filtros de software (-vf),
-    # deixar o FFmpeg decodificar na CPU e enviar direto pro NVENC é muito
-    # mais rápido do que ficar transferindo frames da GPU pra CPU e vice-versa.
     cmd = ["ffmpeg", "-hide_banner", "-loglevel", "error", "-progress", "pipe:1", "-y"]
-    cmd += ["-i", input_path, "-vf", vf]
 
-    if remover_duplicados:
+    if full_gpu_pipeline:
+        # Decodifica direto na GPU → frames ficam na VRAM o tempo todo
+        # Fluxo: GPU decode → scale_cuda (VRAM) → NVENC encode
+        cmd += ["-hwaccel", "cuda", "-hwaccel_output_format", "cuda"]
+
+    cmd += ["-i", input_path]
+
+    if vf:
+        cmd += ["-vf", vf]
+
+    if full_gpu_pipeline:
+        # FPS definido via parâmetro de saída — sem precisar de filtro de software
+        cmd += ["-r", str(fps)]
+    elif remover_duplicados:
         cmd += ["-fps_mode", "vfr"]
 
     # ── Codec de vídeo + CRF/CQ ─────────────────────────────────
@@ -192,12 +211,16 @@ def reencode_video(video_file, modo, resolucao, fps, crf_valor, normalizar_audio
     reducao  = round((1 - final_mb / orig_mb) * 100, 1)
 
     extras = []
+    if full_gpu_pipeline:
+        extras.append("pipeline GPU completo (decode+scale+encode na VRAM)")
+    elif use_gpu:
+        extras.append("encode NVENC / decode+scale na CPU")
     if normalizar_audio and has_audio:
         extras.append("áudio normalizado")
     elif not has_audio:
         extras.append("⚠️ sem faixa de áudio")
     if remover_duplicados:
-        extras.append("frames duplicados removidos (força uso de CPU)")
+        extras.append("mpdecimate ativo → decode forçado na CPU")
     if gpu_fallback:
         extras.append("⚠️ GPU indisponível → usou CPU")
 
@@ -208,7 +231,7 @@ def reencode_video(video_file, modo, resolucao, fps, crf_valor, normalizar_audio
         f"Redução  : {reducao}%\n"
         f"FPS      : {fps}\n"
         f"CRF/CQ   : {crf_valor}\n"
-        f"Codec    : {'NVENC' if use_gpu else 'CPU'}\n"
+        f"Codec    : {'NVENC (pipeline GPU completo)' if full_gpu_pipeline else 'NVENC (encode GPU)' if use_gpu else 'CPU'}\n"
         f"Extras   : {' | '.join(extras) if extras else 'nenhum'}"
     )
 
@@ -256,7 +279,7 @@ with gr.Blocks(title="Super Re-Encoder") as demo:
         normalizar_audio   = gr.Checkbox(value=True,
                                          label="🔊 Normalizar áudio (loudnorm -23 LUFS)")
         remover_duplicados = gr.Checkbox(value=True,
-                                         label="🗑️ Remover frames duplicados (mpdecimate)")
+                                         label="🗑️ Remover frames duplicados (mpdecimate — desative para pipeline GPU completo)")
 
     btn = gr.Button("🚀 RE-ENCODE AGORA", variant="primary", size="large")
 
@@ -271,4 +294,4 @@ with gr.Blocks(title="Super Re-Encoder") as demo:
     )
 
 demo.queue(max_size=5)
-demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft())
+demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft())