Update app.py

app.py

@@ -6,7 +6,7 @@ import tempfile
 import shutil
 import time
 
-st.set_page_config(page_title="TikTok Video Generator - PRO", layout="centered")
+st.set_page_config(page_title="TikTok Video Generator - PRO + Anti-Flop", layout="centered")
 st.title("🎥 TikTok Video Generator - PRO + Anti-Flop")
 
 st.markdown("Envie seus vídeos e gere conteúdo com efeitos, zoom, texto, música, filtros e antiflop!")
@@ -115,6 +115,9 @@ if ativar_antiflop:
 # 💾 Download Automático
 st.write("### 💾 Download Automático")
 download_automatico = st.checkbox("Ativar Download Automático", value=False)
+# Lista para armazenar os vídeos gerados
+videos_gerados = []
+
 # Botão principal
 if st.button("Gerar Vídeo(s)"):
     if not cortes:
@@ -302,70 +305,96 @@ 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)
 
-                    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:
-                        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}...")
-                        # ... [todo o bloco do Anti-Flop aqui, como te enviei antes] ...
-                        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)
+                    # 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]"
                     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
-    )
-                   
+                        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())