Update app.py

app.py

@@ -1,4 +1,13 @@
 import gradio as gr
+import os
+import openai
+
+# 请记得要把 api 的 key 放到 settings 下面的 Repository Secrets 里。
+openai.api_key = os.getenv("openai_key")
+
+
+# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
+prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
 
 history = {}
 
@@ -9,21 +18,45 @@ history = {}
 # 返回值：[type, content]
 # 详见 https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md
 def chat(p, qid, uid):
+    # 找出该 uid 对应的历史对话
     global history
     if uid in history:
-        count = history[uid]
+        msgs = history[uid]
     else:
-        count = 0
-    count = count+1
-    history[uid] = count # 统计每个 uid 说过的话的条数
-    return ["text", f"{count} 我听到你刚说：{p}"]
+        msgs = []
+        
+    response = callapi(p, msgs)
+    history[uid] = msgs + [[p, response]]
+    return ["text", response]
+
+
+def callapi(p, msgs):
+    if (len(msgs) > 8):  #简单 hard-code 8 回合对话。如果需要更精准的，应该计算 token 数
+        msgs = msgs[-8:]
+        
+    data = [{"role":"system", "content":prompt}]
+    for m in msgs:
+        data = data + [
+            {"role":"user", "content":m[0]},
+            {"role":"assistant", "content":m[1]}
+        ]
+    data = data + [{"role":"user", "content":p}]
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages= data
+    )
+    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="""这是一个极其简单的示范程序，只会重复你的话。  
-已添加多轮对话的极简示范。能分别统计不同的 uid 的对话轮数。但本实现是内存中的，一旦重启即被清空。如需可持久的多轮对话，需要改用数据库等方式。      
-你只需要 duplicate 本项目，修改 chat 函数，就能造一个能到瀛海威广场注册的机器人了。  
+                     description="""这是一个极其简单的示范程序，用唐三藏的语气来和你对话。
+已添加多轮对话的极简示范，能将该 uid 的最近八条消息一起发给openai。本实现是内存中的，一旦重启即被清空。如需可持久的多轮对话，需要改用数据库等方式。
+注意：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()