2

app.py

@@ -3,7 +3,7 @@ import json
 import base64
 from data_manager import data_manager
 
-# ============== 状态管理 (封装逻辑) ==============
+# ============== 状态管理 ==============
 class ReviewState:
     def __init__(self):
         self.all_paths = []
@@ -28,11 +28,14 @@ def to_html_frame(html_content):
     b64_content = base64.b64encode(html_content.encode('utf-8')).decode('utf-8')
     return f'<iframe src="data:text/html;base64,{b64_content}" style="width:100%;height:600px;border:none;"></iframe>'
 
+def get_stats_display():
+    stats = data_manager.get_review_stats()
+    return f"✅正确: {stats['correct']} | ❌错误: {stats['incorrect']} | 📝优化: {stats.get('modified', 0)} | 总计: {stats['total']}"
+
 # ============== 交互逻辑 ==============
 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,28 +49,23 @@ 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']}"
+    stats_str = get_stats_display()
     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)]
+    radio_choices = [f"Q{i+1}: {q.question[:30]}..." for i, q in enumerate(qa_list)]
     
     return [
         html_code, 
@@ -75,97 +73,123 @@ def handle_load(source, c_type, c_id, model):
         qa_json, 
         stats_str, 
         prog_str, 
-        f"{source}/{c_type}/{c_id}",
+        f"{source} / {c_type} / {c_id}",
         gr.update(choices=radio_choices, value=radio_choices[0] if radio_choices else None),
-        json.dumps({}) # 占位 review_store
+        "{}" 
     ]
 
 def handle_qa_select(selection, qa_json):
     if not selection or not qa_json:
-        return [""] * 8
+        return [""] * 4
     try:
         qas = json.loads(qa_json)
         idx = int(selection.split(":")[0][1:]) - 1
         curr = qas[idx]
-        return [curr['id'], curr['q'], curr['a'], "correct", "无", "", "", ""]
+        return [curr['id'], curr['q'], curr['a'], ""]
     except:
-        return [""] * 8
+        return [""] * 4
 
 # ============== UI 布局 ==============
 def create_ui():
-    with gr.Blocks(title="审核系统 V2", theme=gr.themes.Soft()) as demo:
-        # 显式初始化 State，避免布尔值陷阱
+    with gr.Blocks(title="图表审核系统 V2", theme=gr.themes.Soft()) as demo:
         qa_store = gr.State(value="[]")
-        review_store = gr.State(value="{}")
-        
-        gr.Markdown("## 📑 图表问答数据集审核终端")
         
+        with gr.Row():
+            gr.Markdown("# 📊 图表问答数据集审核终端")
+            with gr.Column(min_width=200):
+                export_btn = gr.Button("📤 导出审核记录", variant="secondary")
+                download_file = gr.File(label="点击下载导出的文件", visible=False)
+
         with gr.Row():
             with gr.Column(scale=4):
                 with gr.Row():
-                    src_dd = gr.Dropdown(label="数据源", choices=["None"])
-                    typ_dd = gr.Dropdown(label="图表类型")
-                    id_dd = gr.Dropdown(label="图表 ID")
-                    mdl_dd = gr.Dropdown(label="评估模型")
+                    src_dd = gr.Dropdown(label="1. 数据源")
+                    typ_dd = gr.Dropdown(label="2. 图表类型")
+                    id_dd = gr.Dropdown(label="3. 图表 ID")
+                    mdl_dd = gr.Dropdown(label="4. 评估模型")
                 
-                chart_view = gr.HTML(value='<div style="height:500px; background:#f0f0f0;"></div>')
-                path_info = gr.Text(label="当前路径", interactive=False)
+                chart_view = gr.HTML(value='<div style="height:550px; background:#f9f9f9; border-radius:8px;"></div>')
+                path_info = gr.Text(label="当前数据路径", interactive=False)
                 
             with gr.Column(scale=2):
                 with gr.Group():
-                    stats_txt = gr.Text(label="统计信息", interactive=False)
-                    prog_txt = gr.Text(label="审核进度", interactive=False)
+                    stats_txt = gr.Text(label="全库审核统计", value=get_stats_display(), interactive=False)
+                    prog_txt = gr.Text(label="当前库进度", interactive=False)
                 
-                with gr.Accordion("元数据解析", open=False):
-                    meta_md = gr.Markdown()
+                with gr.Accordion("图表元数据 (Meta)", open=False):
+                    meta_md = gr.Markdown("等待加载...")
 
                 gr.Markdown("---")
-                qa_radio = gr.Radio(label="题目列表", choices=[])
+                qa_radio = gr.Radio(label="题目快速切换", choices=[])
                 
                 with gr.Group():
