Update app.py

app.py

@@ -305,96 +305,112 @@ if st.button("Gerar Vídeo(s)"):
                         "-of", "default=noprint_wrappers=1:nokey=1", video_final_raw
                     ], stdout=subprocess.PIPE)
                     dur_video_real = float(dur_proc.stdout.decode().strip())
-                    progresso.progress(35 + n * 5)
-
-                    # Filtros no vídeo principal (zoom, espelho, borda, etc.)
-                    filtros_main = ["scale=720:1280:force_original_aspect_ratio=decrease"]
-                    if zoom != 1.0:
-                        filtros_main.append(f"scale=iw*{zoom}:ih*{zoom}")
-                    filtros_main.append(f"setpts=PTS/{velocidade_cortes}")
-                    if ativar_espelhar:
-                        filtros_main.append("hflip")
-                    if remover_borda and tamanho_borda > 0:
-                        filtros_main.append(f"crop=in_w-{tamanho_borda*2}:in_h-{tamanho_borda*2}")
-                    if ativar_filtro_cor:
-                        filtros_main.append("eq=contrast=1.1:saturation=1.2")
-                    if ativar_borda_personalizada:
-                        cor_ffmpeg = f"0x{cor_borda.lstrip('#')}FF"
-                        anim_map = {
-                            "Nenhuma": "",
-                            "Borda Pulsante": "enable='lt(mod(t,1),0.5)'",
-                            "Cor Animada": "enable='lt(mod(t,1),0.5)'",
-                            "Neon": "enable='lt(mod(t,0.5),0.25)'",
-                            "Ondulada": "enable='gt(sin(t*3.14),0)'"
-                        }
-                        drawbox = f"drawbox=x=0:y=0:w=iw:h=ih:color={cor_ffmpeg}:t=5"
-                        if anim_map[animacao_borda]:
-                            drawbox += f":{anim_map[animacao_borda]}"
-                        filtros_main.append(drawbox)
-
-                    # Fundo com filtros
-                    filtro_complex = f"[0:v]scale=720:1280:force_original_aspect_ratio=increase,crop=720:1280"
-                    if ativar_blur_fundo:
-                        filtro_complex += f",boxblur={blur_strength}:1"
-                    if ativar_sepia:
-                        filtro_complex += ",colorchannelmixer=.393:.769:.189:0:.349:.686:.168:0:.272:.534:.131"
-                    if ativar_granulado:
-                        filtro_complex += ",noise=alls=20:allf=t+u"
-                    if ativar_pb:
-                        filtro_complex += ",hue=s=0.3"
-                    if ativar_vignette:
-                        filtro_complex += ",vignette"
-                    filtro_complex += "[blur];[1:v]" + ",".join(filtros_main) + "[zoomed];[blur][zoomed]overlay=(W-w)/2:(H-h)/2[base]"
-
-                    if ativar_texto and texto_personalizado.strip():
-                        y_pos = "100" if posicao_texto == "Topo" else "(h-text_h)/2" if posicao_texto == "Centro" else "h-text_h-100"
-                        enable = f":enable='lt(t\\,{segundos_texto})'" if duracao_texto == "Apenas primeiros segundos" else ""
-                        texto_clean = texto_personalizado.replace(":", "\\:").replace("'", "\\'")
-                        filtro_complex += f";[base]drawtext=text='{texto_clean}':fontfile=/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf:fontcolor={cor_texto}:fontsize={tamanho_texto}:shadowcolor={cor_sombra}:shadowx=3:shadowy=3:x=(w-text_w)/2:y={y_pos}{enable}[final]"
+                    if musica:
+                        musica_path = os.path.join(temp_dir, "musica_original.mp3")
+                        with open(musica_path, "wb") as f:
+                            f.write(musica.read())
+                        musica_cortada = os.path.join(temp_dir, f"musica_cortada_{n}.aac")
+                        subprocess.run([
+                            "ffmpeg", "-i", musica_path, "-ss", "0", "-t", str(dur_video_real),
+                            "-vn", "-acodec", "aac", "-y", musica_cortada
+                        ], check=True, stderr=subprocess.PIPE)
+                        final_name = f"video_final_{n}_{int(time.time())}.mp4"
+                        subprocess.run([
+                            "ffmpeg", "-i", video_final_raw, "-i", musica_cortada,
+                            "-map", "0:v:0", "-map", "1:a:0",
+                            "-c:v", "copy", "-c:a", "aac",
+                            "-shortest", final_name
+                        ], check=True, stderr=subprocess.PIPE)
                     else:
