2

app.py

@@ -8,10 +8,10 @@ class ReviewState:
     def __init__(self):
         self.all_paths = []
         self.current_idx = -1
-    
+
     def sync_paths(self):
         self.all_paths = data_manager.get_all_chart_paths()
-    
+
     def get_nav_target(self, direction):
         new_idx = self.current_idx + direction
         if 0 <= new_idx < len(self.all_paths):
@@ -32,7 +32,6 @@ def to_html_frame(html_content):
 def handle_source_change(source):
     struct = data_manager.get_dataset_structure()
     types = list(struct.get('sources', {}).get(source, {}).get('chart_types', {}).keys())
-    # 默认选中第一个
     return gr.update(choices=types, value=types[0] if types else None)
 
 def handle_type_change(source, c_type):
@@ -46,29 +45,25 @@ def handle_type_change(source, c_type):
 
 def handle_load(source, c_type, c_id, model):
     if not all([source, c_type, c_id, model]):
-        # 必须返回 8 个 update 对象以匹配 outputs
         return [gr.update()] * 8
-    
-    # 获取数据
+
     chart_data = data_manager.get_chart_data(source, c_type, c_id)
     qa_list = data_manager.get_qa_list(source, c_type, model, c_id)
     stats = data_manager.get_review_stats()
-    
-    # 同步索引位置
+
     nav_state.sync_paths()
     for i, p in enumerate(nav_state.all_paths):
         if p['chart_id'] == c_id and p['model'] == model:
             nav_state.current_idx = i
             break
 
-    # 准备 UI 数据
     html_code = to_html_frame(chart_data.get('html_content', ''))
     meta_md = "\n".join([f"- **{k}**: {v}" for k, v in chart_data.get('label_info', {}).items()])
     qa_json = json.dumps([{"id": q.id, "q": q.question, "a": q.answer} for q in qa_list])
     stats_str = f"✅{stats['correct']} | ❌{stats['incorrect']} | 总{stats['total']}"
     prog_str = f"{nav_state.current_idx + 1} / {len(nav_state.all_paths)}"
     radio_choices = [f"Q{i+1}: {q.question[:20]}..." for i, q in enumerate(qa_list)]
-    
+
     return [
         html_code, 
         meta_md, 
@@ -77,7 +72,7 @@ def handle_load(source, c_type, c_id, model):
         prog_str, 
         f"{source}/{c_type}/{c_id}",
         gr.update(choices=radio_choices, value=radio_choices[0] if radio_choices else None),
-        json.dumps({}) # 占位 review_store
+        json.dumps({}) 
     ]
 
 def handle_qa_select(selection, qa_json):
@@ -94,7 +89,6 @@ def handle_qa_select(selection, qa_json):
 # ============== UI 布局 ==============
 def create_ui():
     with gr.Blocks(title="审核系统 V2", theme=gr.themes.Soft()) as demo:
-        # 显式初始化 State，避免布尔值陷阱
         qa_store = gr.State(value="[]")
         review_store = gr.State(value="{}")
         
@@ -118,7 +112,7 @@ def create_ui():
                 
                 with gr.Accordion("元数据解析", open=False):
                     meta_md = gr.Markdown()
-
+                
                 gr.Markdown("---")
                 qa_radio = gr.Radio(label="题目列表", choices=[])
                 
@@ -126,39 +120,43 @@ def create_ui():
                     curr_qid = gr.Text(visible=False)
                     q_disp = gr.Text(label="问题内容", lines=2)
                     a_disp = gr.Text(label="标准答案")
-                    
                     status_opt = gr.Radio(
                         label="审核结论", 
-                        choices=[("正确", "correct"), ("错误", "incorrect"), ("优化", "modified")],
+                        # 【修正】将 "modified" 改为 "needs_modification" 匹配 DataManager 内部逻辑
+                        choices=[("正确", "correct"), ("错误", "incorrect"), ("优化", "needs_modification")],
                         value="correct"
                     )
                     err_type = gr.Dropdown(label="错误分类", choices=["无", "事实错误", "逻辑错误", "图表无法读取"])
-                    
+                
                 comment = gr.Text(label="审核备注")
