Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import os
-import shutil
 import json
 import time
 
@@ -49,7 +48,7 @@ remove_expired_files()
 # --- Upload ---
 st.header("📤 Upload de Arquivos")
 
-categoria = st.selectbox("Selecione a categoria", CATEGORIES)
+categoria = st.selectbox("Selecione a categoria para upload", CATEGORIES)
 uploaded_files = st.file_uploader("Selecione arquivos", accept_multiple_files=True)
 auto_delete = st.checkbox("Excluir automaticamente após 24 horas")
 
@@ -70,51 +69,53 @@ if uploaded_files:
     st.success("Arquivos enviados com sucesso!")
     st.rerun()
 
-# --- Lista de Arquivos ---
-st.header(f"📄 Arquivos em {categoria}")
+# --- Lista de Arquivos agrupada por pasta ---
+st.header("📄 Arquivos Disponíveis")
 
-folder = os.path.join(BASE_FOLDER, categoria)
-files = os.listdir(folder)
+for categoria in CATEGORIES:
+    folder = os.path.join(BASE_FOLDER, categoria)
+    files = os.listdir(folder)
 
-if not files:
-    st.info("Nenhum arquivo disponível.")
-else:
-    for file in files:
-        st.subheader(f"📄 {file}")
-
-        key = f"{categoria}|||{file}"
-        expira = expirations.get(key)
-
-        if expira:
-            restante = int(expira - time.time())
-            if restante > 0:
-                restante_horas = restante // 3600
-                restante_min = (restante % 3600) // 60
-                st.write(f"⏳ Expira em {restante_horas}h {restante_min}min")
-            else:
-                st.write("⚠ Expiração iminente")
-
-        col1, col2, col3 = st.columns(3)
-
-        with col1:
-            with open(os.path.join(folder, file), "rb") as f_obj:
-                st.download_button("⬇ Download", f_obj, file_name=file, key=f"down_{categoria}_{file}")
-
-        with col2:
-            if st.button("🗑 Excluir", key=f"delete_{categoria}_{file}"):
-                os.remove(os.path.join(folder, file))
-                expirations.pop(key, None)
-                with open(EXPIRATION_FILE, "w") as f:
-                    json.dump(expirations, f)
-                st.success(f"Arquivo '{file}' excluído.")
-                st.rerun()
-
-        with col3:
-            with open(os.path.join(folder, file), "rb") as f_obj:
-                if st.download_button("⬇ Baixar & Apagar", f_obj, file_name=file, key=f"download_delete_{categoria}_{file}"):
+    st.subheader(f"📁 {categoria}")
+
+    if not files:
+        st.info("Nenhum arquivo na categoria.")
+    else:
+        for file in files:
+            st.write(f"**{file}**")
+            key = f"{categoria}|||{file}"
+
+            expira = expirations.get(key)
+            if expira:
+                restante = int(expira - time.time())
+                if restante > 0:
+                    restante_horas = restante // 3600
+                    restante_min = (restante % 3600) // 60
+                    st.caption(f"⏳ Expira em {restante_horas}h {restante_min}min")
+                else:
+                    st.caption("⚠ Expiração iminente")
+
+            col1, col2, col3 = st.columns(3)
+
+            with col1:
+                with open(os.path.join(folder, file), "rb") as f_obj:
+                    st.download_button("⬇ Download", f_obj, file_name=file, key=f"down_{categoria}_{file}")
+
+            with col2:
+                if st.button("🗑 Excluir", key=f"delete_{categoria}_{file}"):
                     os.remove(os.path.join(folder, file))
                     expirations.pop(key, None)
                     with open(EXPIRATION_FILE, "w") as f:
                         json.dump(expirations, f)
-                    st.success(f"Arquivo '{file}' baixado e removido.")
+                    st.success(f"Arquivo '{file}' excluído.")
                     st.rerun()
+
+            with col3:
+                with open(os.path.join(folder, file), "rb") as f_obj:
+                    if st.download_button("⬇ Baixar & Apagar", f_obj, file_name=file, key=f"download_delete_{categoria}_{file}"):
+                        os.remove(os.path.join(folder, file))
+                        expirations.pop(key, None)
+                        with open(EXPIRATION_FILE, "w") as f:
+                            json.dump(expirations, f)
+                        st.success(f"Arquivo '{file}' baixado e removido.")
+                        st.rerun()