Add four-tab interface for caching all stage results, enable Qwen3-8B thinking, and implement proper stage switching

app.py

@@ -886,37 +886,63 @@ async def streaming_ai_analysis(text: str, keywords: List[str], country: str, la
     except Exception as e:
         yield f"分析出错：{str(e)}"
 
-# 三次渐进式输出的审核函数 - 真正的流式输出
+# 三次渐进式输出的审核函数 - 缓存结果并支持切换
 def comprehensive_text_audit_streaming(text, keywords="", country="China", language="简体中文", model="Qwen/Qwen3-8B (Free - SiliconFlow)"):
     """
-    三次渐进式输出的AI内容审核 - 三次请求，真正的流式输出
+    三次渐进式输出的AI内容审核 - 缓存所有阶段结果
     """
     if not text.strip():
-        yield "❌ Please enter text content"
+        yield ("❌ Please enter text content", "", "", "")
         return
 
     # Parse keywords
     keyword_list = [k.strip() for k in keywords.split(",") if k.strip()] if keywords else []
 
-    # 初始化对话历史
+    # 初始化对话历史和结果缓存
     messages = []
+    stage1_result = ""
+    stage2_result = ""
+    stage3_result = ""
 
     try:
-        # 第一阶段：基础分析 - 流式输出
+        # 第一阶段：基础分析
+        yield ("🔍 Stage 1: Basic Analysis in progress...", "", "", "")
+
+        stage1_final = None
         for chunk in perform_streaming_analysis(text, keyword_list, country, language, model, messages, stage=1):
-            yield chunk
+            stage1_final = chunk
+            yield (chunk, "", "", "")
+
+        stage1_result = stage1_final
+
+        # 第二阶段：详细标注
+        yield (stage1_result, "🎨 Stage 2: Detailed Annotation in progress...", "", "")
 
-        # 第二阶段：详细标注 - 流式输出
+        stage2_final = None
         for chunk in perform_streaming_analysis(text, keyword_list, country, language, model, messages, stage=2):
-            yield chunk
+            stage2_final = chunk
+            yield (stage1_result, chunk, "", "")
+
+        stage2_result = stage2_final
+
+        # 第三阶段：修改建议
+        yield (stage1_result, stage2_result, "✨ Stage 3: Revision Suggestions in progress...", "")
 
-        # 第三阶段：修改建议 - 流式输出
+        stage3_final = None
         for chunk in perform_streaming_analysis(text, keyword_list, country, language, model, messages, stage=3):
-            yield chunk
+            stage3_final = chunk
+            yield (stage1_result, stage2_result, chunk, "")
+
+        stage3_result = stage3_final
+
+        # 生成摘要界面
+        summary_interface = generate_analysis_summary()
+        yield (stage1_result, stage2_result, stage3_result, summary_interface)
 
     except Exception as e:
         print(f"Analysis error: {e}")
-        yield generate_error_card(str(e))
+        error_msg = generate_error_card(str(e))
+        yield (error_msg, "", "", "")
 
 # 执行流式分析
 def perform_streaming_analysis(text: str, keywords: List[str], country: str, language: str, model: str, messages: list, stage: int):
@@ -1044,13 +1070,20 @@ def perform_streaming_analysis(text: str, keywords: List[str], country: str, lan
         )
 
         # 发送流式请求
-        response = client.chat.completions.create(
-            model=model_config["model"],
-            messages=current_messages,
-            stream=True,
-            max_tokens=4096,
-            temperature=0.1
-        )
+        request_params = {
+            "model": model_config["model"],
+            "messages": current_messages,
+            "stream": True,
+            "max_tokens": 4096,
+            "temperature": 0.1
+        }
+
+        # 为Qwen3-8B启用thinking
+        if model_config["model"] == "Qwen/Qwen3-8B":
+            request_params["enable_thinking"] = True
+            request_params["thinking_budget"] = 2048
+
+        response = client.chat.completions.create(**request_params)
 
         # 逐步接收并处理响应
         for chunk in response:
@@ -1176,6 +1209,84 @@ def generate_streaming_stage_html(stage: int, title: str, content: str, reasonin
     </div>
     """
 
+# 生成分析摘要
+def generate_analysis_summary() -> str:
+    """生成分析摘要界面"""
+    return f"""
+    <div style="
+        max-width: 900px;
+        margin: 0 auto;
+        padding: 30px;
+        background: linear-gradient(135deg, #E8E3DE 0%, #F2EDE9 100%);
+        border-radius: 20px;
+        font-family: 'Microsoft YaHei', 'Noto Sans SC', sans-serif;
+    ">
+        <div style="
+            background-color: #F2EDE9;
+            border-radius: 15px;
+            padding: 25px;
+            box-shadow: 0 10px 30px rgba(0,0,0,0.1);
+        ">
+            <h2 style="
+                color: #5B5B5B;
+                margin-bottom: 20px;
+                font-size: 28px;
+                border-bottom: 3px solid #9A8F8F;
+                padding-bottom: 10px;
+            ">📊 Analysis Summary</h2>
+
+            <div style="
+                background-color: rgba(255,255,255,0.7);
+                padding: 20px;
+                border-radius: 10px;
+                margin-bottom: 20px;
+            ">
+                <h3 style="color: #5B5B5B; margin-bottom: 15px; font-size: 20px;">🎯 Three-Stage Analysis Complete</h3>
+
+                <div style="margin: 15px 0;">
+                    <h4 style="color: #4CAF50; margin-bottom: 10px;">🔍 Stage 1: Basic Analysis</h4>
+                    <p style="color: #5B5B5B; margin-left: 20px;">Initial risk assessment and compliance evaluation</p>
+                </div>
+
+                <div style="margin: 15px 0;">
+                    <h4 style="color: #FF9800; margin-bottom: 10px;">🎨 Stage 2: Detailed Annotation</h4>
+                    <p style="color: #5B5B5B; margin-left: 20px;">Sentence-level risk marking with color coding</p>
+                </div>
+
+                <div style="margin: 15px 0;">
+                    <h4 style="color: #9C27B0; margin-bottom: 10px;">✨ Stage 3: Revision Suggestions</h4>
+                    <p style="color: #5B5B5B; margin-left: 20px;">Corrected text with compliance improvements</p>
+                </div>
+            </div>
+
+            <div style="
+                background-color: rgba(255,255,255,0.5);
+                padding: 20px;
+                border-radius: 10px;
+                margin-bottom: 20px;
+            ">
+                <h3 style="color: #5B5B5B; margin-bottom: 15px; font-size: 20px;">💡 How to Use</h3>
+                <ul style="color: #5B5B5B; line-height: 1.8;">
+                    <li><strong>Switch between tabs</strong> to view different analysis stages</li>
+                    <li><strong>Click the reasoning sections</strong> to see AI's thought process</li>
+                    <li><strong>Compare original and revised text</strong> in Stage 3</li>
+                    <li><strong>Use color coding</strong> in Stage 2 to identify risk levels</li>
+                </ul>
+            </div>
+
+            <div style="
+                text-align: center;
+                color: #4A7C59;
+                font-size: 18px;
+                font-weight: bold;
+                padding: 20px;
+                background-color: rgba(74, 124, 89, 0.1);
+                border-radius: 8px;
+            ">🎉 Content Audit Complete! Review all stages above.</div>
+        </div>
+    </div>
+    """
+
 # 执行reasoning分析
 def perform_reasoning_analysis(text: str, keywords: List[str], country: str, language: str, model: str, messages: list, stage: int) -> dict:
     """执行带reasoning的AI分析"""
@@ -2475,11 +2586,31 @@ with gr.Blocks(
                 clear_btn = gr.Button("🗑️ Clear", variant="secondary", scale=1)
 
         with gr.Column(scale=2):
-            # 输出区域
-            output_html = gr.HTML(
-                label="AI Audit Report",
-                value="<div style='text-align: center; padding: 50px; color: #888;'>Please enter text and click 'Start Analysis' to begin.</div>"
-            )
+            # 输出区域 - 四个标签页
+            with gr.Tabs():
+                with gr.TabItem("🔍 Stage 1: Basic Analysis"):
+                    stage1_output = gr.HTML(
+                        value="<div style='text-align: center; padding: 50px; color: #888;'>Stage 1 results will appear here...</div>",
+                        label="Stage 1 Results"
+                    )
+
+                with gr.TabItem("🎨 Stage 2: Detailed Annotation"):
+                    stage2_output = gr.HTML(
+                        value="<div style='text-align: center; padding: 50px; color: #888;'>Stage 2 results will appear here...</div>",
+                        label="Stage 2 Results"
+                    )
+
+                with gr.TabItem("✨ Stage 3: Revision Suggestions"):
+                    stage3_output = gr.HTML(
+                        value="<div style='text-align: center; padding: 50px; color: #888;'>Stage 3 results will appear here...</div>",
+                        label="Stage 3 Results"
+                    )
+
+                with gr.TabItem("📊 Summary"):
+                    summary_output = gr.HTML(
+                        value="<div style='text-align: center; padding: 50px; color: #888;'>Analysis summary will appear here...</div>",
+                        label="Summary"
+                    )
 
     # 示例区域
     gr.Markdown("### 📋 Examples")
@@ -2488,16 +2619,20 @@ with gr.Blocks(
         example2_btn = gr.Button("Example 2: English Product Ad", size="sm")
         example3_btn = gr.Button("Example 3: Mixed Content", size="sm")
 
-    # 事件处理 - 使用流式分析
+    # 事件处理 - 使用四个输出
     submit_btn.click(
         fn=comprehensive_text_audit_streaming,
         inputs=[text_input, keywords_input, country_select, language_select, model_select],
-        outputs=output_html
+        outputs=[stage1_output, stage2_output, stage3_output, summary_output]
     )
 
     clear_btn.click(
-        fn=lambda: ("", "", "China", "简体中文", "Qwen/Qwen3-8B (Free - SiliconFlow)", "<div style='text-align: center; padding: 50px; color: #888;'>Please enter text and click 'Start AI Analysis' to begin.</div>"),
-        outputs=[text_input, keywords_input, country_select, language_select, model_select, output_html]
+        fn=lambda: ("", "", "China", "简体中文", "Qwen/Qwen3-8B (Free - SiliconFlow)",
+                   "<div style='text-align: center; padding: 50px; color: #888;'>Stage 1 results will appear here...</div>",
+                   "<div style='text-align: center; padding: 50px; color: #888;'>Stage 2 results will appear here...</div>",
+                   "<div style='text-align: center; padding: 50px; color: #888;'>Stage 3 results will appear here...</div>",
+                   "<div style='text-align: center; padding: 50px; color: #888;'>Analysis summary will appear here...</div>"),
+        outputs=[text_input, keywords_input, country_select, language_select, model_select, stage1_output, stage2_output, stage3_output, summary_output]
     )
 
     # 示例按钮事件