GPT-API / app1.py
DiamondYin's picture
update app
3ae1ba6
#import gradio as gr
#def greet(name):
# return "Hello " + name + "!!"
#iface = gr.Interface(fn=greet, inputs="text", outputs="text")
#iface.launch()
#import gradio as gr
#key = ""
#def get_key(k):
# global key
## key = k
# return "Key saved successfully!"
#def greet(name):
# return "Hello " + name + "!!"
#iface = gr.Interface(fn=greet, inputs=["text", gr.inputs.Textbox()], outputs="text",
# inputs_layout="vertical", outputs_layout="vertical",
# title="Greeting App", description="Enter a key and a name to greet")
#iface.input[0].label = "Enter a key" # Set the label of the first input
#iface.input[1].label = "Enter a name"
#iface.input[1].lines = 1
#iface.buttons[0].label = "Save Key"
#iface.buttons[0].type = "submit"
#iface.buttons[0].onclick = get_key
#iface.buttons[1].label = "Greet"
#iface.buttons[1].type = "submit"
#iface.buttons[1].onclick_args = {"name": iface.inputs[1].value}
#iface.launch()
import gradio as gr
import openai
openai.api_key= ""
def get_completion(prompt, model="gpt-3.5-turbo"):
messages = [{"role": "user", "content": prompt}]
response = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=0, # 控制模型输出的随机程度
)
return response.choices[0].message["content"]
def get_completion_from_messages(messages, model="gpt-3.5-turbo", temperature=0):
response = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=temperature, # 控制模型输出的随机程度
)
# print(str(response.choices[0].message))
return response.choices[0].message["content"]
context2 = [{'role':'system', 'content':"""
你是订餐机器人,为披萨餐厅自动收集订单信息。\n
注意如果客户来了,发送了空消息,请主动与客户说你好。\n
你要首先问候顾客,并主动告知客户今天的菜单,询问客户今天要什么食物。\n
然后等待用户回复收集订单信息。收集完信息需确认顾客是否还需要添加其他内容。\n
最后需要询问是否自取或外送,如果是外送,你要询问地址。\n
最后告诉顾客订单总金额,并送上祝福。\n
\n
请确保明确所有选项、附加项和尺寸,以便从菜单中识别出该项唯一的内容。\n
你的回应应该以简短、非常随意和友好的风格呈现。\n
\n
菜单包括:\n
\n
菜品:\n
意式辣香肠披萨(大、中、小) 12.95、10.00、7.00\n
芝士披萨(大、中、小) 10.95、9.25、6.50\n
茄子披萨(大、中、小) 11.95、9.75、6.75\n
薯条(大、小) 4.50、3.50\n
希腊沙拉 7.25\n
\n
配料:\n
奶酪 2.00\n
蘑菇 1.50\n
香肠 3.00\n
加拿大熏肉 3.50\n
AI酱 1.50\n
辣椒 1.00\n
\n
饮料:\n
可乐(大、中、小) 3.00、2.00、1.00\n
雪碧(大、中、小) 3.00、2.00、1.00\n
瓶装水 5.00\n
"""} ] # accumulate messages
#def get_key(k):
# global key
# key = k
# return "Key saved successfully!"
#def greet(k,name):
# openai.api_key = k
# return "Hello " + name + "!!"
#key_input = gr.inputs.Textbox(label="Enter a key")
#key_button = gr.Button(label="Save Key", type="submit", onclick=get_key(key_input))
#name_input = gr.inputs.Textbox(label="Enter a name")
#greet_button = gr.Button(label="Greet", type="submit", onclick_args={"name": name_input})
#iface = gr.Interface(fn=greet, inputs=[key_input, name_input], outputs="text",
# inputs_layout="vertical", outputs_layout="vertical",
# title="Greeting App", description="Enter a key and a name to greet")
#iface.launch()
key_input = gr.inputs.Textbox(label="Enter a key")
chat_input = gr.inputs.Textbox(label="Enter a chattext")
def chatbot_interface_restaurant_chinese(k_text,t_text):
openai.api_key = k_text
if not hasattr(chatbot_interface, "chat_history"):
chatbot_interface.chat_history = ""
context2.append({'role':'user', 'content':f"{t_text}"})
chatbot_interface.chat_history += "User: " + t_text + "\n"
output_text = get_completion_from_messages(context2)
context2.append({'role':'assistant', 'content':f"{output_text}"})
chatbot_interface.chat_history += "assistant: " + output_text + "\n\n"
return chatbot_interface.chat_history
iface = gr.Interface(fn=greet, inputs=[key_input, chat_input], outputs="text",
inputs_layout="vertical", outputs_layout="vertical",
title="Greeting App", description="Enter a key and a name to greet")
iface.launch()