Update app.py

app.py

@@ -1,4 +1,10 @@
 import gradio as gr
+import os
+import openai
+
+
+# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
+prompt = '假设你是以为优秀的Python老师，一步步教我PYTHONE，每次只交一个python的知识点，并且展示一个例子。现在开始。'
 
 # 修改本函数，来实现你自己的 chatbot
 # p: 对机器人说话的内容  
@@ -7,13 +13,26 @@ import gradio as gr
 # 返回值：[type, content]
 # 详见 https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md
 def chat(p, qid, uid):
-    return ["text", f"我听到你刚说：{p}"]
+    return ["text", callapi(p)]
+
+openai.api_key = os.getenv("openai_key")
+def callapi(p):
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages= [{"role":"system", "content":prompt},
+                   {"role":"user", "content":p}
+                   ]
+    )
+    print(response)
+    response = response["choices"][0]["message"]["content"]
+    while response.startswith("\n"):
+        response = response[1:]
+    return response
 
 iface = gr.Interface(fn=chat, 
                      inputs=["text", "text", "text"], 
                      outputs=["text", "text"],
-                     description="""这是一个极其简单的示范程序，只会重复你的话。
-你只需要 duplicate 本项目，修改 chat 函数，就能造一个能到瀛海威广场注册的机器人了。  
+                     description="""这是一个极其简单的示范程序，用唐三藏的语气来和你对话。
 [对话测试](https://huggingface.co/spaces/BaixingAI/hackathon_test)   [参考文档](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md)    [Q & A](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/qna.md)
                      """)
 iface.launch()