Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -2,15 +2,15 @@ import gradio as gr
|
|
2 |
import requests
|
3 |
#from transformers import AutoTokenizer
|
4 |
#tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
|
|
|
5 |
def chat(input):
|
6 |
-
global conversation_history
|
7 |
url = 'http://dark.21cnai.com:5000/api/chat'
|
8 |
data = { "message": input }
|
9 |
headers = {"Content-Type": "Application/json","Authorization":"Bearer kdfjwoieskdflasdnf"}
|
10 |
-
|
11 |
-
conversation_history += response
|
12 |
response = requests.post(url, json=data, headers=headers)
|
13 |
-
|
|
|
14 |
return conversation_history
|
15 |
|
16 |
css = '''
|
@@ -33,7 +33,27 @@ css = '''
|
|
33 |
font-weight: bold;
|
34 |
border: none;
|
35 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
'''
|
37 |
|
38 |
-
iface = gr.Interface(fn=chat,
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
iface.launch(server_name="0.0.0.0")
|
|
|
2 |
import requests
|
3 |
#from transformers import AutoTokenizer
|
4 |
#tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
|
5 |
+
conversation_history = ""
|
6 |
def chat(input):
|
|
|
7 |
url = 'http://dark.21cnai.com:5000/api/chat'
|
8 |
data = { "message": input }
|
9 |
headers = {"Content-Type": "Application/json","Authorization":"Bearer kdfjwoieskdflasdnf"}
|
10 |
+
user_message = f'<span class="user">您说:{input_text}</span><br>'
|
|
|
11 |
response = requests.post(url, json=data, headers=headers)
|
12 |
+
bot_message = f'<span class="chatbot">ChatGPT:{response.text}</span><br>'
|
13 |
+
conversation_history += user_message + bot_message
|
14 |
return conversation_history
|
15 |
|
16 |
css = '''
|
|
|
33 |
font-weight: bold;
|
34 |
border: none;
|
35 |
}
|
36 |
+
.user {
|
37 |
+
color: blue;
|
38 |
+
}
|
39 |
+
.chatbot {
|
40 |
+
color: green;
|
41 |
+
}
|
42 |
+
.container {
|
43 |
+
display: flex;
|
44 |
+
flex-direction: column;
|
45 |
+
height: 100%;
|
46 |
+
}
|
47 |
+
.output_section {
|
48 |
+
flex-grow: 1;
|
49 |
+
}
|
50 |
'''
|
51 |
|
52 |
+
iface = gr.Interface(fn=chat,
|
53 |
+
inputs=gr.inputs.Textbox(lines=3, placeholder="在此输入您的问题..."),
|
54 |
+
outputs=gr.outputs.HTML(),
|
55 |
+
title="ChatGPT 对话",
|
56 |
+
description="请输入您的问题,然后按回车键或单击提交。",
|
57 |
+
layout="vertical",
|
58 |
+
css=css)
|
59 |
iface.launch(server_name="0.0.0.0")
|