test

app.py

@@ -13,6 +13,10 @@ from dotenv import load_dotenv
 # Carica le variabili di ambiente dal file .env
 load_dotenv()
 
+output_folder = 'output_images'
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
 def numpy_to_pil(image):
     """Convert a numpy array to a PIL Image."""
     if image.dtype == np.uint8:  # Most common case
@@ -34,7 +38,10 @@ def process_image(image):
     result = response.json()
     processed_image_b64 = result["processed_image"]
     processed_image = Image.open(BytesIO(base64.b64decode(processed_image_b64)))
-    return [image, processed_image]  # Return the original and processed images
+    #return [image, processed_image]  # Return the original and processed image
+    image_path = os.path.join(output_folder, "no_bg_image.png")
+    processed_image.save(image_path)
+    return (processed_image, image), image_path
 
 
 # Carica l'esempio di immagine
@@ -45,7 +52,7 @@ output_slider = ImageSlider(label="Processed photo", type="pil")
 demo = gr.Interface(
     fn=process_image,
     inputs=image,
-    outputs=output_slider,
+    outputs=[output_slider, gr.File(label="output png file")],
     title="Magic Eraser",
     examples=[["elephant.jpg"]]  # Esempio locale
 )