import gradio as gr
import testMch as func


def work(imgurl):
    # output = testMch.describeImage2(img)
    # # output = testMch.agent(img)
    # # output = testMch.agent(img)
    # return output
    # imgurl = "r" + imgurl
    # imgurl = imgurl.replace("\\\\\\", "\\")
    # input = "Describe the following image:\n" + imgurl
    out = func.agent(f"Please describe the following image:\n{imgurl}")
    anws = func.toChinese(out['output'])
    return anws




# with gr.Blocks() as demo:
#     image_url = gr.Image(type="filepath", label="请选择一张图片")
#     input = gr.Textbox(label='请描述您的问题', placeholder="", lines=1)
#     output = gr.Textbox(label='答案', placeholder="", lines=2, interactive=False)
#     submit = gr.Button('提问', variant="primary")
#     submit.click(work, inputs=[image_url, input], outputs=output)
# demo.launch()

demo = gr.Interface(title="识别图片",
                    css="",
                    fn=work,
                    inputs=[gr.Image(type="filepath", label="请上传图片")],
                    outputs=[gr.Textbox(lines=3, label="识别结果")])
#
# # demo = gr.Interface(fn=work, inputs="image,text", outputs="text")
#
demo.launch()

# # interface = gr.Interface(fn=agent, inputs="image", outputs="text")
# # interface.launch()