Update app.py

app.py

@@ -1,4 +1,18 @@
 import gradio as gr
+import os
+import requests
+
+# 请记得要把 api 的 key 放到 settings 下面的 Repository Secrets 里。
+# 目前有个特别奇怪的问题： duplicate 的 key 如果和原来的 key 重名，build 就会失败。不知是否是今天正在 migrating 的原因。
+# 作为 workaround，请对 key 使用一个不同的名字，并且记得修改下面这行代码中的 key 的名字。
+key = os.getenv("KEY")
+
+# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
+prompt = """请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。
+########
+"""
+
+url = "https://gpt.baixing.com/"
 
 history = {}
 
@@ -8,22 +22,25 @@ history = {}
 # uid: 用户的唯一标识。例如`'bxuid-Aj8Spso8Xsp...'`。由平台生成并传递给机器人，以便机器人区分用户。可被用于实现多轮对话的功能。  
 # 返回值：[type, content]
 # 详见 https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md
-def chat(p, qid, uid):
+def chat(p, qid, uid): 
     global history
     if uid in history:
-        count = history[uid]
+        count = history[uid] # 还需要添加复杂一些的逻辑，例如超过10条要重温，超过1小时要重温 etc
     else:
         count = 0
+        p = prompt + p #第一次访问，需要加上引导的 prompt
+
     count = count+1
+    print(uid, count)
     history[uid] = count # 统计每个 uid 说过的话的条数
-    return ["text", f"{count} 我听到你刚说：{p}"]
+
+    result = requests.get(url, params={"p": p, "k": key, "session_id": uid}).json()['data']
+    return ["text", result]
 
 iface = gr.Interface(fn=chat, 
                      inputs=["text", "text", "text"], 
                      outputs=["text", "text"],
-                     description="""这是一个极其简单的示范程序，只会重复你的话。  
-已添加多轮对话的极简示范。能分别统计不同的 uid 的对话轮数。但本实现是内存中的，一旦重启即被清空。如需可持久的多轮对话，需要改用数据库等方式。      
-你只需要 duplicate 本项目，修改 chat 函数，就能造一个能到瀛海威广场注册的机器人了。  
-[对话测试](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)
-                     """)
+                     description="""这是一个极其简单的示范程序，用唐三藏的语气来和你对话。)
+                     """
+                    )
 iface.launch()