Update app.py

app.py

@@ -158,3 +158,41 @@ for categoria in CATEGORIES:
                         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)