yt / app.py
pcdoido2's picture
Update app.py
1df1826 verified
raw
history blame
27.5 kB
import streamlit as st
import subprocess
import os
import random
import tempfile
import shutil
import time
import base64
st.set_page_config(page_title="TikTok Video Generator - PRO", layout="centered")
abas = ["🎬 Gerador de Vídeo", "📂 Gerenciador de Arquivos", "🔐 Admin"]
pagina = st.sidebar.radio("Escolha uma página:", abas)
CATEGORIAS = ["AVATAR WORLD","TOCA LIFE","BLOX FRUITS","VOX SEAS","STEAL A BRAINROT","FC MOBILE"]
BASE_ASSETS = "assets"
BASE_SALVOS = "uploaded_files"
os.makedirs("assets/musicas", exist_ok=True)
for cat in CATEGORIAS:
os.makedirs(os.path.join(BASE_ASSETS, cat, "cortes"), exist_ok=True)
os.makedirs(os.path.join(BASE_ASSETS, cat, "tutoriais"), exist_ok=True)
os.makedirs(os.path.join(BASE_SALVOS, cat), exist_ok=True)
if pagina == "🎬 Gerador de Vídeo":
st.title("🎥 TikTok Video Generator - PRO")
categoria_selecionada = st.selectbox("Escolha a categoria:", ["Selecione..."] + CATEGORIAS)
if categoria_selecionada == "Selecione...":
st.warning("👈 Por favor, selecione uma categoria antes de continuar.")
st.stop()
st.markdown("### 🎛️ Opções de Mídia")
usar_tutorial = st.checkbox("Ativar tutorial", value=True)
usar_fundo = st.checkbox("Ativar fundo", value=True)
usar_musica = st.checkbox("Ativar música", value=False)
path_cortes = os.path.join(BASE_ASSETS, categoria_selecionada, "cortes")
path_tutoriais = os.path.join(BASE_ASSETS, categoria_selecionada, "tutoriais")
path_musicas = os.path.join(BASE_ASSETS, "musicas")
cortes_files = [os.path.join(path_cortes, f) for f in os.listdir(path_cortes) if f.endswith(".mp4")]
tutoriais_files = [os.path.join(path_tutoriais, f) for f in os.listdir(path_tutoriais) if f.endswith(".mp4")]
musicas_files = [os.path.join(path_musicas, f) for f in os.listdir(path_musicas) if f.endswith(".mp3")]
if not cortes_files:
st.error("❌ Nenhum vídeo de corte encontrado. Use a aba Admin.")
else:
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)
st.write("## Configuração dos cortes")
aleatorizar = st.checkbox("🎲 Aleatorizar configurações por vídeo")
duracao_corte_min = st.slider("Duração mínima (s)", 1, 10, 3)
duracao_corte_max = st.slider("Duração máxima (s)", duracao_corte_min + 1, 20, 5)
# 🔥 Nova opção: posição do corte
posicao_corte = st.selectbox("Posição do corte", ["Centro", "Esquerda", "Direita"])
zoom = st.slider("Zoom", 1.0, 2.0, 1.0, 0.1)
blur_strength = st.slider("Blur no fundo", 1, 50, 10)
velocidade_final = st.slider("Velocidade final", 0.5, 2.0, 1.0, 0.1)
crf_value = st.slider("Qualidade CRF", 18, 30, 18)
st.write("## Texto no Vídeo")
ativar_texto = st.checkbox("Ativar texto")
if ativar_texto:
texto_personalizado = st.text_input("Texto", "🔥 Siga para mais!")
posicao_texto = st.selectbox("Posição", ["Topo", "Centro", "Base"])
duracao_texto = st.radio("Duração do texto", ["Vídeo todo", "Apenas primeiros segundos"])
segundos_texto = st.slider("Segundos", 1, 30, 5) if duracao_texto == "Apenas primeiros segundos" else 0
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)
st.write("## Filtros no fundo")
ativar_blur_fundo = st.checkbox("Ativar blur", value=True)
ativar_sepia = st.checkbox("Sépia")
ativar_granulado = st.checkbox("Granulado")
ativar_pb = st.checkbox("Preto e branco")
ativar_vignette = st.checkbox("Vignette")
with st.expander("🎨 Filtros Avançados"):
ativar_brilho = st.checkbox("Brilho extra")
ativar_contraste = st.checkbox("Contraste forte")
ativar_colorboost = st.checkbox("Cores intensas")
ativar_azul = st.checkbox("Filtro Azul")
ativar_quente = st.checkbox("Filtro Quente")
ativar_desaturar = st.checkbox("Desaturar")
ativar_vhs = st.checkbox("Efeito VHS")
st.write("## Outros efeitos")
ativar_espelhar = st.checkbox("Espelhar", True)
ativar_filtro_cor = st.checkbox("Ajuste de cor", True)
remover_borda = st.checkbox("Remover borda")
tamanho_borda = st.slider("Tamanho da borda (px)", 0, 200, 0, 5)
ativar_borda_personalizada = st.checkbox("Borda personalizada")
if ativar_borda_personalizada:
cor_borda = st.color_picker("Cor da borda", "#FF0000")
animacao_borda = st.selectbox("Animação", ["Nenhuma", "Pulsante", "Cor Animada", "Neon", "Ondulada"])
salvar_no_gerenciador = st.checkbox("Salvar no gerenciador de arquivos")
if st.button("Gerar Vídeo(s)", key="gerar_video_usuario"):
with st.spinner("🎬 Processando..."):
progresso = st.progress(0)
temp_dir = tempfile.mkdtemp()
def atualizar_barra(n, etapa, total_videos, total_etapas):
progresso.progress(min(100, int(((n + etapa / total_etapas) / total_videos) * 100)))
try:
cortes_names = cortes_files.copy()
tutorials_salvos = tutoriais_files.copy()
musicas_salvas = musicas_files.copy()
for n in range(num_videos_finais):
total_etapas = 6
etapa_atual = 0
tempo_total = 0
cortes_prontos = []
cortes_usados = []
# === Config aleatória ou fixa ===
if aleatorizar:
duracao_corte_min_n = random.randint(1, 5)
duracao_corte_max_n = random.randint(duracao_corte_min_n + 1, 8)
zoom_n = round(random.uniform(1.0, 1.5), 2)
blur_strength_n = random.randint(5, 30)
velocidade_final_n = velocidade_final
remover_borda_n = remover_borda and random.choice([True, False])
tamanho_borda_n = random.randint(0, 50) if remover_borda_n else 0
ativar_blur_fundo_n = random.choice([True, False])
ativar_espelhar_n = random.choice([True, False])
else:
duracao_corte_min_n = duracao_corte_min
duracao_corte_max_n = duracao_corte_max
zoom_n = zoom
blur_strength_n = blur_strength
velocidade_final_n = velocidade_final
remover_borda_n = remover_borda
tamanho_borda_n = tamanho_borda
ativar_blur_fundo_n = ativar_blur_fundo
ativar_espelhar_n = ativar_espelhar
# === Etapa 1 - Fundo ===
if usar_fundo:
fundo_origem = random.choice(cortes_names)
fundo_convertido = os.path.join(temp_dir, f"fundo_convertido_{n}.mp4")
dur_proc = subprocess.run([
"ffprobe", "-v", "error", "-show_entries", "format=duration",
"-of", "default=noprint_wrappers=1:nokey=1", fundo_origem
], stdout=subprocess.PIPE)
try:
dur_total = float(dur_proc.stdout.decode().strip())
ponto_inicio = random.uniform(0, max(0, dur_total - duracao_final))
except:
ponto_inicio = 0.0
subprocess.run([
"ffmpeg", "-ss", str(ponto_inicio), "-i", fundo_origem,
"-t", str(duracao_final),
"-vf",
"scale='if(gt(iw/ih,720/1280),max(720,iw),-2)':'if(gt(iw/ih,720/1280),-2,max(1280,ih))',"
"scale=720:1280:force_original_aspect_ratio=increase,crop=720:1280,fps=30",
"-preset", "ultrafast", "-crf", "18",
fundo_convertido
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
else:
fundo_convertido = os.path.join(temp_dir, f"fundo_vazio_{n}.mp4")
subprocess.run([
"ffmpeg", "-f", "lavfi", "-i", f"color=color=black:size=720x1280:d={duracao_final}:rate=30",
"-c:v", "libx264", "-t", str(duracao_final), "-preset", "ultrafast", "-crf", "18",
fundo_convertido
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
etapa_atual += 1
atualizar_barra(n, etapa_atual, num_videos_finais, total_etapas)
# === Etapa 2 - Cortes ===
tentativas = 0
while tempo_total < duracao_final and tentativas < 100:
tentativas += 1
random.shuffle(cortes_names)
for c in cortes_names:
dur = subprocess.run([
"ffprobe", "-v", "error", "-show_entries", "format=duration",
"-of", "default=noprint_wrappers=1:nokey=1", c
], stdout=subprocess.PIPE).stdout.decode().strip()
if not dur:
continue
try:
d = float(dur)
if d < duracao_corte_min_n + 0.5:
continue
duracao_aleatoria = random.uniform(duracao_corte_min_n, min(d, duracao_corte_max_n))
ini = random.uniform(0, d - duracao_aleatoria)
repetido = any(c == usado[0] and abs(ini - usado[1]) < 0.5 for usado in cortes_usados)
if repetido:
continue
out = os.path.join(temp_dir, f"cut_{random.randint(1000,9999)}.mp4")
# 🔥 Define filtro de crop conforme posição escolhida
if posicao_corte == "Centro":
crop_filter = "crop=720:1280:(in_w-720)/2:(in_h-1280)/2"
elif posicao_corte == "Esquerda":
crop_filter = "crop=720:1280:0:(in_h-1280)/2"
elif posicao_corte == "Direita":
crop_filter = "crop=720:1280:(in_w-720):(in_h-1280)/2"
else:
crop_filter = "scale=trunc(iw/2)*2:trunc(ih/2)*2"
subprocess.run([
"ffmpeg", "-ss", str(ini), "-i", c, "-t", str(duracao_aleatoria),
"-vf", crop_filter,
"-an", "-c:v", "libx264", "-preset", "ultrafast", "-crf", "30", out
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
cortes_prontos.append(out)
cortes_usados.append((c, ini, duracao_aleatoria))
tempo_total += duracao_aleatoria
if tempo_total >= duracao_final:
break
except:
continue
lista = os.path.join(temp_dir, f"lista_{n}.txt")
with open(lista, "w") as f:
for c in cortes_prontos:
f.write(f"file '{c}'\n")
video_raw = os.path.join(temp_dir, f"video_raw_{n}.mp4")
subprocess.run([
"ffmpeg", "-f", "concat", "-safe", "0", "-i", lista,
"-c:v", "libx264", "-preset", "ultrafast", "-crf", "30", video_raw
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
etapa_atual += 1
atualizar_barra(n, etapa_atual, num_videos_finais, total_etapas)
# === Etapa 3 - Filtros ===
filtros_main = ["scale=720:1280:force_original_aspect_ratio=decrease"]
if zoom_n != 1.0:
filtros_main.append(f"scale=iw*{zoom_n}:ih*{zoom_n}")
if ativar_espelhar_n:
filtros_main.append("hflip")
if remover_borda_n and tamanho_borda_n > 0:
filtros_main.append(f"crop=in_w-{tamanho_borda_n*2}:in_h-{tamanho_borda_n*2}")
if ativar_filtro_cor:
filtros_main.append("eq=contrast=1.1:saturation=1.2")
filtros_main.append("scale=trunc(iw/2)*2:trunc(ih/2)*2")
if ativar_borda_personalizada:
cor_ffmpeg = f"0x{cor_borda.lstrip('#')}FF"
filtros_main.append(f"drawbox=x=0:y=0:w=iw:h=ih:color={cor_ffmpeg}:t=5")
filtro_complex = f"[0:v]scale=720:1280:force_original_aspect_ratio=increase,crop=720:1280"
if ativar_blur_fundo_n:
filtro_complex += f",boxblur={blur_strength_n}: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"
if ativar_brilho:
filtro_complex += ",eq=brightness=0.05"
if ativar_contraste:
filtro_complex += ",eq=contrast=1.3"
if ativar_colorboost:
filtro_complex += ",eq=saturation=1.5"
if ativar_azul:
filtro_complex += ",colorbalance=bs=0.5"
if ativar_quente:
filtro_complex += ",colorbalance=rs=0.3"
if ativar_desaturar:
filtro_complex += ",hue=s=0.5"
if ativar_vhs:
filtro_complex += ",noise=alls=10:allf=t+u,format=yuv420p"
filtro_complex += "[blur];"
filtro_complex += f"[1:v]{','.join(filtros_main)}[zoomed];"
filtro_complex += "[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}':"
filtro_complex += (
f"fontfile=/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf:"
f"fontcolor={cor_texto}:fontsize={tamanho_texto}:"
f"shadowcolor={cor_sombra}:shadowx=3:shadowy=3:"
f"x=(w-text_w)/2:y={y_pos}{enable}[final]"
)
else:
filtro_complex += ";[base]null[final]"
video_editado = os.path.join(temp_dir, f"video_editado_{n}.mp4")
subprocess.run([
"ffmpeg", "-i", fundo_convertido, "-i", video_raw,
"-filter_complex", filtro_complex,
"-map", "[final]",
"-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value),
video_editado
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
etapa_atual += 1
atualizar_barra(n, etapa_atual, num_videos_finais, total_etapas)
# === Etapa 4 - VELOCIDADE (corrigido) ===
video_com_velocidade = os.path.join(temp_dir, f"video_com_velocidade_{n}.mp4")
subprocess.run([
"ffmpeg", "-y", "-i", video_editado, "-an",
"-filter:v", f"setpts=PTS/{velocidade_final_n},fps=30",
"-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value),
video_com_velocidade
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
# === Etapa 5 - Tutorial (sem ser acelerado) ===
if usar_tutorial and tutorials_salvos:
tutorial_path = random.choice(tutorials_salvos)
tutorial_mp4 = os.path.join(temp_dir, f"tutorial_conv_{n}.mp4")
subprocess.run([
"ffmpeg", "-i", tutorial_path, "-vf", "scale=720:1280,fps=30",
"-c:v", "libx264", "-preset", "ultrafast", "-crf", str(crf_value),
"-y", tutorial_mp4
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
dur_proc = subprocess.run([
"ffprobe", "-v", "error", "-show_entries", "format=duration",
"-of", "default=noprint_wrappers=1:nokey=1", video_com_velocidade
], stdout=subprocess.PIPE)
dur_f = float(dur_proc.stdout.decode().strip() or 0)
pt = dur_f / 2 if dur_f < 10 else random.uniform(5, max(6, 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_com_velocidade, "-ss", "0", "-t", str(pt),
"-c:v", "libx264", "-preset", "ultrafast", part1],
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
subprocess.run(["ffmpeg", "-i", video_com_velocidade, "-ss", str(pt),
"-c:v", "libx264", "-preset", "ultrafast", part2],
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
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], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
else:
video_final_raw = video_com_velocidade
# === Etapa 6 - Música e saída final ===
dur_final_str = subprocess.run([
"ffprobe", "-v", "error", "-show_entries", "format=duration",
"-of", "default=noprint_wrappers=1:nokey=1", video_final_raw
], stdout=subprocess.PIPE).stdout.decode().strip()
if not dur_final_str:
continue
dur_video_real = float(dur_final_str)
final_name = f"video_final_{n}_{int(time.time())}.mp4"
if usar_musica and musicas_salvas:
musica_path = random.choice(musicas_salvas)
musica_cortada = os.path.join(temp_dir, f"musica_{n}.aac")
subprocess.run(["ffmpeg", "-i", musica_path, "-ss", "0", "-t", str(dur_video_real),
"-vn", "-acodec", "aac", "-y", musica_cortada],
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
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],
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
else:
shutil.copy(video_final_raw, final_name)
if salvar_no_gerenciador:
destino = os.path.join(BASE_SALVOS, categoria_selecionada, final_name)
shutil.move(final_name, destino)
st.success(f"✅ Vídeo {n+1} salvo na categoria '{categoria_selecionada}'.")
else:
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)
except Exception as e:
st.error(f"❌ Erro inesperado: {str(e)}")
finally:
shutil.rmtree(temp_dir)
# ======================== GERENCIADOR ========================
elif pagina == "📂 Gerenciador de Arquivos":
st.header("📂 Vídeos Salvos por Categoria")
for categoria in CATEGORIAS:
pasta = os.path.join(BASE_SALVOS, categoria)
os.makedirs(pasta, exist_ok=True)
arquivos = os.listdir(pasta)
st.subheader(f"📁 {categoria}")
if not arquivos:
st.info("Nenhum vídeo salvo.")
else:
for file in arquivos:
file_path = os.path.join(pasta, file)
st.markdown("---")
st.markdown(f"**🎞️ {file}**")
col1, col2, col3 = st.columns(3)
with col1:
if st.button(f"▶ Assistir", key=f"play_{categoria}_{file}"):
st.video(file_path)
with col2:
with open(file_path, "rb") as f_obj:
st.download_button("⬇ Download", f_obj, file_name=file, key=f"download_{categoria}_{file}")
with col3:
with open(file_path, "rb") as f_obj:
if st.download_button("⬇ Baixar & Apagar", f_obj, file_name=file, key=f"down_del_{categoria}_{file}"):
os.remove(file_path)
st.success(f"✅ Arquivo '{file}' baixado e excluído.")
st.rerun()
if st.button("🗑 Excluir", key=f"delete_{categoria}_{file}"):
os.remove(file_path)
st.success(f"❌ Arquivo '{file}' excluído.")
st.rerun()
# ======================== ADMIN ========================
elif pagina == "🔐 Admin":
st.title("🔐 Área de Administração")
senha_certa = "YWRtaW4xMjM=" # base64 de admin123
if "autenticado" not in st.session_state:
st.session_state.autenticado = False
if not st.session_state.autenticado:
senha = st.text_input("Senha:", type="password")
lembrar = st.checkbox("Lembrar nesta sessão")
login = st.button("Entrar")
if login:
if base64.b64encode(senha.encode()).decode() == senha_certa:
st.success("✅ Acesso liberado!")
st.session_state.autenticado = True
else:
st.error("❌ Senha incorreta.")
st.stop()
if not st.session_state.autenticado:
st.stop()
cat = st.selectbox("Categoria", ["Selecione..."] + CATEGORIAS)
if cat == "Selecione...":
st.warning("👈 Por favor, selecione uma categoria antes de enviar arquivos.")
st.stop()
tipo = st.radio("Tipo", ["Cortes", "Tutoriais", "Músicas"])
path = "assets/musicas" if tipo == "Músicas" else os.path.join(BASE_ASSETS, cat, tipo.lower())
arquivos = st.file_uploader("Enviar arquivos", accept_multiple_files=True, key=f"{cat}_{tipo}")
if arquivos:
with st.spinner("⏳ Processando e padronizando arquivos..."):
for f in arquivos:
extensao = os.path.splitext(f.name)[1].lower()
nome_aleatorio = f"{random.randint(1000000, 9999999)}{extensao}"
temp = f"temp_input_{nome_aleatorio}"
with open(temp, "wb") as out:
out.write(f.read())
final = os.path.join(path, nome_aleatorio)
if tipo != "Músicas":
subprocess.run([
"ffmpeg", "-i", temp,
"-vf", "scale=trunc(iw/2)*2:trunc(ih/2)*2,fps=30",
"-c:v", "libx264", "-preset", "ultrafast", "-crf", "18",
"-y", final
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
else:
shutil.move(temp, final)
os.remove(temp)
st.success("✅ Arquivos enviados, padronizados e renomeados com sucesso!")
st.write("Arquivos existentes:")
for arq in os.listdir(path):
col1, col2 = st.columns([5, 1])
with col1:
st.markdown(f"`{arq}`")
with col2:
if st.button("🗑", key=f"{tipo}_{cat}_{arq}"):
os.remove(os.path.join(path, arq))
st.rerun()