+                
+                # 【新增】状态提示框，用于反馈保存成功与否
+                save_status = gr.Text(label="操作反馈", interactive=False)
                 save_btn = gr.Button("💾 提交单条审核", variant="primary")
                 
                 with gr.Row():
                     prev_btn = gr.Button("⬅️ 上一个图表")
                     next_btn = gr.Button("➡️ 下一个图表")
 
+                gr.Markdown("---")
+                # 【新增】下载区
+                with gr.Row():
+                    export_btn = gr.Button("📥 导出并下载审核记录", variant="secondary")
+                download_file = gr.File(label="下载文件", visible=False)
+
         # --- 事件绑定 ---
-        
-        # 初始化第一级
         demo.load(
             fn=lambda: gr.update(choices=list(data_manager.get_dataset_structure().get('sources', {}).keys())),
             outputs=[src_dd]
         )
 
-        # 联动逻辑
         src_dd.change(handle_source_change, inputs=[src_dd], outputs=[typ_dd])
         typ_dd.change(handle_type_change, inputs=[src_dd, typ_dd], outputs=[id_dd, mdl_dd])
-        
-        # 加载数据 (严格匹配 8 个 output)
+
         load_event_outputs = [chart_view, meta_md, qa_store, stats_txt, prog_txt, path_info, qa_radio, review_store]
         id_dd.change(handle_load, inputs=[src_dd, typ_dd, id_dd, mdl_dd], outputs=load_event_outputs)
         mdl_dd.change(handle_load, inputs=[src_dd, typ_dd, id_dd, mdl_dd], outputs=load_event_outputs)
 
-        # 题目切换
         qa_radio.change(
             handle_qa_select, 
             inputs=[qa_radio, qa_store], 
@@ -173,36 +171,55 @@ def create_ui():
                     gr.update(value=target['source']),
                     gr.update(value=target['chart_type']),
                     gr.update(value=target['chart_id']),
-                    gr.update(value=target['model'])
+                    gr.update(value=target['model']),
+                    gr.update(value="") # 切换时清空状态
                 ]
-            return [gr.update()] * 4
+            return [gr.update()] * 4 + [gr.update()]
+            
+        prev_btn.click(lambda: navigate(-1), outputs=[src_dd, typ_dd, id_dd, mdl_dd, save_status])
+        next_btn.click(lambda: navigate(1), outputs=[src_dd, typ_dd, id_dd, mdl_dd, save_status])
 
-        prev_btn.click(lambda: navigate(-1), outputs=[src_dd, typ_dd, id_dd, mdl_dd])
-        next_btn.click(lambda: navigate(1), outputs=[src_dd, typ_dd, id_dd, mdl_dd])
-
-        # 保存逻辑
+        # 【修复与增强】保存逻辑
         def quick_save(qid, cid, src, status, cmt):
-            if not qid: return "无效操作"
+            if not qid: 
+                return "⚠️ 无效操作：未选择题目", gr.update()
+            
             data_manager.save_review({
                 "qa_id": qid, "chart_id": cid, "source": src,
                 "status": status, "comment": cmt
             })
-            return "已保存"
+            
+            # 【增强】保存后立刻刷新全局统计信息
+            stats = data_manager.get_review_stats()
+            stats_str = f"✅{stats['correct']} | ❌{stats['incorrect']} | 总{stats['total']}"
+            
+            return f"✅ 保存成功 (ID: {qid})", stats_str
 
         save_btn.click(
             quick_save, 
             inputs=[curr_qid, id_dd, src_dd, status_opt, comment],
-            outputs=[gr.Text(visible=False)]
+            # 【修复】将结果输出到已经实例化的组件上
+            outputs=[save_status, stats_txt]
+        )
+
+        # 【新增】导出下载逻辑
+        def handle_export():
+            file_path = data_manager.export_reviews()
+            return gr.update(value=file_path, visible=True)
+
+        export_btn.click(
+            handle_export,
+            inputs=[],
+            outputs=[download_file]
         )
 
     return demo
 
 if __name__ == "__main__":
     app = create_ui()
-    # 强制禁用所有 API 预览功能
     app.launch(
         server_name="0.0.0.0", 
         server_port=7860, 
         show_api=False,
         max_threads=10
-    )
+    )