Update app.py

app.py

@@ -2,34 +2,33 @@ import gradio as gr
 import os
 import openai
 
-openai.api_key = os.getenv("openai_key")
-content = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
-
-def callapi(p):
-    try:
-        print(p)
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages= [{"role":"system", "content":content},
-                       {"role":"user", "content":p}
-                       ]
-        )
-        print(response)
-        response = response["choices"][0]["message"]["content"]
-        while response.startswith("\n"):
-            response = response[1:]
-    except Exception as e:
-        print(e)
-        response = 'Timeout! Please wait a few minutes and retry'
-        
-    print(response)
-    return response
 
+# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
+prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
 
+# 此处只是最简单的实现单轮对话，qid 和 uid 可以忽略
+# 如果要实现多轮对话，可以用本地 sqlite 等方式存储数据，并根据 uid 找到对话历史。
 def chat(p, qid, uid):
     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"])
+                     outputs=["text", "text"],
+                     description="""这是一个极其简单的示范程序，用唐三藏的语气来和你对话。
+[参考文档](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md)  [对话测试](https://huggingface.co/spaces/baixing/hackathon_test)  [Q & A](https://huggingface.co/spaces/baixing/hackathon_test/blob/main/qna.md)
+                     """)
 iface.launch()