File size: 1,837 Bytes
7209f44 464cf39 7209f44 6ef37ba 7209f44 464cf39 0358438 464cf39 0358438 464cf39 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import gradio as gr
import openai
config = open("config").readlines()
openai.api_key = config[0].strip()
openai.api_base = config[1].strip()
'''
gradio: [['第一次说话', 'No'], ['试试第二次', 'Yes']]
openai: [{"role": "user", "content": "第一次说话"},
{"role": "assistant", "content": "Who won the world series in 2020?"}]
'''
def gradio_messages_to_openai_messages(g):
result = []
for pair in g:
result.append({"role": "user", "content": pair[0]})
result.append({"role": "assistant", "content": pair[1]})
return result
def respond(chat_history, message):
print("----------------")
print("chat_histroy:", chat_history)
print("message:", message)
messages = [
{"role": "system", "content": "后面的回答必须简明扼要"},
*gradio_messages_to_openai_messages(chat_history),
{"role": "user", "content": message}
]
print("messages:", messages)
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages
)
print("completion:", completion)
response = completion['choices'][0]['message']['content']
result = chat_history + [[message, response]]
print("result:", result)
return result
with gr.Blocks() as demo:
gr.Markdown("## Chat with GPT")
state = gr.State()
chatbot = gr.Chatbot()
message = gr.Textbox(label = "Message:", placeholder="Enter text")
message.submit(
respond,
[chatbot, message],
chatbot,
)
with gr.Row():
clear = gr.Button("Clear")
clear.click(lambda: None, None, chatbot)
send = gr.Button("Send")
send.click(
respond,
[chatbot, message],
chatbot,
)
demo.launch(server_name="0.0.0.0", server_port=8000)
|