ocr_verify/table_line_generator/streamlit_table_line_editor.py

"""
表格线可视化编辑器
支持人工调整表格线位置
"""

import streamlit as st
from pathlib import Path
import json
from PIL import Image, ImageDraw, ImageFont
import numpy as np
import copy

try:
    from .table_line_generator import TableLineGenerator
except ImportError:
    from table_line_generator import TableLineGenerator


def parse_ocr_data(ocr_data):
    """解析OCR数据，支持多种格式"""
    # 如果是字符串，尝试解析
    if isinstance(ocr_data, str):
        try:
            ocr_data = json.loads(ocr_data)
        except json.JSONDecodeError:
            st.error("❌ JSON 格式错误，无法解析")
            return []
    
    # 检查是否为 PPStructure V3 格式
    if isinstance(ocr_data, dict) and 'parsing_res_list' in ocr_data and 'overall_ocr_res' in ocr_data:
        st.info("🔍 检测到 PPStructure V3 格式")
        
        try:
            table_bbox, text_boxes = TableLineGenerator.parse_ppstructure_result(ocr_data)
            st.success(f"✅ 表格区域: {table_bbox}")
            st.success(f"✅ 表格内文本框: {len(text_boxes)} 个")
            return text_boxes
        except Exception as e:
            st.error(f"❌ 解析 PPStructure 结果失败: {e}")
            return []
    
    # 确保是列表
    if not isinstance(ocr_data, list):
        st.error(f"❌ OCR 数据应该是列表，实际类型: {type(ocr_data)}")
        return []
    
    if not ocr_data:
        st.warning("⚠️ OCR 数据为空")
        return []
    
    first_item = ocr_data[0]
    if not isinstance(first_item, dict):
        st.error(f"❌ OCR 数据项应该是字典，实际类型: {type(first_item)}")
        return []
    
    if 'bbox' not in first_item:
        st.error("❌ OCR 数据缺少 'bbox' 字段")
        st.info("💡 支持的格式示例:\n```json\n[\n  {\n    \"text\": \"文本\",\n    \"bbox\": [x1, y1, x2, y2]\n  }\n]\n```")
        return []
    
    return ocr_data


def draw_table_lines_with_numbers(image, structure, line_width=2, show_numbers=True):
    """
    绘制带编号的表格线（使用线坐标列表）
    
    Args:
        image: PIL Image 对象
        structure: 表格结构字典（包含 horizontal_lines 和 vertical_lines）
        line_width: 线条宽度
        show_numbers: 是否显示编号
    
    Returns:
        绘制了表格线和编号的图片
    """
    img_with_lines = image.copy()
    draw = ImageDraw.Draw(img_with_lines)
    
    # 尝试加载字体
    try:
        font = ImageFont.truetype("/System/Library/Fonts/Helvetica.ttc", 20)
    except:
        font = ImageFont.load_default()
    
    # 🆕 使用线坐标列表
    horizontal_lines = structure.get('horizontal_lines', [])
    vertical_lines = structure.get('vertical_lines', [])
    modified_h_lines = structure.get('modified_h_lines', set())
    modified_v_lines = structure.get('modified_v_lines', set())
    
    # 计算绘制范围
    x_start = vertical_lines[0] if vertical_lines else 0
    x_end = vertical_lines[-1] if vertical_lines else img_with_lines.width
    y_start = horizontal_lines[0] if horizontal_lines else 0
    y_end = horizontal_lines[-1] if horizontal_lines else img_with_lines.height
    
    # 🎨 绘制横线
    for idx, y in enumerate(horizontal_lines):
        color = (255, 0, 0) if idx in modified_h_lines else (0, 0, 255)
        draw.line([(x_start, y), (x_end, y)], fill=color, width=line_width)
        
        # 🔢 绘制行编号
        if show_numbers:
            text = f"R{idx+1}"
            bbox = draw.textbbox((x_start - 35, y - 10), text, font=font)
            draw.rectangle(bbox, fill='white', outline='black')
            draw.text((x_start - 35, y - 10), text, fill=color, font=font)
    
    # 🎨 绘制竖线
    for idx, x in enumerate(vertical_lines):
        color = (255, 0, 0) if idx in modified_v_lines else (0, 0, 255)
        draw.line([(x, y_start), (x, y_end)], fill=color, width=line_width)
        
        # 🔢 绘制列编号
        if show_numbers:
            text = f"C{idx+1}"
            bbox = draw.textbbox((x - 10, y_start - 25), text, font=font)
            draw.rectangle(bbox, fill='white', outline='black')
            draw.text((x - 10, y_start - 25), text, fill=color, font=font)
            bbox = draw.textbbox((x - 10, y_end + 25), text, font=font)
            draw.rectangle(bbox, fill='white', outline='black')
            draw.text((x - 10, y_end + 25), text, fill=color, font=font)
    
    return img_with_lines


