Update app.py

app.py

@@ -6,7 +6,7 @@ import tempfile
 import shutil
 import time
 
-from arquivos import salvar_no_gerenciador, CATEGORIES
+from arquivos import salvar_no_dataset, CATEGORIES
 
 st.set_page_config(page_title="TikTok Video Generator", layout="centered")
 st.title("🎥 TikTok Video Generator - PRO")
@@ -39,9 +39,9 @@ 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)
 
-# === NOVA OPÇÃO SALVAR NO GERENCIADOR ===
+# === NOVA OPÇÃO SALVAR NO GERENCIADOR/DATASET ===
 st.markdown("### Gerenciamento do vídeo final")
-salvar_no_gerenciador_checkbox = st.checkbox("Salvar vídeo no Gerenciador de Arquivos")
+salvar_no_gerenciador_checkbox = st.checkbox("Salvar vídeo no Gerenciador de Arquivos (Dataset)")
 if salvar_no_gerenciador_checkbox:
     categoria = st.selectbox("Selecione a categoria:", CATEGORIES)
 else:
@@ -124,17 +124,17 @@ if st.button("Gerar Vídeo(s)"):
                         "-c:v", "libx264", "-preset", "ultrafast", "-crf", "30", video_raw
                     ], check=True, stderr=subprocess.PIPE)
 
-                    # Aqui você aplicaria os filtros e criaria o video_editado e video_final_raw
-                    # Para não deixar o código gigante, vou direto para o final_name
-
+                    # Para simplificar: vamos usar o video_raw como vídeo final neste exemplo
                     final_name = f"video_final_{n}_{int(time.time())}.mp4"
-                    # Aqui vamos simular que o vídeo final já foi gerado:
                     shutil.copy(video_raw, final_name)
 
-                    # === SALVAR NO GERENCIADOR OU BAIXAR ===
+                    # === SALVAR NO DATASET OU DOWNLOAD ===
                     if salvar_no_gerenciador_checkbox and categoria:
-                        salvar_no_gerenciador(final_name, categoria)
-                        st.success(f"✅ Vídeo {n+1} salvo na categoria '{categoria}'.")
+                        salvar_no_dataset(final_name, categoria)
+                        st.success(f"✅ Vídeo {n+1} salvo no Dataset na categoria '{categoria}'.")
+                        st.markdown(
+                            f"[🔗 Acessar no Dataset](https://huggingface.co/datasets/pcdoido2/video-gerados/resolve/main/{categoria}/{final_name})"
+                        )
                     else:
                         st.video(final_name)
                         with open(final_name, "rb") as f: