import os import time import gradio as gr import modelscope_studio as mgr def resolve_assets(relative_path): return os.path.join(os.path.dirname(__file__), "../resources", relative_path) conversation = [ [None, { "text": "Hello I'm a chatbot", "flushing": False }], ] def get_last_bot_message(chatbot): return chatbot[-1][1] def create_music_bot_message(text: str): return { "text": text, } def create_image_bot_message(text: str): return { "text": text, } def submit(_input, _chatbot): _chatbot.append([_input, None]) yield gr.update(interactive=False, value=None), _chatbot _chatbot[-1][1] = [ "Hello", create_image_bot_message("Hello"), create_music_bot_message("Hello") ] time.sleep(2) get_last_bot_message(_chatbot)[1][ "text"] = f"""Hello, I\'m a image bot\n![image]({resolve_assets("user.jpeg")})""" get_last_bot_message(_chatbot)[2][ "text"] = f"""Hello, I\'m a music bot """ yield { chatbot: _chatbot, } def flushed(): return gr.update(interactive=True) with gr.Blocks() as demo: chatbot = mgr.Chatbot( value=conversation, avatar_image_width=40, avatar_images=[ resolve_assets('user.jpeg'), # default bot avatar and name [{ "name": "bot", "avatar": resolve_assets('bot.jpeg') }, { "name": "image bot", "avatar": resolve_assets('image-bot.jpeg') }, { "name": "music bot", "avatar": resolve_assets('music-bot.jpeg') }] ], height=600, ) input = mgr.MultimodalInput() input.submit(fn=submit, inputs=[input, chatbot], outputs=[input, chatbot]) chatbot.flushed(fn=flushed, outputs=[input]) if __name__ == "__main__": demo.queue().launch()