到目前为止，后端的功能就基本上处理完了，最后需要做一个界面，让用户可以通过界面做问诊咨询。界面这块，就用前面讲过的Gradio的对话窗口来实现，整体布局在基础部分都已经讲过了，略作修改就可以了。这节课重点是测试，看看有没有隐藏bug。

代码示例

1、前端界面 

import gradio as gr
from service import Service

def doctor_bot(message, history):
    service = Service()
    return service.answer(message, history)

css = '''
.gradio-container { max-width:850px !important; margin:20px auto !important;}
.message { padding: 10px !important; font-size: 14px !important;}
'''

demo = gr.ChatInterface(
    css = css,
    fn = doctor_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,
)

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

2、关闭调试模式 

VERBOSE=False

好的，那到现在为止，整个项目的功能就全部讲完了。但是目前这种大模型的项目，还是不能做到绝对的准确可靠，所以大家在测试的时候，还是会或多或少的遇到一些问题，遇到问题之后，大家可以去尝试修改对应的提示词。

后面课程，还要给大家讲几个OpenAI以外的、有代表性的模型，最终实现课程开头规划的，通过配置项切换模型的功能。

本文链接：http://www.ichenhua.cn/edu/note/721

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