【服务器摆设】云端摆设Gradio情感聊天bot应用(ubuntu服务器+Nginx反向署理+内网穿透） - qidao123.com技术社区-IT企服评测·应用市场

sudo apt update
sudo apt-get install python3-venv

复制代码

nano requirements.txt

复制代码

pip install -r requirements.txt

复制代码

mkdir gradio_test/bot

复制代码

import gradio as gr
import json
import erniebot
history = []
# 导入角色设置
with open('../config/enireRoleConfig.json', 'r', encoding='utf-8') as file:
role_config = json.load(file)
# 角色和描述
roles = role_config['roles']
# css样式
custom_css = """
<style>
@import url('https://fonts.googleapis.com/css2?family=Pacifico&family=Permanent+Marker&family=Roboto+Slab&family=Dancing+Script&display=swap');
body, .gradio-container {
display: flex;
flex-direction: column;
min-height: 100vh;
margin: 0;
}
.main {
flex-grow: 1;
display: flex;
flex-direction: column;
}
.contain {
flex-grow: 1;
display: flex;
flex-direction: column;
}
#component-0 {
flex-grow: 1;
display: flex;
flex-direction: column;
}
#component-0 > .chat-interface {
flex-grow: 1;
display: flex;
flex-direction: column;
}
#component-0 > .chat-interface > .message-list {
flex-grow: 1;
overflow-y: auto;
}
</style>
"""
def predict(message, history, option, api_key, role_selector):
model = "ernie-bot"
erniebot.api_type = 'aistudio'
erniebot.access_token = api_key
# 模型选择
if option == "ernie-bot":
model = 'ernie-bot'
elif option == "ernie-bot-turbo":
model = 'ernie-bot-turbo'
elif option == "ernie-bot-4":
model = 'ernie-bot-4'
# 角色选择
system_prompt = roles[role_selector]['prompt']
# 输入消息
json_str = {'role': 'user', 'content': message}
messages = ""
history = [{k: v for k, v in item.items() if k != 'metadata'} for item in history]
history.append(json_str)
# 调用API获取回复
response_stream = erniebot.ChatCompletion.create(
model=model,
messages=history,
stream=True,
system = system_prompt
)
# 流式输出回复
for response in response_stream:
chat = response.get_result()
messages = "".join([messages, chat])
print(chat, end='', flush=True)
yield messages
print("")
print(history)
def update_description(role):
descriptions = {
"温柔女友": f"""<p class='role-description' style='
color: #FF69B4;
font-size: 24px;
font-family: 'Pacifico', cursive;
text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
'>{roles[role]['description']}</p>""",
"可爱甜心": f"""<p class='role-description' style='
color: #DDA0DD;
font-size: 24px;
font-family: 'Permanent Marker', cursive;
text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
'>{roles[role]['description']}</p>""",
"霸道总裁": f"""<p class='role-description' style='
color: #4169E1;
font-size: 24px;
font-family: 'Roboto Slab', serif;
text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
'>{roles[role]['description']}</p>""",
"贴心闺蜜": f"""<p class='role-description' style='
color: #32CD32;
font-size: 24px;
font-family: 'Dancing Script', cursive;
text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
'>{roles[role]['description']}</p>"""
}
return descriptions.get(role, "<p>请选择一个角色</p>")
# 部署页面设计
with gr.Blocks(css=custom_css) as demo:
html_title = gr.HTML("""
<h1 style='text-align:center; font-family: "Brush Script MT", cursive; font-size: 48px; color: #4B0082;'>Elegant AI Chat Assistant</h1>
<style>
@keyframes fadeIn {
from { opacity: 0; }
to { opacity: 1; }
}
#description {
animation: fadeIn 2s ease-in-out;
}
</style>
""")
html_component = gr.HTML(
update_description("温柔女友"), # 默认显示第一个角色的描述
elem_id="role-description"
)
with gr.Row():
with gr.Column():
gr.HTML("<h2 style='text-align:center; font-family: 'Microsoft YaHei'; font-size: 36px; color: #4B0082;'>请输入你的API key</h2>")
api_key = gr.Textbox(
placeholder="输入你的API Key",
value="", # 输入你的apikey
container=False,
interactive=True,
type="password",
label="API Key"
)
option = gr.Radio(
choices=["ernie-bot", "ernie-bot-turbo", "ernie-4.0"],
value="ernie-bot-turbo",
label="模型选择"
)
role_selector = gr.Radio(
choices=["温柔女友", "可爱甜心", "霸道总裁", "贴心闺蜜"],
value="温柔女友",
label="角色选择",
interactive=True,
elem_id="role_selector"
)
role_selector.change(fn=update_description, inputs=role_selector, outputs=html_component)
with gr.Column(elem_id="chatbot-wrapper"):
gr.ChatInterface(
predict,
type="messages",
theme="ocean",
additional_inputs=[option, api_key, role_selector],
fill_height=True,
)
# 本地127.0.0.1，服务端0.0.0.0
demo.launch(server_name="0.0.0.0", server_port=7861,root_path="/bot")

复制代码