-                    curr_qid = gr.Text(visible=False)
-                    q_disp = gr.Text(label="问题内容", lines=2)
-                    a_disp = gr.Text(label="标准答案")
+                    curr_qid = gr.Text(label="QA ID", interactive=False, visible=True)
+                    q_disp = gr.Text(label="问题 (Question)", lines=3)
+                    a_disp = gr.Text(label="答案 (Answer)", lines=3)
                     
                     status_opt = gr.Radio(
                         label="审核结论", 
-                        choices=[("正确", "correct"), ("错误", "incorrect"), ("优化", "modified")],
+                        choices=[("✅ 正确", "correct"), ("❌ 错误", "incorrect"), ("🔧 需优化", "modified")],
                         value="correct"
                     )
-                    err_type = gr.Dropdown(label="错误分类", choices=["无", "事实错误", "逻辑错误", "图表无法读取"])
+                    err_type = gr.Dropdown(label="错误分类 (可选)", choices=["无", "事实错误", "逻辑错误", "图表无法读取", "其他"])
                     
-                comment = gr.Text(label="审核备注")
+                comment = gr.Text(label="审核备注 (选填)")
                 save_btn = gr.Button("💾 提交单条审核", variant="primary")
                 
                 with gr.Row():
-                    prev_btn = gr.Button("⬅️ 上一个图表")
-                    next_btn = gr.Button("➡️ 下一个图表")
+                    prev_btn = gr.Button("⬅️ 上一图表")
+                    next_btn = gr.Button("➡️ 下一图表")
 
         # --- 事件绑定 ---
-        
-        # 初始化第一级
+
+        # 1. 初始加载
         demo.load(
             fn=lambda: gr.update(choices=list(data_manager.get_dataset_structure().get('sources', {}).keys())),
             outputs=[src_dd]
         )
 
-        # 联动逻辑
+        # 2. 联动逻辑
         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]
+        load_event_outputs = [chart_view, meta_md, qa_store, stats_txt, prog_txt, path_info, qa_radio, comment]
         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)
 
-        # 题目切换
+        # 3. 题目切换
         qa_radio.change(
             handle_qa_select, 
             inputs=[qa_radio, qa_store], 
-            outputs=[curr_qid, q_disp, a_disp, status_opt, err_type, gr.State(), gr.State(), comment] 
+            outputs=[curr_qid, q_disp, a_disp, comment] 
+        )
+
+        # 4. 保存功能 (带反馈和刷新)
+        def process_save(qid, cid, src, status, e_type, cmt, model, c_type):
+            if not qid:
+                gr.Warning("请先选择一条题目！")
+                return get_stats_display()
+            
+            data_manager.save_review({
+                "qa_id": qid,
+                "chart_id": cid,
+                "source": src,
+                "chart_type": c_type,
+                "model": model,
+                "status": status,
+                "issue_type": e_type,
+                "comment": cmt
+            })
+            gr.Info(f"保存成功: {qid}")
+            return get_stats_display()
+
+        save_btn.click(
+            process_save, 
+            inputs=[curr_qid, id_dd, src_dd, status_opt, err_type, comment, mdl_dd, typ_dd],
+            outputs=[stats_txt] # 保存后刷新统计数字
         )
 
-        # 导航逻辑
+        # 5. 导航逻辑
         def navigate(direction):
             target = nav_state.get_nav_target(direction)
             if target:
@@ -175,34 +199,26 @@ def create_ui():
                     gr.update(value=target['chart_id']),
                     gr.update(value=target['model'])
                 ]
+            gr.Warning("已到达边界")
             return [gr.update()] * 4
 
         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 "无效操作"
-            data_manager.save_review({
-                "qa_id": qid, "chart_id": cid, "source": src,
-                "status": status, "comment": cmt
-            })
-            return "已保存"
+        # 6. 导出功能
+        def handle_export():
+            file_path = data_manager.export_reviews()
+            return gr.update(value=file_path, visible=True)
 
-        save_btn.click(
-            quick_save, 
-            inputs=[curr_qid, id_dd, src_dd, status_opt, comment],
-            outputs=[gr.Text(visible=False)]
-        )
+        export_btn.click(handle_export, outputs=[download_file])
 
     return demo
 
 if __name__ == "__main__":
+    # 关闭所有可能干扰的 API 解析
     app = create_ui()
-    # 强制禁用所有 API 预览功能
     app.launch(
         server_name="0.0.0.0", 
         server_port=7860, 
-        show_api=False,
-        max_threads=10
+        show_api=False
     )