#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
身份证 OCR 识别测试脚本
使用 PaddleOCR 进行身份证信息提取测试
"""

from utils.ocr_utils import extract_id_card_info, extract_text_from_image
import sys

def test_id_card_recognition(image_path):
    """测试身份证识别"""
    print(f"\n{'='*60}")
    print(f"正在识别身份证图片: {image_path}")
    print(f"{'='*60}\n")

    # 方法1: 提取所有文本
    print("方法1: 提取所有文本")
    print("-" * 60)
    all_text = extract_text_from_image(image_path)
    print(all_text)
    print()

    # 方法2: 直接提取身份证信息
    print("方法2: 提取身份证关键信息")
    print("-" * 60)
    id_info = extract_id_card_info(image_path)
    print(f"姓名: {id_info.get('name')}")
    print(f"身份证号: {id_info.get('id_number')}")
    print()

    # 验证结果
    if id_info.get('name') and id_info.get('id_number'):
        print("✓ 识别成功!")
    else:
        print("✗ 识别失败，请检查图片质量或格式")

    print(f"{'='*60}\n")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("使用方法: python test_id_card_ocr.py <身份证图片路径>")
        print("示例: python test_id_card_ocr.py ./test_images/id_card.jpg")
        sys.exit(1)

    image_path = r'/workspace/campus_activity_system/BaiduHi_2026-8-4_11-9-55.jpg' #sys.argv[1]
    test_id_card_recognition(image_path)