上节课当中,简单体验了一下Gradio的基本用法,用几行简单的代码,就可以实现一个交互界面。因为比较简单,其他组件就不挨个讲了,大家需要的时候,去看一眼文档就可以了。 

这节课,我们要布局的是一个对话的界面,因为是一个常用的场景,所以Gradio做好了封装,直接调就可以了。

https://www.gradio.app/docs/chatinterface

代码示例

1、简单示例

# gradio_02_chat_interface.py
import gradio as gr

def echo(message, history):
    return message

demo = gr.ChatInterface(
    fn=echo, 
    examples=["hello", "hola", "merhaba"], 
    title="Echo Bot"
)

if __name__ == "__main__":
    demo.launch()

2、界面限宽

# 调整页面宽度和占位
css = '''
.gradio-container { max-width:850px !important; margin:20px auto !important;}
.message { padding: 10px !important; font-size: 14px !important;}
'''

3、界面元素调整

调整元素内容,做一个mini版女神聊天机器人。

demo = gr.ChatInterface(
    css = css,
    fn = nvshen_bot, 
    title = '女神聊天机器人',
    chatbot = gr.Chatbot(height=400, bubble_full_width=False),
    theme = gr.themes.Default(spacing_size='sm', radius_size='sm'),
    textbox=gr.Textbox(placeholder="开始跟女神聊天吧~", container=False, scale=7),
    examples = ['在吗', '饿了吗?', '一起去吃点?', '等我去接你'],
    submit_btn = gr.Button('提交', variant='primary'),
    clear_btn = gr.Button('清空记录'),
    retry_btn = None,
    undo_btn = None,
)

4、逻辑处理函数

import time

def nvshen_bot(message, history):
    pos = message.find('吗')
    time.sleep(1)
    if pos != -1:
        return message[:pos]
    else:
        return '嗯'

现在我们的聊天界面就搭建好了,后面要改成医疗问答机器人,就只需要在这个基础上,略作修改就可以了。

本文链接:http://edu.ichenhua.cn/edu/note/709

版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!