import gradio as gr
from PIL import Image
from inference import single_image_wrapped

# 调用 single_image_wrapped 函数的包装函数
def image_with_prompt_to_result(image, prompt):
    # 将图片传递给 single_image_wrapped 函数，得到输出结果
    result = single_image_wrapped(image, prompt)
    return result

# 创建 Gradio 接口
iface = gr.Interface(
    fn=image_with_prompt_to_result,  # 调用的函数
    inputs=[gr.Image(type="pil"), gr.Textbox(placeholder="输入提示词...")],  # 输入：图片 + prompt
    outputs="text",  # 输出：识别结果（文本）
    title="图片与提示词输入",  # 界面标题
    description="上传一张图片并输入提示词，获取识别结果",  # 描述
)

# 启动应用
iface.launch()