Update app.py

app.py

@@ -50,7 +50,7 @@ roadmap_offered = False
 roadmap_done = False
 
 # ============================ 模型设置 ============================
-model_default = "gpt-4o"  # 你可改成 "gpt-3.5-turbo" 等
+model_default = "gpt-4o"
 token_default = 2000
 temp_default = 0.7
 top_p_default = 0.95
@@ -70,14 +70,12 @@ def do_time_roadmap():
 - 工作价值：{wv}
 - 性格：{ps}
 - 理想工作：{dd}
-
 请你使用时间轴方式，给出一个职业路线图，如：
 📍 Now: [当前阶段做什么]
 🎓 Year X: [需要选的课/要做的项目]
 💼 下个暑假: [申请什么实习、参加什么活动]
 📄 毕业前: [考什么证书、准备材料]
 🚀 毕业后: [目标岗位、如何申请]
-
 请用Markdown分段写作，结合学生当前背景合理推断要点，写得具体些。
 """
     try:
@@ -102,7 +100,7 @@ def do_time_roadmap():
     except Exception as e:
         return f"生成路线图时出错: {str(e)}"
 
-# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 深度分析函数（3个具体职业） ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 深度分析函数 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 def do_deep_analysis():
     direction = user_profile["forward_direction_choice"] or "尚未选择"
     bg_info = user_profile["bg_info"] or "未知背景"
@@ -116,7 +114,6 @@ def do_deep_analysis():
 价值观: {wv}
 性格: {ps}
 理想工作: {dd}
-
 请你再深入分析，为学生推荐3个更具体的职业，并说明：
 1. 这些职业对学生背景的契合度
 2. 日常工作内容
@@ -124,7 +121,6 @@ def do_deep_analysis():
 4. 本校资源如何利用
 5. 实习 & 经验积累建议
 6. 简历优化思路
-
 用中文分段写作，字数500+。
 """
     try:
@@ -244,7 +240,7 @@ def predict(message, history):
 
     if not history:
         current_q_index = 0
-        questions = base_questions[:]
+        questions[:] = base_questions  # 用 [:] 保持原列表
         for k in user_profile:
             user_profile[k] = None
 
@@ -259,11 +255,11 @@ def predict(message, history):
         roadmap_offered = False
         roadmap_done = False
 
-    # ========== 如果还在问 base_questions ==========
+    # 如果还在问 base_questions
     if 0 < current_q_index <= len(questions):
         key = questions[current_q_index - 1][0]
+        # 只有回答第一个问题的时候 -> 决定是/否
         if key == "mode" and current_q_index == 1:
-            # 只有回答第一个问题时 => mode
             ans = message.strip()
             user_profile["mode"] = ans
             if "是" in ans:
@@ -280,12 +276,13 @@ def predict(message, history):
 
     mode = user_profile.get("mode") or ""
 
-    # ========== Backward ==========
+    # Backward
     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()
+
             if backward_index < len(backward_additional_questions):
                 k, prompt_text = backward_additional_questions[backward_index]
                 backward_index += 1
@@ -316,7 +313,7 @@ def predict(message, history):
             except Exception as e:
                 return f"发生错误: {str(e)}"
 
-    # ========== Forward ==========
+    # Forward
     else:
         if in_forward_flow and not forward_done and not forward_deep_dive_done and not roadmap_done:
             if forward_index > 0 and forward_index <= len(forward_additional_questions):
@@ -387,71 +384,61 @@ def predict(message, history):
 
     return "信息收集完毕，如还未得到最终回复，请再输入任意文字继续。"
 