-                        filtro_complex += ";[base]null[final]"
-
-                    video_editado = os.path.join(temp_dir, f"video_editado_{n}.mp4")
-                    subprocess.run([
-                        "ffmpeg", "-i", fundo_cortado, "-i", video_raw,
-                        "-filter_complex", filtro_complex,
-                        "-map", "[final]", "-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value),
-                        video_editado
-                    ], check=True, stderr=subprocess.PIPE)
-
-                    # Acelerar vídeo final (sem áudio)
-                    video_acelerado = os.path.join(temp_dir, f"video_acelerado_{n}.mp4")
-                    subprocess.run([
-                        "ffmpeg", "-y", "-i", video_editado, "-an",
-                        "-filter:v", f"setpts=PTS/{velocidade_final}",
-                        "-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value),
-                        video_acelerado
-                    ], check=True, stderr=subprocess.PIPE)
-
-                    progresso.progress(70)
-
-                    # Tutorial no meio (se enviado)
-                    video_final_raw = video_acelerado
-                    if video_tutorial:
-                        dur_proc = subprocess.run([
-                            "ffprobe", "-v", "error", "-show_entries", "format=duration",
-                            "-of", "default=noprint_wrappers=1:nokey=1", video_acelerado
-                        ], stdout=subprocess.PIPE)
-                        dur_f = float(dur_proc.stdout.decode().strip())
-                        pt = dur_f / 2 if dur_f < 10 else random.uniform(5, dur_f - 5)
-                        part1 = os.path.join(temp_dir, f"part1_{n}.mp4")
-                        part2 = os.path.join(temp_dir, f"part2_{n}.mp4")
-                        subprocess.run(["ffmpeg", "-i", video_acelerado, "-ss", "0", "-t", str(pt), "-c:v", "libx264", "-preset", "ultrafast", part1], check=True, stderr=subprocess.PIPE)
-                        subprocess.run(["ffmpeg", "-i", video_acelerado, "-ss", str(pt), "-c:v", "libx264", "-preset", "ultrafast", part2], check=True, stderr=subprocess.PIPE)
-                        final_txt = os.path.join(temp_dir, f"final_{n}.txt")
-                        with open(final_txt, "w") as f:
-                            f.write(f"file '{part1}'\nfile '{tutorial_mp4}'\nfile '{part2}'\n")
-                        video_final_raw = os.path.join(temp_dir, f"video_final_raw_{n}.mp4")
-                        subprocess.run(["ffmpeg", "-f", "concat", "-safe", "0", "-i", final_txt, "-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value), video_final_raw], check=True, stderr=subprocess.PIPE)
-
-                    # 🎵 Música final
-                    dur_proc = subprocess.run([
-                        "ffprobe", "-v", "error", "-show_entries", "format=duration",
-                        "-of", "default=noprint_wrappers=1:nokey=1", video_final_raw
-                    ], stdout=subprocess.PIPE)
-                    dur_video_real = float(dur_proc.stdout.decode().strip())
+                        final_name = f"video_final_{n}_{int(time.time())}.mp4"
+                        shutil.copy(video_final_raw, final_name)
+
+                    # 🔒 APLICAR ANTI-FLOP NO FINAL
+                    if ativar_antiflop:
+                        st.info(f"🔒 Aplicando Anti-Flop no vídeo {n+1}...")
+
+                        # ➡️ Anti-Flop filtros:
+                        encoder_name = f"{fingerprint}Cam_{random.randint(1000,9999)}"
+                        antiflop_out = f"antiflop_{n}_{int(time.time())}.mp4"
+
+                        vf_af = (
+                            f"scale=iw*{zoom_af}:ih*{zoom_af},"
+                            f"crop=iw/{zoom_af}:ih/{zoom_af},"
+                            f"eq=brightness={brilho_af}:contrast={contraste_af}:saturation={saturacao_af},"
+                            f"noise=alls={ruido_af}:allf=t,"
+                            f"rotate={rotacao_af}*PI/180:c=black@0.0,"
+                            "scale=trunc(iw/2)*2:trunc(ih/2)*2"  # Garante resolução par
+                        )
+                        if usar_marca:
+                            vf_af += f",drawtext=text='{encoder_name}':fontcolor=white@0.005:x=10:y=10:fontsize=24"
+                        if usar_glitch:
+                            vf_af += ",blackframe=1:0"
+                        if transicoes:
+                            effects = [
+                                "fade=t=in:st=0:d=0.5",
+                                "fade=t=out:st=3:d=0.5",
+                                "zoompan=z='min(pzoom+0.0015,1.5)':d=1",
+                                "drawbox=x=0:y=0:w=iw:h=ih:color=white@0.02:t=fill"
+                            ]
+                            vf_af += "," + random.choice(effects)
+
+                        af = f"asetrate=44100*{pitch_af},aresample=44100"
+                        profile = "baseline" if fingerprint in ["Android", "Xiaomi"] else "main"
+                        if fingerprint == "iPhone":
+                            profile = "high"
+                        level = "4.0" if fingerprint == "Samsung" else "3.1"
+                        ar = "44100" if fingerprint in ["iPhone", "Xiaomi"] else "48000"
+
+                        subprocess.run([
+                            'ffmpeg', '-i', final_name,
+                            '-vf', vf_af,
+                            '-af', af,
+                            '-r', str(fps_af),
+                            '-c:v', 'libx264', '-profile:v', profile, '-level', level,
+                            '-b:v', f"{bitrate_af}k", '-ar', ar,
+                            '-preset', 'fast', '-tune', 'zerolatency',
+                            '-movflags', '+faststart',
+                            antiflop_out
+                        ], check=True)
+
+                        videos_gerados.append(antiflop_out)
+                        st.video(antiflop_out)
+                        with open(antiflop_out, "rb") as f:
+                            st.download_button(f"📥 Baixar vídeo {n+1} Anti-Flop", f, file_name=antiflop_out)
+                    else:
+                        videos_gerados.append(final_name)
+                        st.video(final_name)
+                        with open(final_name, "rb") as f:
+                            st.download_button(f"📥 Baixar vídeo {n+1}", f, file_name=final_name)
+
+                progresso.progress(100)
+                st.success("✅ Todos os vídeos foram gerados com sucesso!")
+
+except subprocess.CalledProcessError as e:
+    st.error(f"❌ Erro ao gerar vídeo:\n\n{e.stderr.decode(errors='ignore')}")
+
+finally:
+    shutil.rmtree(temp_dir)
+
+# ➡️ Fora do try-finally: Botão "Baixar Todos os Vídeos"
+if videos_gerados:
+    st.markdown("### 📥 Baixar Todos os Vídeos")
+    download_links = ""
+    for v in videos_gerados:
+        download_links += f"<a href='{v}' download='{v}'></a>\n"
+
+    st.markdown(
+        f"""
+        <button onclick="baixarTodos()">📥 Baixar Todos</button>
+        <script>
+        function baixarTodos() {{
+            const links = Array.from(document.querySelectorAll('a[download]'));
+            for (let l of links) {{
+                l.click();
+            }}
+        }}
+        </script>
+        {download_links}
+        """,
+        unsafe_allow_html=True
+    )