Update app.py

app.py

@@ -4,10 +4,13 @@ import shutil
 import json
 import time
 
-UPLOAD_FOLDER = "uploaded_files"
+BASE_FOLDER = "uploaded_files"
+CATEGORIES = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE"]
 EXPIRATION_FILE = "expirations.json"
 
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+# Cria as pastas se não existirem
+for cat in CATEGORIES:
+    os.makedirs(os.path.join(BASE_FOLDER, cat), exist_ok=True)
 
 # Carrega dados de expiração
 if os.path.exists(EXPIRATION_FILE):
@@ -16,7 +19,7 @@ if os.path.exists(EXPIRATION_FILE):
 else:
     expirations = {}
 
-st.title("📂 File Manager Simples")
+st.title("📂 File Manager por Categoria")
 
 # --- Função: apagar arquivos expirados ---
 def remove_expired_files():
@@ -24,16 +27,17 @@ def remove_expired_files():
     now = time.time()
     expired_files = []
 
-    for file, expire_time in list(expirations.items()):
+    for file_full, expire_time in list(expirations.items()):
+        cat, file = file_full.split("|||")
+        file_path = os.path.join(BASE_FOLDER, cat, file)
         if now > expire_time:
-            file_path = os.path.join(UPLOAD_FOLDER, file)
             if os.path.exists(file_path):
                 os.remove(file_path)
-                expired_files.append(file)
+                expired_files.append(file_full)
                 changed = True
 
-    for file in expired_files:
-        expirations.pop(file)
+    for file_full in expired_files:
+        expirations.pop(file_full)
 
     if changed:
         with open(EXPIRATION_FILE, "w") as f:
@@ -44,18 +48,22 @@ remove_expired_files()
 
 # --- Upload ---
 st.header("📤 Upload de Arquivos")
+
+categoria = st.selectbox("Selecione a categoria", CATEGORIES)
 uploaded_files = st.file_uploader("Selecione arquivos", accept_multiple_files=True)
 auto_delete = st.checkbox("Excluir automaticamente após 24 horas")
 
 if uploaded_files:
     for uploaded_file in uploaded_files:
-        file_path = os.path.join(UPLOAD_FOLDER, uploaded_file.name)
+        folder = os.path.join(BASE_FOLDER, categoria)
+        file_path = os.path.join(folder, uploaded_file.name)
         with open(file_path, "wb") as f:
             f.write(uploaded_file.read())
 
-        # Se marcado, registra expiração para 24h depois
+        # Salvar expiração se marcada
         if auto_delete:
-            expirations[uploaded_file.name] = time.time() + 24 * 60 * 60  # 24h em segundos
+            key = f"{categoria}|||{uploaded_file.name}"
+            expirations[key] = time.time() + 24 * 60 * 60
             with open(EXPIRATION_FILE, "w") as f:
                 json.dump(expirations, f)
 
@@ -63,9 +71,10 @@ if uploaded_files:
     st.rerun()
 
 # --- Lista de Arquivos ---
-st.header("📄 Arquivos Disponíveis")
+st.header(f"📄 Arquivos em {categoria}")
 
-files = os.listdir(UPLOAD_FOLDER)
+folder = os.path.join(BASE_FOLDER, categoria)
+files = os.listdir(folder)
 
 if not files:
     st.info("Nenhum arquivo disponível.")
@@ -73,7 +82,9 @@ else:
     for file in files:
         st.subheader(f"📄 {file}")
 
-        expira = expirations.get(file)
+        key = f"{categoria}|||{file}"
+        expira = expirations.get(key)
+
         if expira:
             restante = int(expira - time.time())
             if restante > 0:
@@ -83,27 +94,26 @@ else:
             else:
                 st.write("⚠ Expiração iminente")
 
-        # Botões abaixo do nome do arquivo
         col1, col2, col3 = st.columns(3)
 
         with col1:
-            with open(os.path.join(UPLOAD_FOLDER, file), "rb") as f_obj:
-                st.download_button("⬇ Download", f_obj, file_name=file, key=f"down_{file}")
+            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_{file}"):
-                os.remove(os.path.join(UPLOAD_FOLDER, file))
-                expirations.pop(file, None)
+            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(UPLOAD_FOLDER, file), "rb") as f_obj:
-                if st.download_button("⬇ Baixar & Apagar", f_obj, file_name=file, key=f"download_delete_{file}"):
-                    os.remove(os.path.join(UPLOAD_FOLDER, file))
-                    expirations.pop(file, None)
+            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.")