Update app.py

app.py

@@ -2,15 +2,11 @@ import streamlit as st
 import os
 import json
 import time
-from huggingface_hub import list_repo_files, hf_hub_url
 
 BASE_FOLDER = "uploaded_files"
 CATEGORIES = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE"]
 EXPIRATION_FILE = "expirations.json"
 
-# Dataset onde o vídeo generator salva os vídeos
-DATASET_REPO = "pcdoido2/video-gerados"
-
 # --- Estilos CSS ---
 st.markdown("""
     <style>
@@ -69,6 +65,7 @@ def remove_expired_files():
 
 # --- Apagar arquivos vencidos ao iniciar ---
 remove_expired_files()
+
 # --- Upload ---
 
 st.header("📤 Upload de Arquivos")
@@ -102,14 +99,14 @@ if categoria:
         st.success("Arquivos enviados com sucesso!")
         st.rerun()
 
-# --- Lista de Arquivos locais agrupada por pasta ---
-st.header("📄 Arquivos Locais Disponíveis")
+# --- Lista de Arquivos agrupada por pasta ---
+st.header("📄 Arquivos Disponíveis")
 
 for categoria in CATEGORIES:
     folder = os.path.join(BASE_FOLDER, categoria)
     files = os.listdir(folder)
 
-    st.subheader(f"📁 {categoria} (Local)")
+    st.subheader(f"📁 {categoria}")
 
     if not files:
         st.info("Nenhum arquivo na categoria.")
@@ -157,42 +154,4 @@ for categoria in CATEGORIES:
                         st.success(f"Arquivo '{file}' baixado e removido.")
                         st.rerun()
 
-            st.markdown("</div>", unsafe_allow_html=True)
-# --- Função para listar arquivos do Dataset ---
-def listar_arquivos_dataset(categoria):
-    """
-    Lista arquivos na categoria escolhida do Dataset.
-    """
-    arquivos = []
-    try:
-        todos_arquivos = list_repo_files(DATASET_REPO, repo_type="dataset")
-        prefixo = f"{categoria}/"
-
-        for arquivo in todos_arquivos:
-            if arquivo.startswith(prefixo):
-                arquivos.append(arquivo[len(prefixo):])  # Remove o prefixo da pasta
-    except Exception as e:
-        st.error(f"Erro ao listar arquivos do Dataset: {e}")
-    return arquivos
-
-# --- Lista de Arquivos do Dataset ---
-st.header("🌐 Vídeos Gerados (Dataset)")
-
-for categoria in CATEGORIES:
-    arquivos_dataset = listar_arquivos_dataset(categoria)
-
-    st.subheader(f"📁 {categoria} (Dataset)")
-
-    if not arquivos_dataset:
-        st.info("Nenhum arquivo nesta categoria no Dataset.")
-    else:
-        for arquivo in arquivos_dataset:
-            url = hf_hub_url(
-                repo_id=DATASET_REPO,
-                filename=f"{categoria}/{arquivo}",
-                repo_type="dataset"
-            )
-            st.markdown(f"<div class='file-box'>", unsafe_allow_html=True)
-            st.markdown(f"<div class='file-name'>{arquivo}</div>", unsafe_allow_html=True)
-            st.markdown(f"[⬇ Download]({url})", unsafe_allow_html=True)
-            st.markdown("</div>", unsafe_allow_html=True)
+            st.markdown("</div>", unsafe_allow_html=True)