clarkchan's picture
Update app.py
fc55914
import gradio as gr
import requests
#from transformers import AutoTokenizer
#tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
conversation_history = ""
def chat(input):
global conversation_history
url = 'http://dark.21cnai.com:5000/api/chat'
data = { "message": input }
headers = {"Content-Type": "Application/json","Authorization":"Bearer kdfjwoieskdflasdnf"}
user_message = f'<span class="user">您说:{input}</span><br>'
response = requests.post(url, json=data, headers=headers)
bot_message = f'<span class="chatbot">ChatGPT:{response.text}</span><br>'
conversation_history = user_message + bot_message
return conversation_history
css = '''
.input_text, .output_text {
font-family: Arial, sans-serif;
font-size: 16px;
}
.input_text:focus {
border: 2px solid #2C7BE5;
outline: none;
}
.output_text {
background-color: #F8F9FA;
border: 1px solid #CED4DA;
color: #495057;
}
.input_button {
background-color: #2C7BE5;
color: white;
font-weight: bold;
border: none;
}
.user {
color: blue;
}
.chatbot {
color: green;
}
.container {
display: flex;
flex-direction: column;
height: 100%;
}
.output_section {
flex-grow: 1;
}
'''
iface = gr.Interface(fn=chat,
outputs=gr.outputs.HTML(),
inputs=gr.inputs.Textbox(lines=3, placeholder="在此输入您的问题..."),
title="ChatGPT 对话",
description="请输入您的问题,然后按回车键或单击提交。",
layout="vertical",
css=css)
iface.launch(server_name="0.0.0.0")