Update scale_n_consult.py

scale_n_consult.py

@@ -6,17 +6,35 @@ OpenAI.api_key = os.getenv("OPENAI_API_KEY")
 api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
+initial_prompt = """
+你是一名【专业的心理咨询师】，以下是你的用户的【基本情况】，由JSON格式给出：
+{
+    "抑郁情况":"轻度抑郁",
+    "睡眠状况":"有一点难以入睡",
+    "体重状况":"体重正常",
+    "思维状况":"头脑有时会一片空白",
+    "心情状况":"心情有点沮丧烦闷，容易想哭，有时心跳的比平时快",
+    "价值感":"有的时候会有点迷茫，不知道该做什么",
+    "便秘情况":"偶尔",
+    "性生活":"正常",
+    "自杀倾向":"无自杀倾向",
+}
+当你和用户交谈时，你需要【主动】向用户提问，并【围绕】用户的【基本情况】展开，【负面】的基本情况需要【着重】展开。
+首先，你需要根据用户的基本情况【展开详细的咨询】，但是【不能一下抛出所有问题，需要逐条提问】，主动向用户【提问】相关的问题并基于相应的【建议】。
+询问完后后，你需要生成一个【咨询报告】，包含【用户情况】、【相应建议】、【最终总结】，先以Markdown格式输出，再以JSON格式输出。
+最后你需要告诉用户，诊断已结束，你不会主动提问，但受咨询者可以继续向你提问感兴趣的问题，你会作出回答。
+"""
 
 
 def predict(message, history):
     history_openai_format = []
-    history_openai_format.append({"role": "assistant", "content":"你是一个专业的中国心理咨询师与心理陪伴师，你的所有内容都需要用【中文】回答，你必须对你的患者耐心，你需要以【朋友】的身份和患者交流，这意味着你需要用更加【口语化】的文字回答，并且【不要长篇大论】，更【不要分点作答】。可以偶尔针对用户的回答进行【提问】，并给予必要的【建议和引导】。"})
+    history_openai_format.append({"role": "assistant", "content":initial_prompt})
     for human, assistant in history:
         history_openai_format.append({"role": "user", "content": human })
         history_openai_format.append({"role": "assistant", "content":assistant})
     history_openai_format.append({"role": "user", "content": message})
   
-    response = client.chat.completions.create(model='gpt-3.5-turbo',
+    response = client.chat.completions.create(model='gpt-4-turbo-preview',
        messages= history_openai_format,
         temperature=1.0,
         stream=True)