Update app.py

app.py

@@ -1,164 +1,176 @@
 import streamlit as st
 import subprocess
 import os
+import uuid
 import random
-import tempfile
 import shutil
-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"[email protected]: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}':[email protected]: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:[email protected]: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)
+
+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"[email protected]:t=fill,"
+                f"eq=brightness={brilho_r}:contrast={contraste_r}:saturation={saturacao_r},"
+                f"noise=alls={ruido_r}:allf=t,"
+                f"rotate={rot_r}:[email protected]"
+            )
+
+            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)