Coloring's picture
feat: add antdx
5945dda
import gradio as gr
import modelscope_studio.components.antdx as antdx
import modelscope_studio.components.base as ms
items = [
{
"label": 'Write a report',
"value": 'report'
},
{
"label": 'Draw a picture',
"value": 'draw'
},
{
"label": 'Check some knowledge',
"value": 'knowledge',
"extra": 'Extra Info',
},
]
def select_suggestion(e: gr.EventData):
return gr.update(value=e._data["payload"][0]), gr.update(open=False)
def change_sender(sender_value: str):
splitted_value = sender_value.split("/")
command = sender_value.split("/")[-1]
if len(splitted_value) > 1:
filtered_items = [
item for item in items
if item["label"].upper().find(command.upper()) != -1
and not command.startswith(item["value"])
]
if len(filtered_items) > 0:
return gr.update(open=True, items=filtered_items)
return gr.update(open=False)
with gr.Blocks() as demo:
with ms.Application():
with antdx.XProvider():
with antdx.Suggestion(open=False) as suggestion:
with ms.Slot("children"):
sender = antdx.Sender(
placeholder="Enter / to get suggestions")
sender.change(fn=change_sender,
inputs=[sender],
outputs=[suggestion])
suggestion.select(fn=select_suggestion,
outputs=[sender, suggestion])
if __name__ == "__main__":
demo.queue().launch()