Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import tempfile
|
|
6 |
import shutil
|
7 |
import time
|
8 |
|
9 |
-
from arquivos import
|
10 |
|
11 |
st.set_page_config(page_title="TikTok Video Generator", layout="centered")
|
12 |
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)
|
|
39 |
velocidade_final = st.slider("Velocidade final", 0.5, 2.0, 1.0, 0.1)
|
40 |
crf_value = st.slider("Qualidade CRF", 18, 30, 23)
|
41 |
|
42 |
-
# === NOVA OPÇÃO SALVAR NO GERENCIADOR ===
|
43 |
st.markdown("### Gerenciamento do vídeo final")
|
44 |
-
salvar_no_gerenciador_checkbox = st.checkbox("Salvar vídeo no Gerenciador de Arquivos")
|
45 |
if salvar_no_gerenciador_checkbox:
|
46 |
categoria = st.selectbox("Selecione a categoria:", CATEGORIES)
|
47 |
else:
|
@@ -124,17 +124,17 @@ if st.button("Gerar Vídeo(s)"):
|
|
124 |
"-c:v", "libx264", "-preset", "ultrafast", "-crf", "30", video_raw
|
125 |
], check=True, stderr=subprocess.PIPE)
|
126 |
|
127 |
-
#
|
128 |
-
# Para não deixar o código gigante, vou direto para o final_name
|
129 |
-
|
130 |
final_name = f"video_final_{n}_{int(time.time())}.mp4"
|
131 |
-
# Aqui vamos simular que o vídeo final já foi gerado:
|
132 |
shutil.copy(video_raw, final_name)
|
133 |
|
134 |
-
# === SALVAR NO
|
135 |
if salvar_no_gerenciador_checkbox and categoria:
|
136 |
-
|
137 |
-
st.success(f"✅ Vídeo {n+1} salvo na categoria '{categoria}'.")
|
|
|
|
|
|
|
138 |
else:
|
139 |
st.video(final_name)
|
140 |
with open(final_name, "rb") as f:
|
|
|
6 |
import shutil
|
7 |
import time
|
8 |
|
9 |
+
from arquivos import salvar_no_dataset, CATEGORIES
|
10 |
|
11 |
st.set_page_config(page_title="TikTok Video Generator", layout="centered")
|
12 |
st.title("🎥 TikTok Video Generator - PRO")
|
|
|
39 |
velocidade_final = st.slider("Velocidade final", 0.5, 2.0, 1.0, 0.1)
|
40 |
crf_value = st.slider("Qualidade CRF", 18, 30, 23)
|
41 |
|
42 |
+
# === NOVA OPÇÃO SALVAR NO GERENCIADOR/DATASET ===
|
43 |
st.markdown("### Gerenciamento do vídeo final")
|
44 |
+
salvar_no_gerenciador_checkbox = st.checkbox("Salvar vídeo no Gerenciador de Arquivos (Dataset)")
|
45 |
if salvar_no_gerenciador_checkbox:
|
46 |
categoria = st.selectbox("Selecione a categoria:", CATEGORIES)
|
47 |
else:
|
|
|
124 |
"-c:v", "libx264", "-preset", "ultrafast", "-crf", "30", video_raw
|
125 |
], check=True, stderr=subprocess.PIPE)
|
126 |
|
127 |
+
# Para simplificar: vamos usar o video_raw como vídeo final neste exemplo
|
|
|
|
|
128 |
final_name = f"video_final_{n}_{int(time.time())}.mp4"
|
|
|
129 |
shutil.copy(video_raw, final_name)
|
130 |
|
131 |
+
# === SALVAR NO DATASET OU DOWNLOAD ===
|
132 |
if salvar_no_gerenciador_checkbox and categoria:
|
133 |
+
salvar_no_dataset(final_name, categoria)
|
134 |
+
st.success(f"✅ Vídeo {n+1} salvo no Dataset na categoria '{categoria}'.")
|
135 |
+
st.markdown(
|
136 |
+
f"[🔗 Acessar no Dataset](https://huggingface.co/datasets/pcdoido2/video-gerados/resolve/main/{categoria}/{final_name})"
|
137 |
+
)
|
138 |
else:
|
139 |
st.video(final_name)
|
140 |
with open(final_name, "rb") as f:
|