-    with gr.Blocks(css="""
-    body {
-        background-color: #1e1e1e;
-        color: #ffffff;
-    }
-
-    .gradio-container {
-        font-family: 'Segoe UI', sans-serif;
-    }
 
-    .gr-chatbot {
-        background-color: transparent !important;
-    }
+### ------ Gradio UI ------
+import gradio as gr
 
-     .message.user {
-    background-color: #cce6ff !important;   /* Very light blue */
-    color: #000000 !important;              /* <-- Black text */
+with gr.Blocks(css="""
+body {
+    background-color: #1e1e1e;
+    color: #ffffff;
+}
+.gradio-container {
+    font-family: 'Segoe UI', sans-serif;
+}
+.gr-chatbot {
+    background-color: transparent !important;
+}
+.message.user {
+    background-color: #cce6ff !important;
+    color: #000000 !important;
     border-radius: 10px !important;
     padding: 10px;
     margin: 6px;
-    }
-
-     .message.bot {
-    background-color: #99ccff !important;   /* Light blue */
-    color: #000000 !important;              /* <-- Black text */
+}
+.message.bot {
+    background-color: #99ccff !important;
+    color: #000000 !important;
     border-radius: 10px !important;
     padding: 10px;
     margin: 6px;
-    }
-
-    .gr-button {
-        border-radius: 8px;
-    }
-
-    #custom-send {
-        background-color: #ec4899 !important;
-        color: white !important;
-        border-radius: 999px !important;
-        padding: 10px 24px !important;
-        font-weight: bold;
-        box-shadow: 0 0 10px #ec4899;
-        transition: all 0.3s ease-in-out;
-    }
-
-    #custom-send:hover {
-        background-color: #d63384 !important;
-        box-shadow: 0 0 12px #ec4899;
-    }
-
-    textarea, input {
-        background-color: #ffe4f1 !important;
-        color: #5e2c49 !important;
-        border: 1px solid #ec4899 !important;
-    }
-
-    footer {
-        display: none !important;
-    }
-
-    """) as demo:
-
+}
+.gr-button {
+    border-radius: 8px;
+}
+#custom-send {
+    background-color: #ec4899 !important;
+    color: white !important;
+    border-radius: 999px !important;
+    padding: 10px 24px !important;
+    font-weight: bold;
+    box-shadow: 0 0 10px #ec4899;
+    transition: all 0.3s ease-in-out;
+}
+#custom-send:hover {
+    background-color: #d63384 !important;
+    box-shadow: 0 0 12px #ec4899;
+}
+textarea, input {
+    background-color: #ffe4f1 !important;
+    color: #5e2c49 !important;
+    border: 1px solid #ec4899 !important;
+}
+footer {
+    display: none !important;
+}
+""") as demo:
     with gr.Row():
- 
-
-
         gr.HTML("""
         <div style='display: flex; align-items: center; justify-content: center; gap: 20px; margin-bottom: 10px;'>
         <!-- 左侧动图 -->
@@ -464,8 +451,8 @@ def predict(message, history):
        </div>
        </div>
         """)
-    gr.Markdown("**Let's chat! 💬** Ask me anything you want")
-    
+    gr.Markdown("## 🎓 职业规划助手\n如果你有明确职业目标 => 回答“是”；否则 => 回答“否”")
+
     chatbot = gr.Chatbot(height=500, show_label=False, show_copy_button=True, type="messages")
     msg = gr.Textbox(placeholder="请输入你的回答...")
     send = gr.Button("发送 🚀", elem_id="custom-send")
@@ -477,9 +464,9 @@ def predict(message, history):
         return "", hist
 
     def add_bot_response(hist):
-        user_m = hist[-1]["content"]
-        bot_m = predict(user_m, hist[:-1])
-        hist.append({"role":"assistant","content":bot_m})
+        user_msg = hist[-1]["content"]
+        bot_msg = predict(user_msg, hist[:-1])
+        hist.append({"role":"assistant","content":bot_msg})
         return hist
 
     send.click(add_user_message, [msg, chatbot], [msg, chatbot]) \
@@ -499,27 +486,27 @@ def predict(message, history):
         global roadmap_offered, roadmap_done
 
         current_q_index = 0
-        questions = base_questions[:]
-        for key in user_profile:
-            user_profile[key] = None
+        questions[:] = base_questions
+        for k in user_profile:
+            user_profile[k] = None
 
         in_forward_flow = False
         in_backward_flow = False
         forward_index = 0
         backward_index = 0
-        forward_done = backward_done = False
+        forward_done = False
+        backward_done = False
         forward_recommendation_given = False
         recommendation_round = 0
         forward_deep_dive_done = False
         roadmap_offered = False
         roadmap_done = False
-
         return []
 
     reset.click(reset_state, outputs=chatbot)
 
-    def first_q():
+    def auto_first():
         return [{"role":"assistant","content": questions[0][1]}]
 
-    demo.load(first_q, outputs=chatbot)
+    demo.load(auto_first, outputs=chatbot)
     demo.launch()