Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import streamlit as st
+from streamlit_tree_select import tree_select
 
 # Configuração do diretório de armazenamento
 STORAGE_DIR = "storage"
@@ -8,17 +9,34 @@ STORAGE_DIR = "storage"
 if not os.path.exists(STORAGE_DIR):
     os.makedirs(STORAGE_DIR)
 
-# Função para listar arquivos
-def list_files(directory):
+# Função para listar arquivos em formato de árvore
+def list_files_tree(directory):
     file_tree = {}
     for root, dirs, files in os.walk(directory):
-        for name in files:
-            folder_path = os.path.relpath(root, directory)
-            if folder_path not in file_tree:
-                file_tree[folder_path] = []
-            file_tree[folder_path].append(name)
+        folder_path = os.path.relpath(root, directory)
+        if folder_path not in file_tree:
+            file_tree[folder_path] = {'folders': [], 'files': []}
+        for d in dirs:
+            file_tree[folder_path]['folders'].append(d)
+        for f in files:
+            file_tree[folder_path]['files'].append(f)
     return file_tree
 
+# Função para gerar árvore de arquivos
+def generate_file_tree(directory):
+    tree_data = {}
+    file_tree = list_files_tree(directory)
+    for folder, contents in file_tree.items():
+        path_parts = folder.split(os.sep)
+        current_level = tree_data
+        for part in path_parts:
+            if part not in current_level:
+                current_level[part] = {}
+            current_level = current_level[part]
+        for f in contents['files']:
+            current_level[f] = None
+    return tree_data
+
 # Interface do Streamlit
 st.title("Gerenciamento de Armazenamento em Nuvem")
 
@@ -32,16 +50,15 @@ if uploaded_file is not None:
 
 # Exibir arquivos
 st.header("Arquivos no Armazenamento")
-files = list_files(STORAGE_DIR)
-for folder, files in files.items():
-    st.subheader(folder)
-    for file in files:
-        file_path = os.path.join(folder, file)
-        file_url = f"storage/{file_path}"
-        if file.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp')):
-            st.image(file_url, caption=file, use_column_width=True)
-        else:
-            st.markdown(f"[{file}]({file_url})")
+file_tree_data = generate_file_tree(STORAGE_DIR)
+selected_file = tree_select(file_tree_data)
+
+if selected_file:
+    file_url = os.path.join(STORAGE_DIR, selected_file)
+    if selected_file.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp')):
+        st.image(file_url, caption=selected_file, use_column_width=True)
+    else:
+        st.markdown(f"[{selected_file}]({file_url})")
 
 # Deletar arquivo
 st.header("Deletar Arquivo")