Update app.py

app.py

@@ -105,9 +105,9 @@ is_active_session = st.session_state.get("is_running", False) or \
                     st.session_state.get("resume_pipeline", False)
 
 if not is_active_session and global_lock.locked():
-    st.markdown("<h3 style='text-align: center; color: #e6a23c;'>⏳ 系统忙碌中 / System Busy</h3>", unsafe_allow_html=True)
-    st.warning("🚨 当前有其他用户正在进行极耗算力的长文推演，系统处于满载排他状态。")
-    st.info("💡 预计 15-25 分钟后释放算力，请稍后再刷新页面尝试。")
+    st.markdown("<h3 style='text-align: center; color: #e6a23c;'>⏳ 系统忙碌 / System Busy</h3>", unsafe_allow_html=True)
+    st.warning("🚨 有小伙伴正在写作中")
+    st.info("💡 预计15-20 分钟后可正常使用")
     st.stop()
 # =====================================================================
 
@@ -723,8 +723,12 @@ if st.session_state.start_pipeline or st.session_state.resume_pipeline:
                 theory_injection = "\n\n".join(extra_directives)
                 sys_prompt_m3 = APP_CONFIG["sys_academic_writer"] + "\n" + APP_CONFIG["deep_refinement_directive"]
                 wiz_data_injection = f"\n<USER_ACTUAL_DATA>\n{user_data_md}\n</USER_ACTUAL_DATA>\n" if user_data_md and sec["is_method_or_result"] else ""
-                base_user = f"Generate section '{display_name}'.\n{length_directive}\n{quarantine_rule}\n{heading_instruction}\n{theory_injection}\n{memory_block}\nSynthesis: {synthesis_result}\n{wiz_data_injection}\nPrevious Tail: ...{context_tail}...\nIMPORTANT: Output final text."
+                # 判断是否为续写区块，动态调整引导语
+                action_verb = f"Generate the main section '{sec_title}'" if sec["is_first"] else f"Continue generating the subsequent paragraphs for the section '{sec_title}' (DO NOT restate the main section title)"
 
+                base_user = f"{action_verb}.\n{length_directive}\n{quarantine_rule}\n{heading_instruction}\n{theory_injection}\n{memory_block}\nSynthesis: {synthesis_result}\n{wiz_data_injection}\nPrevious Tail: ...{context_tail}...\nIMPORTANT: Output final text."
+
+            
             if agent._count_tokens(sys_prompt_m3) + agent._count_tokens(base_user) > 60000: 
                 base_user = agent._truncate_context(base_user, 60000 - agent._count_tokens(sys_prompt_m3))
             
@@ -745,8 +749,15 @@ if st.session_state.start_pipeline or st.session_state.resume_pipeline:
                     
                     status.update(label=f"✅ 【{display_name}】撰写完毕！", state="complete", expanded=False)
 
+                # 提取模型文本后的初步正则清理
                 en_draft_clean = re.sub(rf'(?im)^#\s*{re.escape(sec_title)}\s*\n+', '', en_draft_clean).strip()
                 
+                # =====================================================================
+                # [新增拦截] 暴力擦除模型可能生成的带有 (Part X/Y) 的各级幽灵标题
+                # 匹配任意级别的标题(#) + 任意字符 + (Part 数字/数字) + 换行符，将其替换为空
+                # =====================================================================
+                en_draft_clean = re.sub(r'(?im)^#+\s*.*\(Part\s*\d+/\d+\).*?\n+', '', en_draft_clean)
+                
                 en_draft_clean = re.sub(r'(?m)^(#{2,3})\s+[\d\.\s一二三四五六七八九十、]+\s*(.*)', r'\1 \2', en_draft_clean)
                 en_draft_clean = re.sub(r'(?m)^####\s+(.*)', r'**\1**', en_draft_clean)