Upload app.py

app.py

@@ -37,22 +37,15 @@ def contar_frames(path):
 #  Logo — preparação, posição e preview ao vivo
 # ══════════════════════════════════════════════════════════════════
 
-def preparar_logo(logo_path, vid_w, tamanho_pct, opacidade_pct):
+def preparar_logo(logo_path, opacidade_pct):
     """
-    Abre o logo, redimensiona para tamanho_pct% da largura do vídeo
-    e escala o canal alpha por opacidade_pct (100 = totalmente opaco,
-    0 = invisível). Mesmo tratamento simples que o radar-virumania usa.
-    A eliminação do halo verde é feita na composição FFmpeg (em RGBA).
+    SOMENTE aplica a opacidade (escala o canal alpha) via Pillow.
+    O redimensionamento é feito pelo próprio FFmpeg com scale filter —
+    exatamente como o LegendadorBrasileiroWhisperX faz. Assim evitamos
+    re-saves desnecessários da PNG e preservamos as bordas originais.
     """
     img = Image.open(logo_path).convert("RGBA")
 
-    # Redimensiona proporcionalmente à largura alvo
-    alvo_w = max(1, int(vid_w * tamanho_pct / 100))
-    ratio  = alvo_w / img.width
-    alvo_h = max(1, int(img.height * ratio))
-    img    = img.resize((alvo_w, alvo_h), Image.LANCZOS)
-
-    # Escala o canal alpha pela opacidade escolhida
     if opacidade_pct < 100:
         r, g, b, a = img.split()
         fator = opacidade_pct / 100.0
@@ -61,7 +54,7 @@ def preparar_logo(logo_path, vid_w, tamanho_pct, opacidade_pct):
 
     temp_path = "/tmp/logo_overlay.png"
     img.save(temp_path, "PNG")
-    return temp_path, alvo_w, alvo_h
+    return temp_path
 
 
 def calcular_posicao_logo(posicao, margem=20, offset_x=0, offset_y=0):
@@ -251,11 +244,11 @@ def reencode_video(
 
     # ── Prepara a logo (Pillow) ─────────────────────────────────
     logo_tmp = None
+    logo_width_px = 0
     if tem_logo:
         try:
-            logo_tmp, _, _ = preparar_logo(
-                logo_file, vid_w, logo_tamanho, logo_opacidade
-            )
+            logo_tmp      = preparar_logo(logo_file, logo_opacidade)
+            logo_width_px = max(1, int(vid_w * logo_tamanho / 100))
         except Exception as e:
             yield None, f"❌ Erro ao processar logo: {e}"
             return
@@ -288,29 +281,25 @@ def reencode_video(
     if tem_logo:
         cmd += ["-i", logo_tmp]
 
-    # ── filter_complex (overlay em RGBA → yuv420p no fim) ───────
+    # ── filter_complex (IDÊNTICO ao LegendadorBrasileiroWhisperX) ───
     if tem_logo:
         overlay_pos = calcular_posicao_logo(
             logo_posicao, logo_margem, logo_offset_x, logo_offset_y
         )
-        # Correção do halo verde:
-        # BASE e LOGO são convertidos para RGBA antes do overlay, então
-        # a composição acontece em RGB puro (sem YUV 4:2:0 introduzir
-        # sangramento de chroma nas bordas). Só no fim convertemos para
-        # yuv420p para o encoder. Sem premultiply — straight alpha puro.
+        # Regra de ouro: NÃO forçar nenhum format= na cadeia. O overlay
+        # do FFmpeg já lida com alpha PNG corretamente. Conversões
+        # explícitas yuv420p↔rgba causam mismatch de matriz de cor
+        # (BT.601/BT.709) e introduzem halo verde nas bordas.
         if vf_parts:
             fc = (
-                f"[0:v]{','.join(vf_parts)},format=rgba[base];"
-                f"[1:v]format=rgba[logo];"
-                f"[base][logo]overlay={overlay_pos}[composed];"
-                f"[composed]format=yuv420p[vout]"
+                f"[0:v]{','.join(vf_parts)}[base];"
+                f"[1:v]scale={logo_width_px}:-1[logo];"
+                f"[base][logo]overlay={overlay_pos}[vout]"
             )
         else:
             fc = (
-                f"[0:v]format=rgba[base];"
-                f"[1:v]format=rgba[logo];"
-                f"[base][logo]overlay={overlay_pos}[composed];"
-                f"[composed]format=yuv420p[vout]"
+                f"[1:v]scale={logo_width_px}:-1[logo];"
+                f"[0:v][logo]overlay={overlay_pos}[vout]"
             )
         cmd += ["-filter_complex", fc, "-map", "[vout]"]
         if has_audio: