zxzhou8212's picture
Update app.py
f6cc49e
raw
history blame contribute delete
No virus
2.67 kB
import gradio as gr
import os
import openai
# 请记得要把 api 的 key 放到 settings 下面的 Repository Secrets 里。
# 目前有个特别奇怪的问题: duplicate 的 key 如果和原来的 key 重名,build 就会失败。不知是否是今天正在 migrating 的原因。
# 作为 workaround,请对 key 使用一个不同的名字,并且记得修改下面这行代码中的 key 的名字。
openai.api_key = os.getenv("zxzhou8212_key")
# 如果你只打算通过 prompt 来定制机器人的行为,只需要修改这段 prompt 就够了。
prompt = '请你扮演武侠小说的作者金庸,使用金庸的语气、方式和词汇回答问题。不要写任何解释,只需像金庸一样回答问题。你必须掌握金庸的所有作品的所有知识。'
# 修改本函数,来实现你自己的 chatbot
# p: 对机器人说话的内容
# qid: 当前消息的唯一标识。例如 `'bxqid-cManAtRMszw...'`。由平台生成并传递给机器人,以便机器人区分单个问题(写日志、追踪调试、异步回调等)。同步调用可忽略。
# uid: 用户的唯一标识。例如`'bxuid-Aj8Spso8Xsp...'`。由平台生成并传递给机器人,以便机器人区分用户。可被用于实现多轮对话的功能。
# 返回值:[type, content]
# 详见 https://huggingface.co/spaces/baixing/hackathon_test/blob/main/bot-api.md
def chat(p, qid, uid):
return ["text", callapi(p)]
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 本项目后,需要将你自己的 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()