app.py

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("key2")




    
# 如果你只打算通过 prompt 来定制机器人的行为，只需要修改这段 prompt 就够了。
prompt = '你需要帮我做MBTI测试，我会告诉你下面5道题我的选择，你要帮我分析我的人格类型，并且对这种人格类型做详细描述，最后要告诉我这种人格类型和什么样的人格类型的人适合做情侣，又和什么人格类型的人适合做朋友。1，你更喜欢：A. 独处，享受自己的时间B. 和朋友一起，享受社交生活C. 参加大型聚会和活动D. 与家人一起，享受亲密时光。2，当你面临压力和挑战时，你会：A. 冷静分析，找出最佳解决方案B. 寻求他人的帮助和支持C. 直接面对，勇敢克服D. 逃避或放弃，寻找其他选择。3，在工作中，你更喜欢：A. 独立完成任务B. 与团队合作，共同完成任务C. 担任领导角色，指导和管理团队D. 从事创造性工作，追求个人成就。4，当你需要做出决策时，你会：A. 依据事实和逻辑，做出理性决策B. 依据个人感觉和直觉，做出决策C. 询问他人的意见和建议，做出决策D. 拖延决策，直到有更多信息可用。5， 在休闲时间，你更喜欢：A. 阅读、学习新知识B. 运动、户外活动C. 看电影、听音乐D. 与家人或朋友一起，享受社交生活如果你了解了我的要求，我们现在就开始，我的选择是'

# 修改本函数，来实现你自己的 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="""我是人格测试助手，对下面五道题，告诉我你的选择，我会帮你做MBTI测试

1. 你更喜欢：
A. 独处，享受自己的时间
B. 和朋友一起，享受社交生活
C. 参加大型聚会和活动
D. 与家人一起，享受亲密时光

2. 当你面临压力和挑战时，你会：
A. 冷静分析，找出最佳解决方案
B. 寻求他人的帮助和支持
C. 直接面对，勇敢克服
D. 逃避或放弃，寻找其他选择

3. 在工作中，你更喜欢：
A. 独立完成任务
B. 与团队合作，共同完成任务
C. 担任领导角色，指导和管理团队
D. 从事创造性工作，追求个人成就

4. 当你需要做出决策时，你会：
A. 依据事实和逻辑，做出理性决策
B. 依据个人感觉和直觉，做出决策
C. 询问他人的意见和建议，做出决策
D. 拖延决策，直到有更多信息可用

5. 在休闲时间，你更喜欢：
A. 阅读、学习新知识
B. 运动、户外活动
C. 看电影、听音乐
D. 与家人或朋友一起，享受社交生活
 """)

iface.launch()