#!/usr/bin/env python3
"""
圖片文字檢測(cè) Demo - 生成 HTML 報(bào)告（支持遞歸子目錄）
檢查目錄及其所有子目錄中的圖片是否包含文字，結(jié)果以表格形式展示

用法:
    python check_image_has_text.py [圖片目錄](méi) [方法]
    
示例:
    python check_image_has_text.py /path/to/images simple    # 簡(jiǎn)單方法（快速）
"""

import sys
import cv2
import numpy as np
from pathlib import Path
import base64
from datetime import datetime


def image_to_base64(image_path: Path) -> str:
    """將圖片轉(zhuǎn)為 base64，用于 HTML 顯示"""
    try:
        with open(image_path, "rb") as f:
            return base64.b64encode(f.read()).decode()
    except:
        return ""


def check_text_by_simple_analysis(image_path: Path) -> dict:
    """改進(jìn)版 - 更嚴(yán)格的文字檢測(cè)（盡量降低假陽(yáng)性）"""
    try:
        img = cv2.imread(str(image_path))
        if img is None:
            return {"has_text": False, "confidence": 0, "detail": "無(wú)法讀取圖片"}

        # ── 預(yù)處理 ────────────────────────────────────────
        if img.shape[0] * img.shape[1] > 4000 * 4000:  # 太大就縮小加快速度
            scale = 2000 / max(img.shape[:2])
            img = cv2.resize(img, None, fx=scale, fy=scale)

        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        
        # 嘗試局部自適應(yīng)二值化（對(duì)光照不均更魯棒）
        binary = cv2.adaptiveThreshold(
            gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
            cv2.THRESH_BINARY_INV, 51, 9
        )

        # 輕度去噪
        kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3,3))
        binary = cv2.morphologyEx(binary, cv2.MORPH_OPEN, kernel, iterations=1)

        # ── 連通區(qū)域分析 ─────────────────────────────────
        num_labels, labels, stats, _ = cv2.connectedComponentsWithStats(
            binary, connectivity=8
        )

        text_like_count = 0
        total_text_area = 0
        stroke_widths = []

        h, w = gray.shape

        for i in range(1, num_labels):
            x, y, bw, bh, area = stats[i]

            if area < 15 or area > 8000:
                continue

            aspect = bw / max(bh, 1)
            fill_ratio = area / (bw * bh + 1e-6)

            if not (0.08 < aspect < 8.0):
                continue
            if not (0.25 < fill_ratio < 0.92):
                continue
            if bw < 5 or bh < 5:
                continue

            roi = binary[y:y+bh, x:x+bw]
            if roi.size == 0:
                continue
            dist = cv2.distanceTransform(roi, cv2.DIST_L2, 3)
            max_dist = dist.max()
            if max_dist > 0:
                stroke_widths.append(max_dist * 2)

            text_like_count += 1
            total_text_area += area

        edges = cv2.Canny(gray, 60, 180)
        edge_density = np.mean(edges > 0)

        if text_like_count > 0:
            local_std = np.std(gray[binary > 0])
        else:
            local_std = np.std(gray)

        score_density   = min(text_like_count / 45.0, 1.0)
        score_area      = min(total_text_area / (w * h * 0.018), 1.0)
        score_aspect    = 1.0 if 3 <= len(stroke_widths) <= 180 else 0.2
        score_edge      = min(edge_density * 12, 1.0) if edge_density < 0.25 else 0.0
        score_contrast  = min(local_std / 38.0, 1.0) if local_std > 18 else 0.0

        confidence = (
            score_density   * 0.30 +
            score_area      * 0.25 +
            score_edge      * 0.15 +
            score_contrast  * 0.20 +
            score_aspect    * 0.10
        )

        has_text = (
            confidence > 0.58 and
            text_like_count >= 12 and
            total_text_area >= w * h * 0.004 and
            (np.mean(stroke_widths) < 18 if stroke_widths else True)
        )

        detail = (
            f"區(qū)域:{text_like_count} | 面積比:{total_text_area/(w*h):.4f} | "
            f"邊緣:{edge_density:.4f} | 對(duì)比:{local_std:.1f} | sw:{np.mean(stroke_widths):.1f}"
            if stroke_widths else "無(wú)有效筆畫(huà)寬度"
        )

        return {
            "has_text": has_text,
            "confidence": round(float(confidence), 3),
            "edge_density": round(float(edge_density), 4),
            "contrast": round(float(local_std), 1),
            "text_regions": text_like_count,
            "text_area_ratio": round(total_text_area / (w * h), 4),
            "detail": detail
        }

    except Exception as e:
        return {"has_text": False, "confidence": 0, "detail": f"錯(cuò)誤: {str(e)}"}


