pcdoido2 commited on
Commit
d462dfc
·
verified ·
1 Parent(s): 3b2c327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -6,7 +6,7 @@ import tempfile
6
  import shutil
7
  import time
8
 
9
- from arquivos import salvar_no_gerenciador, CATEGORIES
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
- # Aqui você aplicaria os filtros e criaria o video_editado e video_final_raw
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 GERENCIADOR OU BAIXAR ===
135
  if salvar_no_gerenciador_checkbox and categoria:
136
- salvar_no_gerenciador(final_name, categoria)
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: