# app.py
import gradio as gr
from handler import EndpointHandler

# 初始化你的 ONNX 模型
handler = EndpointHandler(path=".")

def api_call(image, parameters):
    data = {
        "inputs": image,      # PIL.Image
        "parameters": parameters or {}
    }
    result = handler(data)[0]
    return result  # JSON only（含 base64 mask）

demo = gr.Interface(
    fn=api_call,
    inputs=[
        gr.Image(type="pil", label="image"),
        gr.JSON(label="parameters (point_coords, point_labels, return_mask_image)")
    ],
    outputs=gr.JSON(label="result"),
    title="Edge SAM API",
    description="Pure API Space — send image + parameters, get mask JSON."
)

if __name__ == "__main__":
    demo.launch()