Update app.py

app.py

@@ -73,11 +73,11 @@ top_p_default = 0.95
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 def answer_abc_questions(selected, bg_info, wv, ps, dd):
     desired_parts = []
-    if "a" in selected:
+    if "a" in selected.lower():
         desired_parts.append("A")
-    if "b" in selected:
+    if "b" in selected.lower():
         desired_parts.append("B")
-    if "c" in selected:
+    if "c" in selected.lower():
         desired_parts.append("C")
 
     if not desired_parts:
@@ -223,10 +223,12 @@ def recommend_3jobs_for_direction(direction, bg_info, wv, ps, dd):
 2) 典型工作职责
 3) 技能/证书要求
 4) 未来发展前景
-最后再用如下格式提示学生选择：
-1. XXX职业
-2. XXX职业
-3. XXX职业
+最后以这样的格式结束（不要在前面重复这些选项）：
+
+请选择你的理想职业：
+1. [职业名称1]
+2. [职业名称2]
+3. [职业名称3]
 如都不满意，可输入'换'。
 """
     try:
@@ -249,13 +251,21 @@ def recommend_3jobs_for_direction(direction, bg_info, wv, ps, dd):
         )
         output_text = resp.choices[0].message.content
 
-        # 示例：从回复中解析出职业名称
-        # 这里为简单起见使用模拟数据，实际应用中可以从output_text中解析
-        user_profile["recommended_jobs"] = [
-            f"{direction} - 职业A",
-            f"{direction} - 职业B",
-            f"{direction} - 职业C"
-        ]
+        # 从回复中尝试解析职业名称 - 简单示例实现
+        import re
+        job_pattern = r'(\d+)\.\s+\[?(.*?)\]?(?:\n|$)'
+        job_matches = re.findall(job_pattern, output_text)
+        job_list = [match[1].strip() for match in job_matches]
+        
+        # 如果没有找到合适的匹配，创建默认值
+        if len(job_list) < 3:
+            job_list = [
+                f"{direction} - 职业A",
+                f"{direction} - 职业B",
+                f"{direction} - 职业C"
+            ]
+        
+        user_profile["recommended_jobs"] = job_list[:3]  # 确保只取前3个
         
         # 设置为已经推荐过职业
         global jobs_recommended
@@ -274,18 +284,22 @@ def recommend_3directions():
     dd = user_profile["dream_day"] or "无"
 
     rec_prompt = f"""
-请根据以下信息，写一份条理清晰的人物画像+3个大方向，并结尾让学生选1~3(或'换')，其中称呼对面用"你"不要用"他/她"：
+请根据以下信息，写一份条理清晰的人物画像+3个大方向，其中称呼对面用"你"不要用"他/她"：
 - 学术背景: {bg_info}
 - 工作意义: {wv}
 - 性格: {ps}
 - 理想工作: {dd}
 分两部分：
 [一、人物画像]：1) 学术背景 2) 价值观 3) 性格 4) 理想工作
-[二、推荐3个大方向]：每个方向(1~2段文字)
-最后用如下结尾：
-1. XXX方向
-2. XXX方向
-3. XXX方向
+
+[二、推荐3个大方向]：每个方向写1~2段文字分析，并说明为什么适合学生。
+
+最后结束时只有这样的选项（不要在前面重复这些选项）：
+
+请选择你的职业方向：
+1. [方向名称1]
+2. [方向名称2]
+3. [方向名称3]
 如都不满意，可输入'换'。
 """
     try:
@@ -307,13 +321,21 @@ def recommend_3directions():
         )
         output_text = resp.choices[0].message.content
 
-        # 示例：从回复中解析方向名称
-        # 这里为简单起见使用模拟数据，实际应用中可以从output_text中解析
-        user_profile["recommended_directions"] = [
-            "方向1",
-            "方向2",
-            "方向3"
-        ]
+        # 从回复中尝试解析方向名称 - 简单示例实现
+        import re
+        dir_pattern = r'(\d+)\.\s+\[?(.*?)\]?(?:\n|$)'
+        dir_matches = re.findall(dir_pattern, output_text)
+        dir_list = [match[1].strip() for match in dir_matches]
+        
+        # 如果没有找到合适的匹配，创建默认值
+        if len(dir_list) < 3:
+            dir_list = [
+                "方向1", 
+                "方向2", 
+                "方向3"
+            ]
+            
+        user_profile["recommended_directions"] = dir_list[:3]  # 确保只取前3个
         return output_text
     except Exception as e:
         return f"多方向推荐出错: {str(e)}"