Update app.py

app.py

@@ -1,4 +1,10 @@
 import gradio as gr
+import os
+import openai
+
+
+# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
+prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
 
 # 修改本函数，来实现你自己的 chatbot
 # p: 对机器人说话的内容  
@@ -7,13 +13,27 @@ 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="""这是一个极其简单的示范程序，用唐三藏的语气来和你对话。
+注意：duplicate 本项目后，需要将你自己的 openai apikey 设置到 settings 的 Repository Secrets 里，否则运行会报错。[了解详情](https://huggingface.co/spaces/baixing/hackathon_chatbot_openai_api/blob/main/%E6%B7%BB%E5%8A%A0%20secret%20%E7%9A%84%E6%96%B9%E6%B3%95.jpg)    
 [对话测试](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()
+iface.launch()