# 🆕 新增：用于保存的纯净表格线绘制函数
def draw_clean_table_lines(image, structure, line_width=2, line_color=(0, 0, 0)):
    """
    绘制纯净的表格线（用于保存）
    - 所有线用黑色
    - 不显示编号
    
    Args:
        image: PIL Image 对象
        structure: 表格结构字典
        line_width: 线条宽度
        line_color: 线条颜色，默认黑色 (0, 0, 0)
    
    Returns:
        绘制了纯净表格线的图片
    """
    img_with_lines = image.copy()
    draw = ImageDraw.Draw(img_with_lines)
    
    horizontal_lines = structure.get('horizontal_lines', [])
    vertical_lines = structure.get('vertical_lines', [])
    
    if not horizontal_lines or not vertical_lines:
        return img_with_lines
    
    # 计算绘制范围
    x_start = vertical_lines[0]
    x_end = vertical_lines[-1]
    y_start = horizontal_lines[0]
    y_end = horizontal_lines[-1]
    
    # 🖤 绘制横线（统一黑色）
    for y in horizontal_lines:
        draw.line([(x_start, y), (x_end, y)], fill=line_color, width=line_width)
    
    # 🖤 绘制竖线（统一黑色）
    for x in vertical_lines:
        draw.line([(x, y_start), (x, y_end)], fill=line_color, width=line_width)
    
    return img_with_lines


def init_undo_stack():
    """初始化撤销/重做栈"""
    if 'undo_stack' not in st.session_state:
        st.session_state.undo_stack = []
    if 'redo_stack' not in st.session_state:
        st.session_state.redo_stack = []


def save_state_for_undo(structure):
    """保存当前状态到撤销栈"""
    # 深拷贝当前结构
    state = copy.deepcopy(structure)
    st.session_state.undo_stack.append(state)
    # 清空重做栈
    st.session_state.redo_stack = []
    
    # 限制栈深度（最多保存20个历史状态）
    if len(st.session_state.undo_stack) > 20:
        st.session_state.undo_stack.pop(0)


def undo_last_action():
    """撤销上一个操作"""
    if st.session_state.undo_stack:
        # 保存当前状态到重做栈
        current_state = copy.deepcopy(st.session_state.structure)
        st.session_state.redo_stack.append(current_state)
        
        # 恢复上一个状态
        st.session_state.structure = st.session_state.undo_stack.pop()
        return True
    return False


def redo_last_action():
    """重做上一个操作"""
    if st.session_state.redo_stack:
        # 保存当前状态到撤销栈
        current_state = copy.deepcopy(st.session_state.structure)
        st.session_state.undo_stack.append(current_state)
        
        # 恢复重做的状态
        st.session_state.structure = st.session_state.redo_stack.pop()
        return True
    return False


def get_structure_hash(structure, line_width, show_numbers):
    """生成结构的哈希值，用于判断是否需要重新绘制"""
    import hashlib
    
    # 🔧 使用线坐标列表生成哈希
    key_data = {
        'horizontal_lines': structure.get('horizontal_lines', []),
        'vertical_lines': structure.get('vertical_lines', []),
        'modified_h_lines': sorted(list(structure.get('modified_h_lines', set()))),
        'modified_v_lines': sorted(list(structure.get('modified_v_lines', set()))),
        'line_width': line_width,
        'show_numbers': show_numbers
    }
    
    key_str = json.dumps(key_data, sort_keys=True)
    return hashlib.md5(key_str.encode()).hexdigest()


