think 不折叠

app.py

@@ -451,6 +451,7 @@ class Gradio_Events:
                         in_thinking = False
                         history[-1]["meta"]["is_thinking"] = False
                         history[-1]["meta"]["thinking_done"] = True
+                        history[-1]["meta"]["just_finished_thinking"] = True  # 标记刚完成thinking
                         
                         # 分离thinking内容和后续内容
                         think_parts = accumulated_content.split("</think>", 1)
@@ -466,10 +467,15 @@ class Gradio_Events:
                                 history[-1]["content"] = current_content + after_think_content
                         
                         accumulated_content = ""  # 重置累积内容
+                        # 先yield一次，让前端检测到just_finished_thinking状态
                         yield {
                             chatbot: gr.update(items=history),
                             state: gr.update(value=state_value)
                         }
+                        
+                        # 然后清除just_finished_thinking标记（如果有内容的话）
+                        if history[-1]["content"] and history[-1]["content"].strip():
+                            history[-1]["meta"]["just_finished_thinking"] = False
                         continue
                     
                     # 如果在thinking模式中，只更新thinking内容，不修改content
@@ -484,6 +490,10 @@ class Gradio_Events:
                             if not history[-1]["content"]:  # 如果content为空才初始化
                                 history[-1]["content"] = ""
                         history[-1]["content"] += content
+                        # 清除"刚完成thinking"标记，因为现在在正常输出内容
+                        # 只在第一次输出内容时清除标记
+                        if history[-1]["meta"].get("just_finished_thinking") and history[-1]["content"].strip():
+                            history[-1]["meta"]["just_finished_thinking"] = False
 
                 yield {
                     chatbot: gr.update(items=history),
@@ -1346,23 +1356,33 @@ with gr.Blocks(css=css, fill_width=True) as demo:
                                             const thinking_content = bubble?.meta?.thinking_content || ""
                                             const is_thinking = bubble?.meta?.is_thinking || false
                                             const thinking_done = bubble?.meta?.thinking_done || false
+                                            const just_finished_thinking = bubble?.meta?.just_finished_thinking || false
                                             
-                                            // 判断是否应该折叠thinking区域
-                                            // 当thinking完成且有正式回答内容时，自动折叠
-                                            const shouldCollapse = thinking_done && content && content.trim().length > 0
+                                            // 只在刚完成thinking且有回答内容时自动折叠
+                                            const shouldAutoCollapse = just_finished_thinking && content && content.trim().length > 0
                                             
-                                            // 使用key来重新渲染组件，触发新的默认状态
-                                            const collapseKey = thinking_done ? 'collapsed' : 'expanded'
+                                            // 使用组合的key和状态策略
+                                            let collapseKey = 'thinking'
+                                            let collapseProps = {}
+                                            
+                                            if (shouldAutoCollapse) {
+                                                // 刚完成thinking且有内容时，强制折叠一次
+                                                collapseKey = 'thinking-auto-collapsed'
+                                                collapseProps.active_key = []
+                                            } else {
+                                                // 其他情况使用默认行为，允许用户控制
+                                                collapseKey = thinking_done ? 'thinking-done' : 'thinking-active'
+                                                collapseProps.default_active_key = ['1']
+                                            }
                                             
                                             return {
-                                                thinking_collapse_props: {
-                                                    key: collapseKey, // 重新渲染组件以应用新的默认状态
-                                                    default_active_key: shouldCollapse ? [] : ['1'],
+                                                thinking_collapse_props: Object.assign({
+                                                    key: collapseKey,
                                                     style: { 
                                                         display: (thinking_content || is_thinking) ? 'block' : 'none',
                                                         marginBottom: thinking_content || is_thinking ? '12px' : '0'
                                                     }
-                                                },
+                                                }, collapseProps), // 动态属性：active_key 或 default_active_key
                                                 thinking_label: is_thinking ? '🤔 正在思考...' : '🤔 思考过程',
                                                 thinking_markdown: {
                                                     value: thinking_content || '思考中...'