Update app.py

app.py

@@ -1,176 +1,164 @@
 import streamlit as st
 import subprocess
 import os
-import uuid
 import random
+import tempfile
 import shutil
-
-st.set_page_config(page_title="Anti-Flop PRO MAX", layout="centered")
-st.title("🚀 Anti-Flop PRO MAX – Gerador com Fingerprint Invisível")
-
-uploaded_file = st.file_uploader("📤 Envie seu vídeo (MP4)", type=["mp4"])
-
-if uploaded_file:
-    raw_input_filename = f"raw_{uuid.uuid4().hex}.mp4"
-    with open(raw_input_filename, "wb") as f:
-        f.write(uploaded_file.read())
-
-    input_filename = f"cleaned_{uuid.uuid4().hex}.mp4"
-    subprocess.run(['ffmpeg', '-i', raw_input_filename, '-map_metadata', '-1', '-c', 'copy', input_filename])
-    os.remove(raw_input_filename)
-
-    st.video(input_filename)
-    st.success("🎥 Vídeo carregado e metadados limpos!")
-
-    st.subheader("🎛️ Ajustes Visuais e Áudio (padrão)")
-
-    if st.button("🎲 RANDOMIZAR TUDO"):
-        st.session_state.zoom_default = random.uniform(1.00, 1.05)
-        st.session_state.brilho_default = random.uniform(-0.05, 0.05)
-        st.session_state.contraste_default = random.uniform(0.95, 1.05)
-        st.session_state.saturacao_default = random.uniform(0.95, 1.05)
-        st.session_state.ruido_default = random.randint(3, 15)
-        st.session_state.rotacao_default = random.uniform(0.2, 1.5)
-        st.session_state.pitch_default = random.uniform(1.00, 1.08)
-        st.session_state.fps_val = random.choice([30, 29.97, 25])
-        st.session_state.bitrate_default = random.randint(900, 1500)
-        st.session_state.velocidade_default = random.uniform(1.00, 1.03)
-        st.rerun()
-
-    zoom_default = st.slider("Zoom (%)", 100, 105, int(st.session_state.get('zoom_default', 1.01) * 100)) / 100
-    brilho_default = st.slider("Brilho", -0.1, 0.1, st.session_state.get('brilho_default', 0.01))
-    contraste_default = st.slider("Contraste", 0.9, 1.1, st.session_state.get('contraste_default', 1.01))
-    saturacao_default = st.slider("Saturação", 0.9, 1.1, st.session_state.get('saturacao_default', 1.01))
-    ruido_default = st.slider("Ruído Visual (0-20)", 0, 20, st.session_state.get('ruido_default', 5))
-    rotacao_default = st.slider("Rotação (graus)", 0.0, 2.0, st.session_state.get('rotacao_default', 0.5))
-    pitch_default = st.slider("Pitch do Áudio (%)", 100, 110, int(st.session_state.get('pitch_default', 1.01) * 100)) / 100
-    fps = st.selectbox("FPS Final", [30, 29.97, 25], index=[30, 29.97, 25].index(st.session_state.get('fps_val', 30)))
-    bitrate_default = st.slider("Bitrate (kbps)", 800, 2000, st.session_state.get('bitrate_default', 1100))
-    velocidade_default = st.slider("Velocidade do Vídeo (%)", 100, 103, int(st.session_state.get('velocidade_default', 1.01) * 100)) / 100
-
-    st.subheader("🧩 Extras Anti-Detecção")
-    usar_marca = st.checkbox("Marca d’água Fantasma com perfil", value=True)
-    usar_glitch = st.checkbox("Inserir Frame Preto Aleatório", value=True)
-    res_random = st.checkbox("Resolução Aleatória", value=True)
-    cortar = st.checkbox("Cortar trecho aleatório do vídeo", value=True)
-    embaralhar = st.checkbox("Embaralhar trechos do vídeo", value=True)
-
-    st.subheader("🎨 Avançado")
-    lote_efeitos = st.checkbox("🎨 Aplicar efeitos por lote (aleatórios)", value=True)
-    transicoes = st.checkbox("🔄 Transições visuais aleatórias", value=True)
-
-    st.subheader("🕵️‍♂️ Perfil Fingerprint")
-    fingerprint = st.selectbox("Escolha o Perfil", ["Android", "iPhone", "Câmera", "CapCut", "Samsung", "Xiaomi", "LG"])
-
-    qtd_videos = st.number_input("Quantidade de vídeos únicos:", min_value=1, max_value=10, value=3)
-
-    if st.button("🎬 Gerar Vídeos Únicos"):
-        st.info("Processando vídeos... aguarde!")
-        output_files = []
-
-        if embaralhar:
-            temp_dir = f"temp_{uuid.uuid4().hex}"
-            os.makedirs(temp_dir, exist_ok=True)
-            subprocess.run(['ffmpeg', '-i', input_filename, '-c', 'copy', '-f', 'segment', '-segment_time', '2', f'{temp_dir}/out%03d.mp4'])
-            files = os.listdir(temp_dir)
-            files.sort()
-            random.shuffle(files)
-            concat_list = f"{temp_dir}/list.txt"
-            with open(concat_list, 'w') as f:
-                for file in files:
-                    f.write(f"file '{os.path.join(temp_dir, file)}'\n")
-            shuffled_input = f"shuffled_{uuid.uuid4().hex}.mp4"
-            subprocess.run(['ffmpeg', '-f', 'concat', '-safe', '0', '-i', concat_list, '-c', 'copy', shuffled_input])
-            shutil.rmtree(temp_dir)
-            base_video = shuffled_input if os.path.exists(shuffled_input) else input_filename
-        else:
-            base_video = input_filename
-
-        for _ in range(qtd_videos):
-            encoder_name = f"{fingerprint}Cam_{random.randint(1000,9999)}"
-            final_out = f"{encoder_name}_video_{uuid.uuid4().hex}.mp4"
-
-            if lote_efeitos:
-                zoom_r = random.uniform(1.00, 1.05)
-                brilho_r = random.uniform(-0.05, 0.05)
-                contraste_r = random.uniform(0.95, 1.05)
-                saturacao_r = random.uniform(0.95, 1.05)
-                ruido_r = random.randint(3, 15)
-                rot_r = random.uniform(0.2, 1.5) * 0.01745
-                pitch_r = random.uniform(1.00, 1.08)
-                velocidade_r = random.uniform(1.00, 1.03)
-                bitrate_r = f"{random.randint(900, 1500)}k"
-            else:
-                zoom_r = zoom_default
-                brilho_r = brilho_default
-                contraste_r = contraste_default
-                saturacao_r = saturacao_default
-                ruido_r = ruido_default
-                rot_r = rotacao_default * 0.01745
-                pitch_r = pitch_default
-                velocidade_r = velocidade_default
-                bitrate_r = f"{bitrate_default}k"
-
-            vf = (
-                f"scale=iw*{zoom_r}:ih*{zoom_r},"
-                f"crop=iw/{zoom_r}:ih/{zoom_r},"
-                f"drawbox=color=black@0.05:t=fill,"
-                f"eq=brightness={brilho_r}:contrast={contraste_r}:saturation={saturacao_r},"
-                f"noise=alls={ruido_r}:allf=t,"
-                f"rotate={rot_r}:c=black@0.0"
-            )
-
-            if usar_marca:
-                vf += f",drawtext=text='{encoder_name}':fontcolor=white@0.005:x=10:y=10:fontsize=24"
-            if usar_glitch:
-                vf += ",blackframe=1:0"
-            if res_random:
-                w = random.choice([720, 960, 1080])
-                h = random.choice([1280, 1440, 1920])
-                vf += f",scale={w}:{h}"
-            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 += "," + random.choice(effects)
-
-            af = f"asetrate=44100*{pitch_r},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"
-
-            codec_cmd = ['-c:v', 'libx264', '-profile:v', profile, '-level', level]
-            audio_cmd = ['-ar', ar]
-            filters = ['-vf', vf, '-af', af, '-r', str(fps)]
-            speed_filter = ['-filter:v', f"setpts={1/velocidade_r}*PTS"]
-            speed_audio = ['-filter:a', f"atempo={velocidade_r}"]
-            duration_cmd = ['-ss', str(random.uniform(0, 1))] if cortar else []
-
-            subprocess.run([
-                'ffmpeg', '-i', base_video, *duration_cmd,
-                *filters, *speed_filter, *speed_audio,
-                *codec_cmd, '-b:v', bitrate_r,
-                *audio_cmd,
-                '-preset', 'fast', '-tune', 'zerolatency',
-                '-movflags', '+faststart',
-                final_out
-            ])
-
-            output_files.append(final_out)
-
-        if base_video != input_filename and os.path.exists(base_video):
-            os.remove(base_video)
-        os.remove(input_filename)
-
-        st.success(f"✅ {qtd_videos} vídeos gerados com fingerprint invisível!")
-
-        for file in output_files:
-            st.video(file)
-            with open(file, "rb") as f:
-                st.download_button(f"📥 Baixar {file}", data=f, file_name=file, mime="video/mp4")
-            os.remove(file)
+import time
+
+st.set_page_config(page_title="TikTok Video Generator", layout="centered")
+st.title("🎥 TikTok Video Generator - PRO")
+
+st.markdown("Envie seus vídeos e gere conteúdo com efeitos, zoom, texto, música e filtros!")
+
+# Uploads dos vídeos de cortes (principal)
+st.markdown("<div style='background-color:#F0F0FF;padding:10px;border-radius:8px'>", unsafe_allow_html=True)
+st.subheader("🎬 Envie os vídeos principais (cortes)")
+cortes = st.file_uploader("Vídeos de cortes", type=["mp4"], accept_multiple_files=True)
+st.markdown("</div>", unsafe_allow_html=True)
+
+# Upload do vídeo de fundo (opcional)
+st.markdown("<div style='background-color:#FFF0F0;padding:10px;border-radius:8px;margin-top:15px;'>", unsafe_allow_html=True)
+st.subheader("🌄 Envie o vídeo de fundo (opcional)")
+video_fundo = st.file_uploader("Vídeo de Fundo (com blur e efeitos)", type=["mp4", "mov"])
+st.markdown("</div>", unsafe_allow_html=True)
+
+# Uploads adicionais
+video_tutorial = st.file_uploader("📎 Tutorial (opcional)", type="mp4")
+musica = st.file_uploader("🎵 Música (opcional, MP3)", type=["mp3"])
+
+# Configurações gerais
+num_videos_finais = st.number_input("Quantos vídeos gerar?", 1, 10, 1)
+duracao_final = st.number_input("Duração final (s)", 10, 300, 30)
+duracao_corte = st.number_input("Duração de cada corte (s)", 1, 30, 5)
+zoom = st.slider("Zoom no vídeo principal", 1.0, 2.0, 1.0, 0.1)
+blur_strength = st.slider("Blur no fundo", 1, 50, 10)
+velocidade_cortes = st.slider("Velocidade dos cortes", 0.5, 2.0, 1.0, 0.1)
+velocidade_final = st.slider("Velocidade final", 0.5, 2.0, 1.0, 0.1)
+crf_value = st.slider("Qualidade CRF", 18, 30, 23)
+
+# Texto com emojis
+st.write("### Texto no Vídeo")
+ativar_texto = st.checkbox("Ativar texto", value=False)
+if ativar_texto:
+    texto_personalizado = st.text_input("Texto (emojis permitidos)", "🔥 Siga para mais!")
+    posicao_texto = st.selectbox("Posição do texto", ["Topo", "Centro", "Base"])
+    duracao_texto = st.radio("Duração do texto", ["Vídeo todo", "Apenas primeiros segundos"])
+    segundos_texto = 5
+    if duracao_texto == "Apenas primeiros segundos":
+        segundos_texto = st.slider("Segundos", 1, 30, 5)
+    cor_texto = st.color_picker("Cor do texto", "#FFFF00")
+    cor_sombra = st.color_picker("Cor da sombra", "#000000")
+    tamanho_texto = st.slider("Tamanho da fonte", 20, 100, 60, step=2)
+
+# Filtros no fundo
+st.write("### Filtros no fundo")
+ativar_blur_fundo = st.checkbox("Ativar blur no fundo", value=True)
+ativar_sepia = st.checkbox("Sépia", False)
+ativar_granulado = st.checkbox("Granulado", False)
+ativar_pb = st.checkbox("Preto e branco", False)
+ativar_vignette = st.checkbox("Vignette", False)
+
+# Efeitos extras
+st.write("### Outros efeitos")
+ativar_espelhar = st.checkbox("Espelhar vídeo", True)
+ativar_filtro_cor = st.checkbox("Ajuste de cor", True)
+remover_borda = st.checkbox("Remover borda do vídeo")
+tamanho_borda = st.slider("Tamanho da borda (px)", 0, 200, 0, 5)
+
+# Borda personalizada
+ativar_borda_personalizada = st.checkbox("Borda personalizada", False)
+if ativar_borda_personalizada:
+    cor_borda = st.color_picker("Cor da borda", "#FF0000")
+    animacao_borda = st.selectbox("Animação da borda", ["Nenhuma", "Borda Pulsante", "Cor Animada", "Neon", "Ondulada"])
+
+# 💥 Anti-Flop
+st.write("### 🔒 Anti-Flop PRO MAX")
+ativar_antiflop = st.checkbox("Ativar Anti-Flop", value=False)
+if ativar_antiflop:
+    perfil_fingerprint = st.selectbox("Perfil Fingerprint", ["Android", "iPhone", "Câmera", "CapCut", "Samsung", "Xiaomi", "LG"])
+    lote_efeitos = st.checkbox("Aplicar efeitos por lote (aleatórios)", value=True)
+    usar_marca = st.checkbox("Marca d’água fantasma", value=True)
+    usar_glitch = st.checkbox("Inserir frame preto aleatório", value=True)
+    res_random = st.checkbox("Resolução aleatória", value=True)
+    transicoes = st.checkbox("Transições visuais aleatórias", value=True)
+
+# BOTÃO PRINCIPAL
+if st.button("Gerar Vídeo(s)"):
+    if not cortes:
+        st.error("❌ Envie os vídeos de cortes.")
+    else:
+        with st.spinner("🎬 Processando..."):
+            progresso = st.progress(0)
+            temp_dir = tempfile.mkdtemp()
+
+            try:
+                # (Aqui segue todo o processamento original do video-generator.py...)
+
+                # Após gerar cada vídeo final:
+                # video_final é o vídeo já pronto (com música)
+
+                for n in range(num_videos_finais):
+                    final_name = f"video_final_{n}_{int(time.time())}.mp4"
+
+                    # Aqui você insere o código Anti-Flop, se ativado
+                    if ativar_antiflop:
+                        antiflop_out = os.path.join(temp_dir, f"antiflop_{n}.mp4")
+                        zoom_r = random.uniform(1.00, 1.05) if lote_efeitos else 1.02
+                        brilho_r = random.uniform(-0.05, 0.05) if lote_efeitos else 0.01
+                        contraste_r = random.uniform(0.95, 1.05) if lote_efeitos else 1.01
+                        saturacao_r = random.uniform(0.95, 1.05) if lote_efeitos else 1.01
+                        ruido_r = random.randint(3, 15) if lote_efeitos else 5
+                        rot_r = random.uniform(0.2, 1.5) * 0.01745 if lote_efeitos else 0.5 * 0.01745
+                        pitch_r = random.uniform(1.00, 1.08) if lote_efeitos else 1.01
+                        velocidade_r = random.uniform(1.00, 1.03) if lote_efeitos else 1.01
+                        bitrate_r = f"{random.randint(900, 1500)}k" if lote_efeitos else "1100k"
+                        encoder_name = f"{perfil_fingerprint}Cam_{random.randint(1000,9999)}"
+
+                        vf = (
+                            f"scale=iw*{zoom_r}:ih*{zoom_r},"
+                            f"crop=iw/{zoom_r}:ih/{zoom_r},"
+                            f"drawbox=color=black@0.05:t=fill,"
+                            f"eq=brightness={brilho_r}:contrast={contraste_r}:saturation={saturacao_r},"
+                            f"noise=alls={ruido_r}:allf=t,"
+                            f"rotate={rot_r}:c=black@0.0"
+                        )
+                        if usar_marca:
+                            vf += f",drawtext=text='{encoder_name}':fontcolor=white@0.005:x=10:y=10:fontsize=24"
+                        if usar_glitch:
+                            vf += ",blackframe=1:0"
+                        if res_random:
+                            w = random.choice([720, 960, 1080])
+                            h = random.choice([1280, 1440, 1920])
+                            vf += f",scale={w}:{h}"
+                        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 += "," + random.choice(effects)
+
+                        af = f"asetrate=44100*{pitch_r},aresample=44100"
+                        subprocess.run([
+                            "ffmpeg", "-i", final_name,
+                            "-vf", vf, "-af", af,
+                            "-c:v", "libx264", "-preset", "fast", "-b:v", bitrate_r,
+                            "-c:a", "aac", "-movflags", "+faststart",
+                            antiflop_out
+                        ], check=True)
+                        final_name = antiflop_out
+
+                    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)