Update app.py

app.py

@@ -6,54 +6,92 @@ import tempfile
 import shutil
 import time
 
+# ================= NOVO: categorias e pastas de cortes salvos =================
+CATEGORIAS = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE", "FC MOBILE"]
+BASE_ASSETS = "assets"
+for cat in CATEGORIAS:
+    os.makedirs(os.path.join(BASE_ASSETS, cat, "cortes"), exist_ok=True)
+# ============================================================================
+
 st.set_page_config(page_title="Shorts Generator", layout="centered")
 st.title("🎥 Shorts Generator - Simples")
 
 st.markdown("Envie seus vídeos e gere conteúdo com efeitos e controle total de duração!")
 
-# Upload dos vídeos
+# ================= NOVO: seleção de categoria para usar cortes salvos =========
+categoria_selecionada = st.selectbox(
+    "Escolha a categoria (opcional, para usar cortes já salvos):",
+    ["Selecione..."] + CATEGORIAS
+)
+
+cortes_salvos_paths = []
+if categoria_selecionada != "Selecione...":
+    path_cortes_salvos = os.path.join(BASE_ASSETS, categoria_selecionada, "cortes")
+    if os.path.isdir(path_cortes_salvos):
+        cortes_salvos_paths = [
+            os.path.join(path_cortes_salvos, f)
+            for f in os.listdir(path_cortes_salvos)
+            if f.lower().endswith(".mp4")
+        ]
+    st.caption(f"Encontrados {len(cortes_salvos_paths)} cortes salvos em: {path_cortes_salvos}")
+# ============================================================================
+
+# Upload dos vídeos (MANTIDO)
 cortes = st.file_uploader("Envie os vídeos de cortes", type=["mp4"], accept_multiple_files=True)
 
-# Configurações
+# Configurações (MANTIDO)
 num_videos_finais = st.number_input("Quantos vídeos finais gerar?", min_value=1, max_value=10, value=1)
 duracao_final = st.number_input("Duração final do vídeo (em segundos)", min_value=10, max_value=300, value=30)
 
-# Min e Max duração dos cortes
+# Min e Max duração dos cortes (MANTIDO)
 duracao_min = st.slider("Duração mínima de cada corte (s)", 1, 30, 3)
 duracao_max = st.slider("Duração máxima de cada corte (s)", 1, 60, 5)
 if duracao_min > duracao_max:
     st.warning("⚠️ A duração mínima não pode ser maior que a máxima.")
 
-# Corte lateral e zoom
+# Corte lateral e zoom (MANTIDO)
 corte_lateral = st.slider("Corte lateral (%)", 0, 50, 0, 5)
 zoom = st.slider("Zoom Central (1.0 = normal)", 1.0, 2.0, 1.0, 0.1)
 
-# Velocidades e qualidade
+# Velocidades e qualidade (MANTIDO)
 velocidade_cortes = st.slider("Velocidade dos cortes", 0.5, 2.0, 1.0, 0.1)
 velocidade_final = st.slider("Velocidade final do vídeo", 0.5, 2.0, 1.0, 0.1)
 crf_value = st.slider("Qualidade CRF (menor = melhor qualidade)", 18, 30, 18)
 
-# Outros
+# Outros (MANTIDO)
 st.write("### Outros")
 ativar_espelhar = st.checkbox("Espelhar Vídeo", value=True)
 ativar_filtro_cor = st.checkbox("Filtro de Cor (Contraste/Saturação)", value=True)
-
-# Botão principal
+# Botão principal (MANTIDO, mas agora combinando uploads + salvos)
 if st.button("Gerar Vídeo(s)"):
-    if not cortes:
-        st.error("❌ Envie os vídeos para continuar.")
+    # Verifica se temos pelo menos upload OU salvos
+    if not cortes and len(cortes_salvos_paths) == 0:
+        st.error("❌ Envie vídeos OU selecione uma categoria com cortes salvos para continuar.")
     else:
         with st.spinner('🎥 Seu vídeo está sendo gerado...'):
             progresso = st.progress(0)
             temp_dir = tempfile.mkdtemp()
 
             try:
+                # === PREPARAR LISTA FINAL DE ARQUIVOS DE CORTE ===
                 cortes_names = []
-                for idx, corte in enumerate(cortes):
-                    nome = os.path.join(temp_dir, f"corte_{idx}_{random.randint(1000,9999)}.mp4")
-                    with open(nome, "wb") as f:
-                        f.write(corte.read())
-                    cortes_names.append(nome)
+
+                # 1) Arquivos enviados (MANTIDO)
+                if cortes:
+                    for idx, corte in enumerate(cortes):
+                        nome = os.path.join(temp_dir, f"corte_{idx}_{random.randint(1000,9999)}.mp4")
+                        with open(nome, "wb") as f:
+                            f.write(corte.read())
+                        cortes_names.append(nome)
+
+                # 2) Arquivos salvos na categoria (NOVO – usamos os caminhos diretamente)
+                if len(cortes_salvos_paths) > 0:
+                    cortes_names.extend(cortes_salvos_paths)
+
+                if len(cortes_names) == 0:
+                    st.error("❌ Não há cortes disponíveis após combinar uploads e salvos.")
+                    shutil.rmtree(temp_dir)
+                    st.stop()
 
                 progresso.progress(10)