Update app.py

app.py

@@ -1,32 +1,22 @@
 import gradio as gr
 import os
 
-# Diretório onde as imagens serão salvas
-IMAGE_DIR = "imagens"
-
-# Cria o diretório se não existir
-if not os.path.exists(IMAGE_DIR):
-    os.makedirs(IMAGE_DIR)
+# Crie a pasta 'imagens' se ela não existir
+if not os.path.exists('imagens'):
+    os.makedirs('imagens')
 
 def save_image(image):
-    if image:
-        # Define o caminho completo para salvar a imagem
-        image_path = os.path.join(IMAGE_DIR, image.name)
-        # Salva a imagem no diretório especificado
-        with open(image_path, "wb") as f:
-            f.write(image.read())
-        return f"Imagem salva em: {image_path}"
-    else:
-        return "Nenhuma imagem enviada."
+    image_name = image.name.split('/')[-1]
+    image_path = os.path.join('imagens', image_name)
+    image.save(image_path)
+    return f"Imagem salva em: {image_path}"
 
-# Interface do Gradio
-iface = gr.Interface(
-    fn=save_image,
-    inputs=gr.File(label="Selecione uma imagem"),
-    outputs="text",
-    title="Upload de Imagem",
-    description="Envie uma imagem e ela será salva na pasta 'imagens'."
-)
+# Interface Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Envie uma imagem")
+    file_input = gr.File(label="Selecione uma imagem", type="file")
+    output = gr.Textbox()
+    submit_btn = gr.Button("Enviar")
+    submit_btn.click(save_image, inputs=file_input, outputs=output)
 
-if __name__ == "__main__":
-    iface.launch()
+demo.launch()