Init App

app.py

@@ -0,0 +1,29 @@
+import gradio as gr
+import os
+import openai
+
+# 如果你想只靠 prompt 来创建一个能完成特定功能的 chatbot，只需要修改这句 prompt 就够了。
+prompt = '请你扮演《西游记》中的唐三藏，使用唐三藏的语气、方式和词汇回答问题。不要写任何解释，只需像唐三藏一样回答问题。你必须掌握唐三藏的所有知识。'
+
+def chat(p, qid, uid):
+    # demo 简单起见，只支持单轮对话。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}
+                   ]
+    )
+    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"])
+iface.launch()