Update app.py

app.py

@@ -5,39 +5,56 @@ import io
 import uuid
 import os
 
+# Cartella per output
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
+# Funzione principale
 def remove_background(image: Image.Image):
     if image is None:
         return None, None
 
-    # Rimuove lo sfondo
     output = remove(image)
-
-    # Salva l'immagine in un file temporaneo
     file_id = str(uuid.uuid4())
     output_path = os.path.join(OUTPUT_DIR, f"{file_id}.png")
     output.save(output_path)
 
-    # Restituisce: immagine modificata, link file per download
     return output, output_path
 
-# Interfaccia Gradio
-demo = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(
-    type="pil",
-    label="Carica un'immagine o usa la webcam",
-    sources=["upload", "webcam"],
-    ),
-    outputs=[
-        gr.Image(type="pil", label="Immagine senza sfondo"),
-        gr.File(label="Scarica il file PNG")
-    ],
-    title="Rimozione Sfondo da Immagine",
-    description="Carica un'immagine, rimuovi lo sfondo e scarica il file PNG risultante."
+# Immagini della galleria (metti i file nella cartella "gallery")
+gallery_images = [f"gallery/{img}" for img in sorted(os.listdir("gallery")) if img.endswith((".png", ".jpg", ".jpeg"))][:5]
+
+# Funzione per usare immagine dalla galleria
+def load_gallery_image(image_path):
+    return Image.open(image_path)
+
+# Tema localizzato in italiano (solo per gradio >=4.15)
+italian_ui = gr.themes.Base(
+    language="it"
 )
 
+with gr.Blocks(theme=italian_ui, title="Rimozione Sfondo") as demo:
+    gr.Markdown("# ✂️ Rimozione Sfondo da Immagine")
+    gr.Markdown("Carica una tua immagine, usane una dalla webcam oppure scegli dalla galleria.")
+
+    with gr.Row():
+        image_input = gr.Image(
+            type="pil",
+            label="Carica immagine",
+            sources=["upload", "webcam"],
+        )
+        gallery = gr.Gallery(label="Galleria di esempio").style(grid=5, height="auto")
+        gallery.value = gallery_images
+
+    output_image = gr.Image(type="pil", label="Risultato")
+    output_file = gr.File(label="Scarica PNG")
+
+    # Quando si seleziona un'immagine dalla galleria
+    gallery.select(fn=load_gallery_image, outputs=image_input)
+
+    # Bottone per elaborare
+    btn = gr.Button("Rimuovi Sfondo")
+    btn.click(fn=remove_background, inputs=image_input, outputs=[output_image, output_file])
+
 if __name__ == "__main__":
     demo.launch()