def generate_html_report(results: list, root_directory: Path, method: str) -> str:
    """生成 HTML 報(bào)告 - 支持顯示相對(duì)路徑"""
    
    has_text_count = sum(1 for r in results if r.get("has_text"))
    no_text_count = len(results) - has_text_count
    
    html = f"""<!DOCTYPE html>
<html lang="zh-CN">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>圖片文字檢測(cè)結(jié)果（遞歸）</title>
    <style>
        * {{ margin: 0; padding: 0; box-sizing: border-box; }}
        body {{ font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; background: #f5f5f5; padding: 20px; line-height: 1.6; }}
        .container {{ max-width: 1400px; margin: 0 auto; background: white; border-radius: 8px; box-shadow: 0 2px 8px rgba(0,0,0,0.1); padding: 30px; }}
        h1 {{ color: #333; margin-bottom: 10px; font-size: 24px; }}
        .meta {{ color: #666; font-size: 14px; margin-bottom: 20px; padding-bottom: 15px; border-bottom: 1px solid #eee; }}
        .stats {{ display: flex; gap: 20px; margin-bottom: 25px; }}
        .stat-item {{ flex: 1; background: #f8f9fa; padding: 15px 20px; border-radius: 6px; text-align: center; }}
        .stat-number {{ font-size: 32px; font-weight: bold; color: #1890ff; }}
        .stat-label {{ color: #666; font-size: 14px; margin-top: 5px; }}
        .stat-item.success .stat-number {{ color: #52c41a; }}
        .stat-item.error .stat-number {{ color: #ff4d4f; }}
        table {{ width: 100%; border-collapse: collapse; font-size: 14px; }}
        th {{ background: #fafafa; padding: 12px; text-align: left; font-weight: 600; color: #333; border-bottom: 2px solid #e8e8e8; position: sticky; top: 0; }}
        td {{ padding: 12px; border-bottom: 1px solid #e8e8e8; vertical-align: middle; }}
        tr:hover {{ background: #f5f5f5; }}
        .preview-img {{ width: 120px; height: 80px; object-fit: cover; border-radius: 4px; border: 1px solid #d9d9d9; cursor: pointer; transition: transform 0.2s; }}
        .preview-img:hover {{ transform: scale(2); z-index: 100; box-shadow: 0 4px 12px rgba(0,0,0,0.15); }}
        .status {{ display: inline-block; padding: 4px 12px; border-radius: 12px; font-size: 12px; font-weight: 500; }}
        .status-yes {{ background: #f6ffed; color: #52c41a; border: 1px solid #b7eb8f; }}
        .status-no {{ background: #fff2f0; color: #ff4d4f; border: 1px solid #ffccc7; }}
        .confidence {{ font-weight: 600; color: #1890ff; }}
        .confidence-high {{ color: #52c41a; }}
        .confidence-low {{ color: #ff4d4f; }}
        .detail {{ color: #666; font-size: 12px; max-width: 300px; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }}
        .detail:hover {{ white-space: normal; word-break: break-all; }}
        .index {{ color: #999; font-size: 12px; }}
        .relpath {{ font-family: monospace; font-size: 12px; color: #333; }}
        @media (max-width: 768px) {{ .container {{ padding: 15px; }} .stats {{ flex-direction: column; }} th, td {{ padding: 8px; font-size: 12px; }} .preview-img {{ width: 80px; height: 60px; }} }}
    </style>
</head>
<body>
    <div class="container">
        <h1>??? 圖片文字檢測(cè)報(bào)告（遞歸子目錄）</h1>
        <div class="meta">
            <div>?? 根目錄: {root_directory}</div>
            <div>?? 方法: {method}</div>
            <div>? 生成時(shí)間: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</div>
        </div>
        
        <div class="stats">
            <div class="stat-item"><div class="stat-number">{len(results)}</div><div class="stat-label">總圖片數(shù)</div></div>
            <div class="stat-item success"><div class="stat-number">{has_text_count}</div><div class="stat-label">有文字</div></div>
            <div class="stat-item error"><div class="stat-number">{no_text_count}</div><div class="stat-label">無(wú)文字</div></div>
        </div>
        
        <table>
            <thead>
                <tr>
                    <th style="width: 50px;">#</th>
                    <th style="width: 140px;">預(yù)覽</th>
                    <th>相對(duì)路徑 / 文件名</th>
                    <th style="width: 100px;">結(jié)果</th>
                    <th style="width: 100px;">置信度</th>
                    <th>詳細(xì)信息</th>
                </tr>
            </thead>
            <tbody>
"""
    
    for i, r in enumerate(results, 1):
        img_base64 = image_to_base64(Path(r["path"]))
        img_src = f"data:image/jpeg;base64,{img_base64}" if img_base64 else ""
        
        status_class = "status-yes" if r.get("has_text") else "status-no"
        status_text = "? 有文字" if r.get("has_text") else "? 無(wú)文字"
        
        conf = r.get("confidence", 0)
        conf_class = "confidence-high" if conf > 0.7 else ("confidence-low" if conf < 0.3 else "")
        
        rel_path = str(Path(r["path"]).relative_to(root_directory))
        
        html += f"""
                <tr>
                    <td class="index">{i}</td>
                    <td>{f'<img src="{img_src}" class="preview-img" alt="預(yù)覽">' if img_src else 'N/A'}</td>
                    <td class="relpath" title="{rel_path}">{rel_path}</td>
                    <td><span class="status {status_class}">{status_text}</span></td>
                    <td class="confidence {conf_class}">{conf:.3f}</td>
                    <td class="detail" title="{r.get('detail', '')}">{r.get('detail', '')}</td>
                </tr>
"""
    
    html += """
            </tbody>
        </table>
    </div>
</body>
</html>
"""
    return html


