import gradio as gr
from rembg import remove
from PIL import Image
import io, base64

def remove_bg(input_image):
    if input_image is None:
        return None

    # Background remove
    output = remove(input_image)

    # Convert to Base64
    buffer = io.BytesIO()
    output.save(buffer, format="PNG")
    b64 = base64.b64encode(buffer.getvalue()).decode()
    return "data:image/png;base64," + b64

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            inp = gr.Image(type="pil", label="Upload Image")
            btn = gr.Button("Remove Background")
        with gr.Column():
            out = gr.Textbox(label="Base64 Encoded PNG")

    btn.click(remove_bg, inputs=inp, outputs=out)

# 👇 ye line jaruri hai taaki API bhi kaam kare
demo.queue(api_open=True).launch()