def get_cached_table_lines_image(image, structure, line_width, show_numbers):
    """
    获取缓存的表格线图片，如果缓存不存在或失效则重新绘制
    
    Args:
        image: PIL Image 对象
        structure: 表格结构字典
        line_width: 线条宽度
        show_numbers: 是否显示编号
    
    Returns:
        绘制了表格线和编号的图片
    """
    # 初始化缓存
    if 'cached_table_image' not in st.session_state:
        st.session_state.cached_table_image = None
    if 'cached_table_hash' not in st.session_state:
        st.session_state.cached_table_hash = None
    
    # 计算当前结构的哈希
    current_hash = get_structure_hash(structure, line_width, show_numbers)
    
    # 检查缓存是否有效
    if (st.session_state.cached_table_hash == current_hash and 
        st.session_state.cached_table_image is not None):
        # 缓存有效，直接返回
        return st.session_state.cached_table_image
    
    # 缓存失效，重新绘制
    img_with_lines = draw_table_lines_with_numbers(
        image, 
        structure, 
        line_width=line_width,
        show_numbers=show_numbers
    )
    
    # 更新缓存
    st.session_state.cached_table_image = img_with_lines
    st.session_state.cached_table_hash = current_hash
    
    return img_with_lines


def clear_table_image_cache():
    """清除表格图片缓存"""
    if 'cached_table_image' in st.session_state:
        st.session_state.cached_table_image = None
    if 'cached_table_hash' in st.session_state:
        st.session_state.cached_table_hash = None


def load_structure_from_config(config_path: Path) -> dict:
    """
    从配置文件加载表格结构
    
    Args:
        config_path: 配置文件路径
    
    Returns:
        表格结构字典
    """
    with open(config_path, 'r', encoding='utf-8') as f:
        structure = json.load(f)
    
    # 🔧 兼容旧版配置（补充缺失字段）
    if 'horizontal_lines' not in structure:
        # 从 rows 生成横线坐标
        horizontal_lines = []
        for row in structure.get('rows', []):
            horizontal_lines.append(row['y_start'])
        if structure.get('rows'):
            horizontal_lines.append(structure['rows'][-1]['y_end'])
        structure['horizontal_lines'] = horizontal_lines
    
    if 'vertical_lines' not in structure:
        # 从 columns 生成竖线坐标
        vertical_lines = []
        for col in structure.get('columns', []):
            vertical_lines.append(col['x_start'])
        if structure.get('columns'):
            vertical_lines.append(structure['columns'][-1]['x_end'])
        structure['vertical_lines'] = vertical_lines
    
    # 🔧 转换修改标记（从列表转为集合）
    if 'modified_h_lines' in structure:
        structure['modified_h_lines'] = set(structure['modified_h_lines'])
    else:
        structure['modified_h_lines'] = set()
    
    if 'modified_v_lines' in structure:
        structure['modified_v_lines'] = set(structure['modified_v_lines'])
    else:
        structure['modified_v_lines'] = set()
    
    # 🔧 转换旧版的 modified_rows/modified_cols（如果存在）
    if 'modified_rows' in structure and not structure['modified_h_lines']:
        structure['modified_h_lines'] = set(structure.get('modified_rows', []))
    if 'modified_cols' in structure and not structure['modified_v_lines']:
        structure['modified_v_lines'] = set(structure.get('modified_cols', []))
    
    return structure


