Update app.py

app.py

@@ -1,92 +1,67 @@
-import gradio as gr
 import os
 import shutil
-import re
-from PIL import Image
+import gradio as gr
 
-# Diretório onde os arquivos serão armazenados
+# Configuração do diretório de armazenamento
 STORAGE_DIR = "storage"
-ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'webp'}
-MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
-
-# Nome do espaço HuggingFace
-SPACE_NAME = os.getenv('SPACE_NAME', '13ze/cloud-storage')
-
-# Função para inicializar o diretório de armazenamento
-def initialize_storage():
-    if not os.path.exists(STORAGE_DIR):
-        os.makedirs(STORAGE_DIR)
-
-# Função para verificar o tipo de arquivo
-def allowed_file(filename):
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
-
-# Função para normalizar o nome do arquivo
-def secure_filename(filename):
-    filename = re.sub(r'[^a-zA-Z0-9_.-]', '_', filename)
-    return filename
 
-# Função para verificar o conteúdo do arquivo
-def is_image(file_path):
-    try:
-        with Image.open(file_path) as img:
-            img.verify()
-        return True
-    except (IOError, SyntaxError) as e:
-        return False
+# Criar o diretório de armazenamento se não existir
+if not os.path.exists(STORAGE_DIR):
+    os.makedirs(STORAGE_DIR)
 
-# Função para fazer upload do arquivo
+# Função para upload de arquivos
 def upload_file(file):
-    if file:
-        # Normalizar nome do arquivo
-        filename = secure_filename(file.name)
+    # Salvar o arquivo no diretório de armazenamento
+    file_path = os.path.join(STORAGE_DIR, file.name)
+    with open(file_path, "wb") as f:
+        f.write(file.read())
+    return "Arquivo enviado com sucesso!"
+
+# Função para listar arquivos
+def list_files():
+    file_tree = {}
+    for root, dirs, files in os.walk(STORAGE_DIR):
+        for name in files:
+            folder_path = os.path.relpath(root, STORAGE_DIR)
+            if folder_path not in file_tree:
+                file_tree[folder_path] = []
+            file_tree[folder_path].append(name)
+    return file_tree
+
+# Função para deletar arquivos
+def delete_file(file_path):
+    abs_path = os.path.join(STORAGE_DIR, file_path)
+    if os.path.isfile(abs_path):
+        os.remove(abs_path)
+        return f"Arquivo {file_path} deletado com sucesso!"
+    else:
+        return f"Arquivo {file_path} não encontrado!"
 
-        # Verificar extensão do arquivo
-        if not allowed_file(filename):
-            return "Tipo de arquivo não permitido."
-
-        # Verificar tamanho do arquivo
-        if os.path.getsize(file.name) > MAX_FILE_SIZE:
-            return "Arquivo excede o tamanho máximo permitido de 10MB."
-
-        # Salvar arquivo
-        file_path = os.path.join(STORAGE_DIR, filename)
-        shutil.copyfile(file.name, file_path)
-
-        # Verificar conteúdo do arquivo
-        if not is_image(file_path):
-            os.remove(file_path)
-            return "O conteúdo do arquivo não é uma imagem válida."
+# Interface Gradio
+def file_explorer_ui():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Gerenciamento de Armazenamento em Nuvem")
 
-        base_url = f"https://{SPACE_NAME}.hf.space/files/{filename}"
-        delete_url = f"https://{SPACE_NAME}.hf.space/delete?file={filename}"
-        return f"Arquivo disponível em: {base_url}\nLink para deletar: {delete_url}"
-    return "Falha no upload do arquivo."
+        with gr.Row():
+            with gr.Column():
+                upload = gr.File(label="Enviar Arquivo")
+                upload_btn = gr.Button("Enviar")
+                upload_output = gr.Textbox()
 
-# Função para deletar o arquivo
-def delete_file(file_name):
-    file_path = os.path.join(STORAGE_DIR, file_name)
-    if os.path.isfile(file_path):
-        os.remove(file_path)
-        return f"Arquivo {file_name} deletado com sucesso."
-    return f"Arquivo {file_name} não encontrado."
+            with gr.Column():
+                delete_file_input = gr.Textbox(label="Caminho do arquivo para deletar")
+                delete_file_btn = gr.Button("Deletar")
+                delete_output = gr.Textbox()
 
-# Inicializar o diretório de armazenamento
-initialize_storage()
+        with gr.Row():
+            list_files_btn = gr.Button("Listar Arquivos")
+            file_list_output = gr.JSON()
 
-# Interface Gradio
-with gr.Blocks() as app:
-    with gr.Row():
-        gr.Markdown("## Gerenciamento de Arquivos na Nuvem")
-    with gr.Row():
-        upload = gr.File(label="Selecione um arquivo para upload")
-        output = gr.Textbox(label="Status")
-    with gr.Row():
-        delete_file_name = gr.Textbox(label="Nome do arquivo para deletar")
-        delete_button = gr.Button("Deletar arquivo")
-        delete_output = gr.Textbox(label="Status de deleção")
+        upload_btn.click(upload_file, inputs=upload, outputs=upload_output)
+        delete_file_btn.click(delete_file, inputs=delete_file_input, outputs=delete_output)
+        list_files_btn.click(list_files, outputs=file_list_output)
 
-    upload.upload(upload_file, inputs=[upload], outputs=[output])
-    delete_button.click(delete_file, inputs=[delete_file_name], outputs=[delete_output])
+    return demo
 
-app.launch()
+# Rodar a aplicação
+file_explorer_ui().launch()