Spaces:
Running
Running
File size: 1,792 Bytes
2e77168 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import time
import gradio as gr
import modelscope_studio.components.antd as antd
import modelscope_studio.components.antdx as antdx
import modelscope_studio.components.base as ms
import modelscope_studio.components.pro as pro
def submit(input_value, chatbot_value):
chatbot_value.append({
"role":
"user",
"content": [{
"type": "text",
"content": input_value["text"]
}, {
"type": "file",
"content": [file for file in input_value["files"]]
}]
})
chatbot_value.append({
"role": "assistant",
"loading": True,
"status": "pending"
})
yield gr.update(value=None, loading=True), gr.update(value=chatbot_value)
time.sleep(2)
chatbot_value[-1]["loading"] = False
chatbot_value[-1]["content"] = "Hello"
chatbot_value[-1]["status"] = "done"
yield gr.update(loading=False), gr.update(value=chatbot_value)
def cancel(chatbot_value):
chatbot_value[-1]["loading"] = False
chatbot_value[-1]["footer"] = "canceled"
chatbot_value[-1]["status"] = "done"
return gr.update(loading=False), gr.update(value=chatbot_value)
with gr.Blocks() as demo, ms.Application(), antdx.XProvider():
with antd.Flex(elem_style=dict(height=600), vertical=True):
chatbot = pro.Chatbot(height=0, elem_style=dict(flex=1))
input = pro.MultimodalInput()
submit_event = input.submit(fn=submit,
inputs=[input, chatbot],
outputs=[input, chatbot])
input.cancel(fn=cancel,
cancels=[submit_event],
inputs=[chatbot],
outputs=[input, chatbot],
queue=False)
if __name__ == "__main__":
demo.queue().launch()
|