import gradio as gr
from PIL import Image
from infer import infer

MODEL = None
LOCAL_REPO_PATH = None

def process_image(image):
    """调用 infer() 进行处理并返回结果"""
    global MODEL, LOCAL_REPO_PATH
    ret, MODEL, LOCAL_REPO_PATH= infer(image, MODEL, LOCAL_REPO_PATH)
    return ret

iface = gr.Interface(
    fn=process_image,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="numpy"),
    title="RGB手部姿态估计（手部mesh重建正在建设中……）",
    description="上传或拍摄一张右手单手的RGB图片，输出手部姿态估计结果。"
)

if __name__ == "__main__":
    iface.launch(debug=True)