#!/usr/bin/env python3
"""
Test script to check PaddleOCR GPU support and available parameters
"""

import paddleocr
import sys

def test_paddleocr_gpu():
    print("Testing PaddleOCR GPU support...")
    print(f"Python version: {sys.version}")
    print(f"PaddleOCR version: {paddleocr.__version__}")
    
    # Check available parameters by inspecting the PaddleOCR class
    import inspect
    sig = inspect.signature(paddleocr.PaddleOCR.__init__)
    print(f"\nAvailable PaddleOCR parameters: {list(sig.parameters.keys())}")
    
    # Try to initialize with different GPU parameters
    print("\nTesting different GPU parameter combinations:")
    
    # Test 1: Basic initialization (CPU)
    try:
        ocr1 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en')
        print("✓ Basic initialization (CPU) successful")
        del ocr1
    except Exception as e:
        print(f"✗ Basic initialization failed: {e}")
    
    # Test 2: Try use_gpu parameter
    try:
        ocr2 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', use_gpu=True)
        print("✓ GPU initialization with use_gpu=True successful")
        del ocr2
    except Exception as e:
        print(f"✗ GPU initialization with use_gpu=True failed: {e}")
    
    # Test 3: Try gpu_id parameter
    try:
        ocr3 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', gpu_id=0)
        print("✓ GPU initialization with gpu_id=0 successful")
        del ocr3
    except Exception as e:
        print(f"✗ GPU initialization with gpu_id=0 failed: {e}")
    
    # Test 4: Try enable_mkldnn parameter (Intel acceleration)
    try:
        ocr4 = paddleocr.PaddleOCR(use_angle_cls=True, lang='en', enable_mkldnn=True)
        print("✓ CPU initialization with enable_mkldnn=True successful")
        del ocr4
    except Exception as e:
        print(f"✗ CPU initialization with enable_mkldnn=True failed: {e}")
    
    print("\nGPU configuration test completed.")

if __name__ == "__main__":
    test_paddleocr_gpu()