Update app.py

app.py

@@ -12,7 +12,7 @@ user_profile = {
     "dream_day": None,
     "forward_direction_choice": None,
 
-    # 新增：用于 Forward 模式下的更多状态
+    # Forward模式下更多状态
     "recommended_directions": [],  # 3个大方向
     "selected_direction": None,    # 学生最终选定的大方向
     "recommended_jobs": [],        # 3个具体职业
@@ -48,33 +48,64 @@ backward_index = 0
 forward_done = False
 backward_done = False
 forward_recommendation_given = False
-
 recommendation_round = 0
 forward_deep_dive_done = False
 roadmap_offered = False
 roadmap_done = False
 
-# 新增：控制大方向→具体职业的流程
 direction_chosen = False    # 是否已经选定大方向
 jobs_recommended = False    # 是否已给出3个具体职业
 job_chosen = False          # 是否选定了具体职业
 
+# ========== 新增：处理 A/B/C 三部分问题的标记 ==========
+post_career_detail_asked = False  # 是否已询问用户要看A/B/C
+post_career_detail_done = False   # 是否已回答完A/B/C
+
+
 # ============================ 模型设置 ============================
 model_default = "gpt-4o"
 token_default = 2000
 temp_default = 0.7
 top_p_default = 0.95
 
-# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 提供具体职业后，再深入介绍 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-def do_selected_career_detail(selected_career, bg_info, wv, ps, dd):
+# ~~~~~~~~~~~~~~~~~~~~~~~ 新增函数：answer_abc_questions ~~~~~~~~~~~~~~~~~~~~~~~
+def answer_abc_questions(selected, bg_info, wv, ps, dd):
     """
-    学生在3个具体职业里选定某一个后，详细介绍：
-    1. 薪资水平
-    2. 工作环境
-    3. 晋升难度
-    4. 关键技能和考证
-    5. 日常工作节奏
+    根据用户选择 A/B/C/AB/ABC 等，返回相应的额外建议内容。
+    你可以自定义下面三段文案，以符合你的需要。
     """
+    text_A = f"""
+**A: 专业/选课方向 & 求职入门建议**  
+- （这里可结合bg_info, wv, ps, dd给一些更针对的建议）
+"""
+
+    text_B = f"""
+**B: 本校资源利用建议**  
+- （这里介绍如何利用所在学校的Career Center、社团、实验室资源、校友网等）
+"""
+
+    text_C = f"""
+**C: 实习 & Experience 积累 (networking)**  
+- （这里可提到怎么找实习、如何做networking、如何累积实操项目等）
+"""
+
+    sel = selected.lower()
+    response_parts = []
+    if "a" in sel:
+        response_parts.append(text_A)
+    if "b" in sel:
+        response_parts.append(text_B)
+    if "c" in sel:
+        response_parts.append(text_C)
+
+    if not response_parts:
+        return "好的，你暂时不需要查看这三个专项。"
+
+    return "\n".join(response_parts)
+
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 提供具体职业后，再深入介绍 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+def do_selected_career_detail(selected_career, bg_info, wv, ps, dd):
     prompt = f"""
 你是一位专业的职业规划顾问。
 学生最终选择的具体职业: {selected_career}
@@ -116,8 +147,8 @@ def do_selected_career_detail(selected_career, bg_info, wv, ps, dd):
 
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 路线图生成函数 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 def do_time_roadmap():
-    # 优先取 final_choice
-    direction = user_profile.get("final_choice") or user_profile.get("selected_direction") or user_profile.get("forward_direction_choice") or user_profile.get("specific_career")
+    direction = user_profile.get("final_choice") or user_profile.get("selected_direction") or \
+                user_profile.get("forward_direction_choice") or user_profile.get("specific_career")
     bg_info = user_profile["bg_info"] or "未知专业"
     wv = user_profile["work_value"] or "暂无"
     ps = user_profile["personality_summary"] or "暂无"
@@ -163,9 +194,6 @@ def do_time_roadmap():
 
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 给定大方向后，再给3个具体职业 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 def recommend_3jobs_for_direction(direction, bg_info, wv, ps, dd):
-    """
-    当学生选定了某个大方向后，AI再推荐3个该方向下的具体职业
-    """
     prompt = f"""
 你是一位专业的职业规划顾问。
 学生选定的大方向：{direction}
@@ -219,7 +247,6 @@ def recommend_3jobs_for_direction(direction, bg_info, wv, ps, dd):
 
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 深度分析函数（可做备用） ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 def do_deep_analysis():
-    # 保留原有，若后续需要也可用
     direction = user_profile["forward_direction_choice"] or "尚未选择"
     bg_info = user_profile["bg_info"] or "未知背景"
     wv = user_profile["work_value"] or "无"
@@ -371,6 +398,9 @@ def predict(message, history):
 
     global direction_chosen, jobs_recommended, job_chosen
 
+    # 新增全局标记
+    global post_career_detail_asked, post_career_detail_done
+
     # ======================== 初始化 =========================
     if not history:
         current_q_index = 0
@@ -394,6 +424,10 @@ def predict(message, history):
         jobs_recommended = False
         job_chosen = False
 
+        # 新增标记重置
+        post_career_detail_asked = False
+        post_career_detail_done = False
+
     # ======================== 先处理基础问题 =========================
     if 0 < current_q_index <= len(questions):
         key = questions[current_q_index - 1][0]
@@ -421,7 +455,6 @@ def predict(message, history):
     # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
     if mode == "是":
         if in_backward_flow and not backward_done:
-            # 录入回答
             if backward_index > 0 and backward_index <= len(backward_additional_questions):
                 prev_key = backward_additional_questions[backward_index - 1][0]
                 user_profile[prev_key] = message.strip()
@@ -433,7 +466,6 @@ def predict(message, history):
             else:
                 backward_done = True
 
-        # backward_done 表示可以给最终回复
         if backward_done:
             try:
                 api_key = os.environ.get("API_TOKEN")
@@ -461,7 +493,7 @@ def predict(message, history):
     #       Forward模式处理
     # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
     else:
-        # 1) 先问 forward_additional_questions
+        # 如果还没选大方向/没问完前置问题
         if in_forward_flow and not forward_done and not direction_chosen and not jobs_recommended and not job_chosen and not roadmap_done:
             if forward_index > 0 and forward_index <= len(forward_additional_questions):
                 prev_key = forward_additional_questions[forward_index - 1][0]
@@ -472,7 +504,6 @@ def predict(message, history):
                 forward_index += 1
                 return prompt_text
 
-            # 已完成4个问题，准备推荐3大方向
             elif not forward_recommendation_given:
                 forward_recommendation_given = True
                 return recommend_3directions()
@@ -483,11 +514,9 @@ def predict(message, history):
                 if choice in ["1","2","3"]:
                     idx = int(choice) - 1
                     if idx < len(user_profile["recommended_directions"]):
-                        # 选定的大方向
                         sel_dir = user_profile["recommended_directions"][idx]
                         user_profile["selected_direction"] = sel_dir
                         direction_chosen = True
-                        # 进入：给出该方向下的3个具体职业
                         return recommend_3jobs_for_direction(
                             direction=sel_dir,
                             bg_info=user_profile["bg_info"] or "",
@@ -503,15 +532,13 @@ def predict(message, history):
                         forward_done = True
                         return "已多次换方向，结束推荐。"
                     else:
-                        # 再次推荐3大方向
                         return recommend_3directions()
                 else:
                     return "请回复1/2/3选择方向，或输入'换'来重新推荐"
 
-        # 2) 已选定大方向 -> 推荐了3个具体职业
+        # 已选定大方向 -> 推荐了3个具体职业
         elif direction_chosen and not jobs_recommended and not job_chosen:
             # 此时 message 应该是“AI 给出3个具体职业”后，用户的输入
-            # 学生现在选“1/2/3”或“换”
             choice = message.strip().lower()
             if choice in ["1","2","3"]:
                 idx = int(choice) - 1
@@ -520,7 +547,7 @@ def predict(message, history):
                     user_profile["final_choice"] = final_job
                     job_chosen = True
 
-                    # 生成详细介绍
+                    # 生成更详细介绍
                     detail_msg = do_selected_career_detail(
                         selected_career=final_job,
                         bg_info=user_profile["bg_info"] or "",
@@ -529,17 +556,25 @@ def predict(message, history):
                         dd=user_profile["dream_day"] or ""
                     )
                     forward_deep_dive_done = True
-                    return detail_msg
+
+                    # 关键：现在我们“询问 A/B/C” -> 设置 post_career_detail_asked = True
+                    global post_career_detail_asked
+                    post_career_detail_asked = True
+
+                    return detail_msg + "\n\n我还可以针对以下三方面提供建议：\n" \
+                                        "A：专业/选课 & 求职方向\n" \
+                                        "B：本校资源利用建议\n" \
+                                        "C：实习 & networking 积累\n" \
+                                        "如需要其中一个或多个，请输入 A / B / C / AB / BC / AC / ABC\n" \
+                                        "如果都不需要，请回复“不需要”。"
                 else:
                     return "无效的选项，请重新输入1/2/3或'换'"
             elif choice == "换":
-                # 学生对这3个具体职业不满意
                 recommendation_round += 1
                 if recommendation_round > 2:
                     forward_done = True
                     return "已多次换职业，结束推荐。"
                 else:
-                    # 重新推荐3个具体职业
                     sel_dir = user_profile.get("selected_direction") or "未知方向"
                     return recommend_3jobs_for_direction(
                         direction=sel_dir,
@@ -551,7 +586,29 @@ def predict(message, history):
             else:
                 return "请回复1/2/3选择具体职业，或输入'换'来重新推荐"
 
-        # 3) 已选定具体职业 -> 问是否需要时间轴
+        # ============== 新增：处理 A/B/C答疑 ==============
+        elif post_career_detail_asked and not post_career_detail_done:
+            user_choice = message.strip().lower()
+            if user_choice in ["不需要", "no", "n"]:
+                post_career_detail_done = True
+                # 接下来问时间轴
+                roadmap_offered = True
+                return "好的，不查看这三个专项。那需要一个时间轴式职业路线图吗？如需，则回复【要】，否则回复【不要】。"
+            else:
+                # 调用 answer_abc_questions
+                abc_text = answer_abc_questions(
+                    selected=user_choice,
+                    bg_info=user_profile["bg_info"],
+                    wv=user_profile["work_value"],
+                    ps=user_profile["personality_summary"],
+                    dd=user_profile["dream_day"]
+                )
+                post_career_detail_done = True
+                # 问时间轴
+                roadmap_offered = True
+                return abc_text + "\n\n以上是你所选的A/B/C专项信息。需要一个时间轴式职业路线图吗？如需，则回复【要】，否则回复【不要】。"
+
+        # 已选定具体职业 -> 问是否需要时间轴
         elif job_chosen and not roadmap_offered and not roadmap_done:
             roadmap_offered = True
             return "需要一个时间轴式职业路线图吗？如需，则回复【要】，否则回复【不要】。"
@@ -591,7 +648,6 @@ def predict(message, history):
             except Exception as e:
                 return f"发生错误: {str(e)}"
 
-    # 如果都没被return，执行这里：
     return "信息收集完毕，如还未得到最终回复，请再输入任意文字继续。"
 
 
@@ -687,6 +743,9 @@ footer {
 
         global direction_chosen, jobs_recommended, job_chosen
 
+        # 新增
+        global post_career_detail_asked, post_career_detail_done
+
         current_q_index = 0
         questions[:] = base_questions
         for k in user_profile:
@@ -708,6 +767,9 @@ footer {
         jobs_recommended = False
         job_chosen = False
 
+        post_career_detail_asked = False
+        post_career_detail_done = False
+
         return []
 
     reset.click(reset_state, outputs=chatbot)
@@ -716,4 +778,4 @@ footer {
         return [{"role":"assistant","content": questions[0][1]}]
 
     demo.load(auto_first, outputs=chatbot)
-    demo.launch()
+    demo.launch()