Update app.py

app.py

@@ -38,27 +38,106 @@
 #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
 
-key = ""
+#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})
 
-def get_key(k):
-    global key
-    key = k
-    return "Key saved successfully!"
 
-def greet(k,name):
-    global key
-    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 = 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()