Update app.py

app.py

@@ -5,12 +5,12 @@ import re
 from PIL import Image
 
 # Diretório onde os arquivos serão armazenados
-STORAGE_DIR = "/home/user/storage"
+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')
+SPACE_NAME = os.getenv('SPACE_NAME', '13ze/cloud-storage')
 
 # Função para inicializar o diretório de armazenamento
 def initialize_storage():
@@ -66,7 +66,7 @@ def upload_file(file):
 # 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):
+    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."