Update app.py

app.py

@@ -9,6 +9,20 @@ STORAGE_DIR = "/home/user/storage"
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'webp'}
 MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
 
+# Função para inicializar o diretório import gradio as gr
+import os
+import shutil
+import re
+from PIL import Image
+
+# Diretório onde os arquivos serão armazenados
+STORAGE_DIR = "/home/user/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')
+
 # Função para inicializar o diretório de armazenamento
 def initialize_storage():
     if not os.path.exists(STORAGE_DIR):
@@ -32,6 +46,84 @@ def is_image(file_path):
     except (IOError, SyntaxError) as e:
         return False
 
+# Função para fazer upload do arquivo
+def upload_file(file):
+    if file:
+        # Normalizar nome do arquivo
+        filename = secure_filename(file.name)
+
+        # 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."
+
+        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."
+
+# Função para deletar o arquivo
+def delete_file(file_name):
+    file_path = os.path.join(STORAGE_DIR, file_name)
+    if os.path.exists(file_path):
+        os.remove(file_path)
+        return f"Arquivo {file_name} deletado com sucesso."
+    return f"Arquivo {file_name} não encontrado."
+
+# Inicializar o diretório de armazenamento
+initialize_storage()
+
+# 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.upload(upload_file, inputs=[upload], outputs=[output])
+    delete_button.click(delete_file, inputs=[delete_file_name], outputs=[delete_output])
+
+app.launch()
+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
+
 # Função para fazer upload do arquivo
 def upload_file(file):
     if file: