Spaces:
Paused
Paused
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") |