Update app.py

app.py

@@ -1,11 +1,6 @@
 # -*- coding: utf-8 -*-
 """
 AI Background Remover – Hugging Face Space
-Features
----------
-1. Upload an image and get a transparent‑background PNG.
-2. A “Save Image” button (base64 Data‑URI) appears right below the result.
-3. A ZIP file containing the PNG is created automatically for download.
 """
 
 import spaces
@@ -13,7 +8,6 @@ import gradio as gr
 from transparent_background import Remover
 from PIL import Image
 import numpy as np
-import io
 import base64
 import tempfile
 import zipfile
@@ -21,44 +15,29 @@ import zipfile
 # --------------------------------------------------------------------------- #
 # Core processing                                                             #
 # --------------------------------------------------------------------------- #
-@spaces.GPU  # Drop this decorator if you want to force CPU only
+@spaces.GPU  # Remove if you prefer CPU‑only execution
 def remove_background(image: Image.Image | np.ndarray):
-    """
-    Parameters
-    ----------
-    image : PIL.Image or np.ndarray
-        Input image from the Gradio uploader.
-
-    Returns
-    -------
-    png_path : str
-        Path to the processed PNG (for gr.Image).
-    download_link_html : str
-        An <a> tag with an embedded base64 data URI so users can save the PNG.
-    zip_path : str
-        Path to a ZIP file containing the PNG (for gr.File).
-    """
-    # 1. Background removal
     remover = Remover()
-    if isinstance(image, Image.Image):
-        output = remover.process(image)
-    elif isinstance(image, np.ndarray):
-        output = remover.process(Image.fromarray(image))
-    else:
+
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    elif not isinstance(image, Image.Image):
         raise TypeError("Unsupported image type")
 
-    # 2. Save PNG to a temporary file
+    # 1) Remove background
+    output = remover.process(image)
+
+    # 2) Save PNG
     png_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
     output.save(png_tmp, format="PNG")
     png_tmp.close()
     png_path = png_tmp.name
 
-    # 3. Build base64 download link
+    # 3) Build base64 download button
     with open(png_path, "rb") as f:
-        base64_image = base64.b64encode(f.read()).decode("utf-8")
-
+        b64 = base64.b64encode(f.read()).decode()
     download_link_html = f"""
-    <a href="data:application/octet-stream;base64,{base64_image}"
+    <a href="data:application/octet-stream;base64,{b64}"
        download="background_removed.png"
        style="
          display:inline-block;
@@ -74,56 +53,59 @@ def remove_background(image: Image.Image | np.ndarray):
     </a>
     """
 
-    # 4. Package PNG into a ZIP archive
+    # 4) Zip
     zip_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
     with zipfile.ZipFile(zip_tmp.name, "w", zipfile.ZIP_DEFLATED) as zf:
         zf.write(png_path, arcname="background_removed.png")
     zip_tmp.close()
-    zip_path = zip_tmp.name
 
-    return png_path, download_link_html, zip_path
+    return png_path, download_link_html, zip_tmp.name
 
 
 # --------------------------------------------------------------------------- #
-# Gradio UI                                                                   #
+# UI                                                                          #
 # --------------------------------------------------------------------------- #
 TITLE = "AI Background Remover – Automatically Remove Image Backgrounds"
-DESCRIPTION = """
+
+DESCRIPTION_MD = """
 Upload an image and our AI‑powered background remover will return a PNG with a transparent background.
 
 ### How to Use
 1. Click **Upload Image** or drag & drop a picture.  
 2. Press **Start Processing** and wait a moment.  
 3. The processed image appears on the right.  
-4. Click **📥 Save Image** to download the PNG, or download the ZIP below.
+4. Click **📥 Save Image** to download the PNG, or download the ZIP below.
 
 ### Application Scenarios
 - **E‑commerce** – Clean product photos with transparent backgrounds.  
-- **Graphic Design** – Quick cutouts for composites or layouts.  
+- **Graphic Design** – Quick cut‑outs for composites or layouts.  
 - **Social Media** – Avatars and icons that blend perfectly anywhere.
 
 Our tool leverages state‑of‑the‑art AI to remove backgrounds quickly and accurately—saving you time and effort.
 """
 
 with gr.Blocks(title=TITLE) as demo:
-    gr.Markdown(f"# {TITLE}\n{DESCRIPTION}")
+    gr.Markdown(f"# {TITLE}")
 
+    # -------- Accordion with instructions (default open) -------- #
+    with gr.Accordion("Instructions & Scenarios (click to collapse)", open=True):
+        gr.Markdown(DESCRIPTION_MD)
+
+    # -------- I/O widgets -------- #
     with gr.Row():
-        inp = gr.Image(label="Upload Image", type="numpy")
-        btn = gr.Button("Start Processing", variant="primary")
+        input_img = gr.Image(label="Upload Image", type="numpy")
+        process_btn = gr.Button("Start Processing", variant="primary")
 
-    out_img = gr.Image(label="Output Image (PNG)", type="filepath", height=350)
-    out_dl  = gr.HTML()
-    out_zip = gr.File(label="Download ZIP")
+    output_img = gr.Image(label="Output Image (PNG)", type="filepath", height=350)
+    download_html = gr.HTML()
+    zip_file = gr.File(label="Download ZIP")
 
-    btn.click(
+    process_btn.click(
         fn=remove_background,
-        inputs=inp,
-        outputs=[out_img, out_dl, out_zip],
+        inputs=input_img,
+        outputs=[output_img, download_html, zip_file]
     )
 
-# --------------------------------------------------------------------------- #
-# Entry point                                                                 #
 # --------------------------------------------------------------------------- #
 if __name__ == "__main__":
     demo.launch(show_error=True, show_api=False)