def create_table_line_editor():
    """创建表格线编辑器界面"""
    # 🆕 配置页面为宽屏模式
    st.set_page_config(
        page_title="表格线编辑器",
        page_icon="📏",
        layout="wide",
        initial_sidebar_state="expanded"
    )
    
    st.title("📏 表格线编辑器")
    
    # 初始化 session_state
    if 'loaded_json_name' not in st.session_state:
        st.session_state.loaded_json_name = None
    if 'loaded_image_name' not in st.session_state:
        st.session_state.loaded_image_name = None
    if 'loaded_config_name' not in st.session_state:
        st.session_state.loaded_config_name = None
    if 'ocr_data' not in st.session_state:
        st.session_state.ocr_data = None
    if 'image' not in st.session_state:
        st.session_state.image = None
    
    # 初始化撤销/重做栈
    init_undo_stack()
    
    # 🆕 添加工作模式选择
    st.sidebar.header("📂 工作模式")
    work_mode = st.sidebar.radio(
        "选择模式",
        ["🆕 新建标注", "📂 加载已有标注"],
        index=0
    )
    
    if work_mode == "🆕 新建标注":
        # 原有的上传流程
        st.sidebar.subheader("上传文件")
        uploaded_json = st.sidebar.file_uploader("上传OCR结果JSON", type=['json'], key="new_json")
        uploaded_image = st.sidebar.file_uploader("上传对应图片", type=['jpg', 'png'], key="new_image")
        
        # 检查是否需要重新加载 JSON
        if uploaded_json is not None:
            if st.session_state.loaded_json_name != uploaded_json.name:
                try:
                    raw_data = json.load(uploaded_json)
                    
                    with st.expander("🔍 原始数据结构"):
                        if isinstance(raw_data, dict):
                            st.json({k: f"<{type(v).__name__}>" if not isinstance(v, (str, int, float, bool, type(None))) else v 
                                    for k, v in list(raw_data.items())[:5]})
                        else:
                            st.json(raw_data[:3] if len(raw_data) > 3 else raw_data)
                    
                    ocr_data = parse_ocr_data(raw_data)
                    
                    if not ocr_data:
                        st.error("❌ 无法解析 OCR 数据，请检查 JSON 格式")
                        st.stop()
                    
                    st.session_state.ocr_data = ocr_data
                    st.session_state.loaded_json_name = uploaded_json.name
                    st.session_state.loaded_config_name = None  # 清除配置文件标记
                    
                    # 清除旧的分析结果、历史记录和缓存
                    if 'structure' in st.session_state:
                        del st.session_state.structure
                    if 'generator' in st.session_state:
                        del st.session_state.generator
                    st.session_state.undo_stack = []
                    st.session_state.redo_stack = []
                    clear_table_image_cache()
                    
                    st.success(f"✅ 成功加载 {len(ocr_data)} 条 OCR 记录")
                    
                except Exception as e:
                    st.error(f"❌ 加载数据失败: {e}")
                    st.stop()
        
        # 检查是否需要重新加载图片
        if uploaded_image is not None:
            if st.session_state.loaded_image_name != uploaded_image.name:
                try:
                    image = Image.open(uploaded_image)
                    
                    st.session_state.image = image
                    st.session_state.loaded_image_name = uploaded_image.name
                    
                    if 'structure' in st.session_state:
                        del st.session_state.structure
                    if 'generator' in st.session_state:
                        del st.session_state.generator
                    st.session_state.undo_stack = []
                    st.session_state.redo_stack = []
                    clear_table_image_cache()
                    
                    st.success(f"✅ 成功加载图片: {uploaded_image.name}")
                    
                except Exception as e:
                    st.error(f"❌ 加载图片失败: {e}")
                    st.stop()
    
    else:  # 📂 加载已有标注
        st.sidebar.subheader("加载已保存的标注")
        
        # 🆕 上传配置文件
        uploaded_config = st.sidebar.file_uploader(
            "上传配置文件 (*_structure.json)",
            type=['json'],
            key="load_config"
        )
        
        # 🆕 上传对应的图片（可选，用于重新标注）
        uploaded_image_for_config = st.sidebar.file_uploader(
            "上传对应图片（可选）",
            type=['jpg', 'png'],
            key="load_image"
        )
        
        # 处理配置文件加载
        if uploaded_config is not None:
            if st.session_state.loaded_config_name != uploaded_config.name:
                try:
                    # 🔧 直接从配置文件路径加载
                    import tempfile
                    
                    # 创建临时文件
                    with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False, encoding='utf-8') as tmp:
                        tmp.write(uploaded_config.getvalue().decode('utf-8'))
                        tmp_path = tmp.name
                    
                    # 加载结构
                    structure = load_structure_from_config(Path(tmp_path))
                    
                    # 清理临时文件
                    Path(tmp_path).unlink()
                    
                    st.session_state.structure = structure
                    st.session_state.loaded_config_name = uploaded_config.name
                    
                    # 清除历史记录和缓存
                    st.session_state.undo_stack = []
                    st.session_state.redo_stack = []
                    clear_table_image_cache()
                    
                    st.success(f"✅ 成功加载配置: {uploaded_config.name}")
                    st.info(
                        f"📊 表格结构: {len(structure['rows'])}行 x {len(structure['columns'])}列\n\n"
                        f"📏 横线数: {len(structure.get('horizontal_lines', []))}\n\n"
                        f"📏 竖线数: {len(structure.get('vertical_lines', []))}"
                    )
                    
                    # 🆕 显示配置文件详情
                    with st.expander("📋 配置详情"):
                        st.json({
                            "行数": len(structure['rows']),
                            "列数": len(structure['columns']),
                            "横线数": len(structure.get('horizontal_lines', [])),
                            "竖线数": len(structure.get('vertical_lines', [])),
                            "行高": structure.get('row_height'),
                            "列宽": structure.get('col_widths'),
                            "已修改的横线": list(structure.get('modified_h_lines', set())),
                            "已修改的竖线": list(structure.get('modified_v_lines', set()))
                        })
                    
                except Exception as e:
                    st.error(f"❌ 加载配置失败: {e}")
                    import traceback
                    st.code(traceback.format_exc())
                    st.stop()
        
        # 处理图片加载（用于显示）
        if uploaded_image_for_config is not None:
            if st.session_state.loaded_image_name != uploaded_image_for_config.name:
                try:
                    image = Image.open(uploaded_image_for_config)
                    st.session_state.image = image
                    st.session_state.loaded_image_name = uploaded_image_for_config.name
                    
                    clear_table_image_cache()
                    
                    st.success(f"✅ 成功加载图片: {uploaded_image_for_config.name}")
                    
                except Exception as e:
                    st.error(f"❌ 加载图片失败: {e}")
                    st.stop()
        
        # 🆕 如果配置已加载但没有图片，提示用户
        if 'structure' in st.session_state and st.session_state.image is None:
            st.warning("⚠️ 已加载配置，但未加载对应图片。请上传图片以查看效果。")
            st.info("💡 提示：配置文件已加载，您可以：\n1. 上传对应图片查看效果\n2. 直接编辑配置并保存")
    
    # 检查必要条件
    if work_mode == "🆕 新建标注":
        if st.session_state.ocr_data is None or st.session_state.image is None:
            st.info("👆 请在左侧上传 OCR 结果 JSON 文件和对应的图片")
            
            with st.expander("📖 使用说明"):
                st.markdown("""
                ### 🆕 新建标注模式
                
                **支持的OCR格式**
                
                **1. PPStructure V3 格式 (推荐)**
                ```json
                {
                  "parsing_res_list": [...],
                  "overall_ocr_res": {
                    "rec_boxes": [[x1, y1, x2, y2], ...],
                    "rec_texts": ["文本1", "文本2", ...]
                  }
                }
                ```
                
                **2. 标准格式**
                ```json
                [
                  {
                    "text": "文本内容",
                    "bbox": [x1, y1, x2, y2]
                  }
                ]
                ```
                
                ### 📂 加载已有标注模式
                
                1. 上传之前保存的 `*_structure.json` 配置文件
                2. 上传对应的图片（可选）
                3. 继续调整表格线位置
                4. 保存更新后的配置
                """)
            return
        
        ocr_data = st.session_state.ocr_data
        image = st.session_state.image
        
        st.info(f"📂 已加载: {st.session_state.loaded_json_name} + {st.session_state.loaded_image_name}")
        
        if 'generator' not in st.session_state or st.session_state.generator is None:
            try:
                generator = TableLineGenerator(image, ocr_data)
                st.session_state.generator = generator
            except Exception as e:
                st.error(f"❌ 初始化失败: {e}")
                st.stop()
    
    else:  # 加载已有标注模式
        if 'structure' not in st.session_state:
            st.info("👆 请在左侧上传配置文件 (*_structure.json)")
            
            with st.expander("📖 使用说明"):
                st.markdown("""
                ### 📂 加载已有标注
                
                **步骤：**
                
                1. **上传配置文件**：选择之前保存的 `*_structure.json`
                2. **上传图片**（可选）：上传对应的图片以查看效果
                3. **调整表格线**：使用下方的工具调整横线/竖线位置
                4. **保存更新**：保存修改后的配置
                
                **提示：**
                - 即使没有图片，也可以直接编辑配置文件中的坐标
                - 配置文件包含完整的表格结构信息
                - 可以应用到同类型的其他页面
                """)
            return
        
        if st.session_state.image is None:
            st.warning("⚠️ 仅加载了配置，未加载图片。部分功能受限。")
        
        # 🆕 使用配置中的信息
        structure = st.session_state.structure
        image = st.session_state.image
        
        if image is None:
            # 如果没有图片，创建一个虚拟的空白图片用于显示坐标信息
            if 'table_bbox' in structure:
                bbox = structure['table_bbox']
                dummy_width = bbox[2] + 100
                dummy_height = bbox[3] + 100
            else:
                dummy_width = 2000
                dummy_height = 2000
            
            image = Image.new('RGB', (dummy_width, dummy_height), color='white')
            st.info(f"💡 使用虚拟画布 ({dummy_width}x{dummy_height}) 显示表格结构")
    
    # 显示设置
    st.sidebar.divider()
    st.sidebar.subheader("🖼️ 显示设置")
    
    line_width = st.sidebar.slider("线条宽度", 1, 5, 2)
    display_mode = st.sidebar.radio("显示模式", ["对比显示", "仅显示划线图", "仅显示原图"], index=1)
    zoom_level = st.sidebar.slider("图片缩放", 0.25, 2.0, 1.0, 0.25)
    show_line_numbers = st.sidebar.checkbox("显示线条编号", value=True)
    
    # 撤销/重做按钮
    st.sidebar.divider()
    st.sidebar.subheader("↩️ 撤销/重做")
    
    col1, col2 = st.sidebar.columns(2)
    with col1:
        if st.button("↩️ 撤销", disabled=len(st.session_state.undo_stack) == 0):
            if undo_last_action():
                clear_table_image_cache()
                st.success("✅ 已撤销")
                st.rerun()
    
    with col2:
        if st.button("↪️ 重做", disabled=len(st.session_state.redo_stack) == 0):
            if redo_last_action():
                clear_table_image_cache()
                st.success("✅ 已重做")
                st.rerun()
    
    st.sidebar.info(f"📚 历史记录: {len(st.session_state.undo_stack)} 条")
    
    # 分析表格结构（仅在新建模式显示）
    if work_mode == "🆕 新建标注" and st.button("🔍 分析表格结构"):
        with st.spinner("分析中..."):
            try:
                generator = st.session_state.generator
                structure = generator.analyze_table_structure(
                    y_tolerance=y_tolerance,
                    x_tolerance=x_tolerance,
                    min_row_height=min_row_height
                )
                
                if not structure:
                    st.warning("⚠️ 未检测到表格结构")
                    st.stop()
                
                structure['modified_h_lines'] = set()
                structure['modified_v_lines'] = set()
                
                st.session_state.structure = structure
                
                st.session_state.undo_stack = []
                st.session_state.redo_stack = []
                clear_table_image_cache()
                
                st.success(
                    f"✅ 检测到 {len(structure['rows'])} 行（{len(structure['horizontal_lines'])} 条横线），"
                    f"{len(structure['columns'])} 列（{len(structure['vertical_lines'])} 条竖线）"
                )
                
                col1, col2, col3, col4 = st.columns(4)
                with col1:
                    st.metric("行数", len(structure['rows']))
                with col2:
                    st.metric("横线数", len(structure['horizontal_lines']))
                with col3:
                    st.metric("列数", len(structure['columns']))
                with col4:
                    st.metric("竖线数", len(structure['vertical_lines']))
            
            except Exception as e:
                st.error(f"❌ 分析失败: {e}")
                import traceback
                st.code(traceback.format_exc())
                st.stop()
    
    # 显示结果（两种模式通用）
    if 'structure' in st.session_state and st.session_state.structure:
        structure = st.session_state.structure
        
        # 使用缓存机制绘制表格线
        img_with_lines = get_cached_table_lines_image(
            image, 
            structure, 
            line_width=line_width,
            show_numbers=show_line_numbers
        )
        
        # 根据显示模式显示图片
        if display_mode == "对比显示":
            col1, col2 = st.columns(2)
            with col1:
                st.subheader("原图")
                st.image(image, use_container_width=True)
            
            with col2:
                st.subheader("添加表格线")
                st.image(img_with_lines, use_container_width=True)
                
        elif display_mode == "仅显示划线图":
            display_width = int(img_with_lines.width * zoom_level)
            
            st.subheader(f"表格线图 (缩放: {zoom_level:.0%})")
            st.image(img_with_lines, width=display_width)
            
        else:
            display_width = int(image.width * zoom_level)
            
            st.subheader(f"原图 (缩放: {zoom_level:.0%})")
            st.image(image, width=display_width)
        
        # 显示详细信息
        with st.expander("📊 表格结构详情"):
            st.json({
                "行数": len(structure['rows']),
                "列数": len(structure['columns']),
                "横线数": len(structure.get('horizontal_lines', [])),
                "竖线数": len(structure.get('vertical_lines', [])),
                "横线坐标": structure.get('horizontal_lines', []),
                "竖线坐标": structure.get('vertical_lines', []),
                "标准行高": structure.get('row_height'),
                "列宽度": structure.get('col_widths'),
                "修改的横线": list(structure.get('modified_h_lines', set())),
                "修改的竖线": list(structure.get('modified_v_lines', set()))
            })
        
        # 🆕 手动调整 - 使用线坐标列表
        st.subheader("🛠️ 手动调整")
        
        adjust_type = st.radio(
            "调整类型",
            ["调整横线", "调整竖线", "添加横线", "删除横线", "添加竖线", "删除竖线"],
            horizontal=True
        )
        
        if adjust_type == "调整横线":
            horizontal_lines = structure.get('horizontal_lines', [])
            if len(horizontal_lines) > 0:
                line_index = st.selectbox(
                    "选择横线",
                    range(len(horizontal_lines)),
                    format_func=lambda x: f"第 {x+1} 条横线 (Y: {horizontal_lines[x]}) {'🔴已修改' if x in structure.get('modified_h_lines', set()) else ''}"
                )
                
                new_y = st.number_input(
                    "新的Y坐标",
                    value=int(horizontal_lines[line_index]),
                    step=1
                )
                
                if st.button("应用调整"):
                    save_state_for_undo(structure)
                    
                    structure['horizontal_lines'][line_index] = new_y
                    structure['modified_h_lines'].add(line_index)
                    
                    # 🔧 同步更新 rows
                    if line_index < len(structure['rows']):
                        structure['rows'][line_index]['y_start'] = new_y
                    if line_index > 0:
                        structure['rows'][line_index - 1]['y_end'] = new_y
                    
                    clear_table_image_cache()
                    st.success("✅ 已调整")
                    st.rerun()
            else:
                st.warning("⚠️ 没有检测到横线")
        
        elif adjust_type == "调整竖线":
            vertical_lines = structure.get('vertical_lines', [])
            if len(vertical_lines) > 0:
                line_index = st.selectbox(
                    "选择竖线",
                    range(len(vertical_lines)),
                    format_func=lambda x: f"第 {x+1} 条竖线 (X: {vertical_lines[x]}) {'🔴已修改' if x in structure.get('modified_v_lines', set()) else ''}"
                )
                
                new_x = st.number_input(
                    "新的X坐标",
                    value=int(vertical_lines[line_index]),
                    step=1
                )
                
                if st.button("应用调整"):
                    save_state_for_undo(structure)
                    
                    structure['vertical_lines'][line_index] = new_x
                    structure['modified_v_lines'].add(line_index)
                    
                    # 🔧 同步更新 columns
                    if line_index < len(structure['columns']):
                        structure['columns'][line_index]['x_start'] = new_x
                    if line_index > 0:
                        structure['columns'][line_index - 1]['x_end'] = new_x
                    
                    clear_table_image_cache()
                    st.success("✅ 已调整")
                    st.rerun()
            else:
                st.warning("⚠️ 没有检测到竖线")
        
        elif adjust_type == "删除横线":
            horizontal_lines = structure.get('horizontal_lines', [])
            if len(horizontal_lines) > 0:
                lines_to_delete = st.multiselect(
                    "选择要删除的横线（可多选）",
                    range(len(horizontal_lines)),
                    format_func=lambda x: f"第 {x+1} 条横线 (Y: {horizontal_lines[x]}) {'🔴已修改' if x in structure.get('modified_h_lines', set()) else ''}"
                )
                
                if lines_to_delete and st.button("🗑️ 批量删除", type="primary"):
                    save_state_for_undo(structure)
                    
                    # 🔧 删除线坐标
                    for idx in sorted(lines_to_delete, reverse=True):
                        del structure['horizontal_lines'][idx]
                    
                    # 🔧 重新计算 rows（删除线后重建行区间）
                    new_rows = []
                    for i in range(len(structure['horizontal_lines']) - 1):
                        new_rows.append({
                            'y_start': structure['horizontal_lines'][i],
                            'y_end': structure['horizontal_lines'][i + 1],
                            # 'bboxes': []
                        })
                    structure['rows'] = new_rows
                    
                    # 更新修改标记
                    structure['modified_h_lines'] = set()
                    
                    clear_table_image_cache()
                    st.success(f"✅ 已删除 {len(lines_to_delete)} 条横线")
                    st.rerun()
                
                st.info(f"💡 当前有 {len(horizontal_lines)} 条横线，已选择 {len(lines_to_delete)} 条")
            else:
                st.warning("⚠️ 没有可删除的横线")
        
        elif adjust_type == "删除竖线":
            vertical_lines = structure.get('vertical_lines', [])
            if len(vertical_lines) > 0:
                lines_to_delete = st.multiselect(
                    "选择要删除的竖线（可多选）",
                    range(len(vertical_lines)),
                    format_func=lambda x: f"第 {x+1} 条竖线 (X: {vertical_lines[x]}) {'🔴已修改' if x in structure.get('modified_v_lines', set()) else ''}"
                )
                
                if lines_to_delete and st.button("🗑️ 批量删除", type="primary"):
                    save_state_for_undo(structure)
                    
                    # 🔧 删除线坐标
                    for idx in sorted(lines_to_delete, reverse=True):
                        del structure['vertical_lines'][idx]
                    
                    # 🔧 重新计算 columns
                    new_columns = []
                    for i in range(len(structure['vertical_lines']) - 1):
                        new_columns.append({
                            'x_start': structure['vertical_lines'][i],
                            'x_end': structure['vertical_lines'][i + 1]
                        })
                    structure['columns'] = new_columns
                    
                    # 重新计算列宽
                    structure['col_widths'] = [
                        col['x_end'] - col['x_start'] 
                        for col in new_columns
                    ]
                    
                    # 更新修改标记
                    structure['modified_v_lines'] = set()
                    
                    clear_table_image_cache()
                    st.success(f"✅ 已删除 {len(lines_to_delete)} 条竖线")
                    st.rerun()
                
                st.info(f"💡 当前有 {len(vertical_lines)} 条竖线，已选择 {len(lines_to_delete)} 条")
            else:
                st.warning("⚠️ 没有可删除的列")
        
        # 保存配置
        st.divider()
        
        save_col1, save_col2, save_col3 = st.columns(3)
        
        with save_col1:
            save_structure = st.checkbox("保存表格结构配置", value=True)
        
        with save_col2:
            save_image = st.checkbox("保存表格线图片", value=True)
        
        with save_col3:
            # 🆕 线条颜色选择
            line_color_option = st.selectbox(
                "保存时线条颜色",
                ["黑色", "蓝色", "红色"],
                index=0
            )
        
        if st.button("💾 保存", type="primary"):
            output_dir = Path("output/table_structures")
            output_dir.mkdir(parents=True, exist_ok=True)
            
            base_name = Path(st.session_state.loaded_image_name).stem
            saved_files = []
            
            if save_structure:
                structure_path = output_dir / f"{base_name}_structure.json"
                
                # 🔧 保存线坐标列表
                save_structure_data = {
                    'rows': structure['rows'],
                    'columns': structure['columns'],
                    'horizontal_lines': structure.get('horizontal_lines', []),
                    'vertical_lines': structure.get('vertical_lines', []),
                    'row_height': structure['row_height'],
                    'col_widths': structure['col_widths'],
                    'table_bbox': structure['table_bbox'],
                    'modified_h_lines': list(structure.get('modified_h_lines', set())),
                    'modified_v_lines': list(structure.get('modified_v_lines', set()))
                }
                
                with open(structure_path, 'w', encoding='utf-8') as f:
                    json.dump(save_structure_data, f, indent=2, ensure_ascii=False)
                
                saved_files.append(("配置文件", structure_path))
                
                with open(structure_path, 'r') as f:
                    st.download_button(
                        "📥 下载配置文件",
                        f.read(),
                        file_name=f"{base_name}_structure.json",
                        mime="application/json"
                    )
            
            if save_image:
                # 🆕 根据选择的颜色绘制纯净表格线
                color_map = {
                    "黑色": (0, 0, 0),
                    "蓝色": (0, 0, 255),
                    "红色": (255, 0, 0)
                }
                selected_color = color_map[line_color_option]
                
                # 🎯 使用纯净绘制函数
                clean_img = draw_clean_table_lines(
                    image,
                    structure,
                    line_width=line_width,
                    line_color=selected_color
                )
                
                output_image_path = output_dir / f"{base_name}_with_lines.png"
                clean_img.save(output_image_path)
                saved_files.append(("表格线图片", output_image_path))
                
                # 🆕 提供下载按钮
                import io
                buf = io.BytesIO()
                clean_img.save(buf, format='PNG')
                buf.seek(0)
                
                st.download_button(
                    "📥 下载表格线图片",
                    buf,
                    file_name=f"{base_name}_with_lines.png",
                    mime="image/png"
                )
            
            if saved_files:
                st.success(f"✅ 已保存 {len(saved_files)} 个文件:")
                for file_type, file_path in saved_files:
                    st.info(f"  • {file_type}: {file_path}")


if __name__ == "__main__":
    create_table_line_editor()