def process_directory(root_dir: Path, method: str = "simple"):
    """遞歸處理目錄及其所有子目錄中的圖片，并生成 HTML 報(bào)告"""
    
    image_extensions = {'.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp'}
    
    # 遞歸查找所有圖片
    images = [
        p for p in root_dir.rglob("*")
        if p.is_file() and p.suffix.lower() in image_extensions
    ]
    # 按相對(duì)路徑排序，便于閱讀
    images.sort(key=lambda p: str(p.relative_to(root_dir)))
    
    if not images:
        print(f"? 在 {root_dir} 及其子目錄中沒(méi)有找到任何圖片")
        return
    
    print(f"?? 根目錄: {root_dir}")
    print(f"???  找到 {len(images)} 張圖片（包含所有子目錄）")
    print(f"?? 使用方法: {method}")
    print("? 正在處理...")
    
    results = []
    
    for i, img_path in enumerate(images, 1):
        print(f"  [{i}/{len(images)}] {img_path.relative_to(root_dir)}", end="\r")
        
        if method == "ocr":
            result = check_text_by_ocr(img_path)
        else:
            result = check_text_by_simple_analysis(img_path)
        
        results.append({
            "file": str(img_path.relative_to(root_dir)),  # 相對(duì)路徑作為標(biāo)識(shí)
            "path": str(img_path.absolute()),
            "method": method,
            **result
        })
    
    print()  # 換行
    
    # 生成并保存 HTML
    html_content = generate_html_report(results, root_dir, method)
    
    output_file = root_dir / "text_detection_report_recursive.html"
    output_file.write_text(html_content, encoding='utf-8')
    
    # 統(tǒng)計(jì)
    has_text_count = sum(1 for r in results if r.get("has_text"))
    
    print("\n" + "=" * 70)
    print("?? 檢測(cè)結(jié)果匯總（遞歸）：")
    print(f"   有文字: {has_text_count} 張")
    print(f"   無(wú)文字: {len(images) - has_text_count} 張")
    print(f"   總計(jì): {len(images)} 張")
    print("-" * 70)
    print(f"?? HTML 報(bào)告已生成: {output_file}")
    print(f"?? 請(qǐng)?jiān)跒g覽器中打開(kāi)查看（表格顯示相對(duì)路徑）")


def main():
    default_dir = "/home/michah/桌面/nlp/docu_intel/storage/temp"
    
    directory = Path(sys.argv[1]) if len(sys.argv) > 1 else Path(default_dir)
    method = sys.argv[2] if len(sys.argv) > 2 else "simple"
    
    if method not in ["simple"]:
        print("用法: python check_image_has_text.py [目錄](méi) [simple]")
        print("  simple : 簡(jiǎn)單圖像分析（快速，無(wú)需額外依賴）")
        sys.exit(1)
    
    if not directory.is_dir():
        print(f"? 路徑不是目錄或不存在: {directory}")
        sys.exit(1)
    
    process_directory(directory, method)


if __name__ == "__main__":
    main()

pip install opencv-python

# 掃描當(dāng)前目錄及其子目錄（simple 模式）
python check_images_text.py .

# 指定目錄 + ocr 模式
python check_images_text.py /path/to/your/folder ocr