paddleocr-skills 1.0.0 → 1.1.0

package/templates/{paddleocr-vl/scripts/paddleocr-vl → paddleocr-vl-1.5/scripts/paddleocr-vl-1.5}/configure.py

@@ -1,270 +1,270 @@
-#!/usr/bin/env python3
-"""
-PaddleOCR-VL Configuration Wizard
-
-Supports two modes:
-1. Interactive mode (default): python configure.py
-2. CLI mode: python configure.py --api-url URL --token TOKEN
-
-Interactive configuration for PaddleOCR-VL API credentials.
-Saves configuration to .env file in project root.
-"""
-
-import argparse
-import os
-import sys
-from pathlib import Path
-
-
-def save_config(api_url: str, token: str, project_root: Path, quiet: bool = False) -> bool:
-    """
-    Save configuration to .env file
-
-    Args:
-        api_url: VL API URL
-        token: VL access token
-        project_root: Project root directory
-        quiet: If True, suppress output messages
-
-    Returns:
-        True if successful, False otherwise
-    """
-    env_file = project_root / ".env"
-
-    # Read existing .env if it exists
-    existing_config = {}
-    if env_file.exists():
-        if not quiet:
-            print(f"Found existing .env file: {env_file}")
-            overwrite = input("Overwrite? [Y/n]: ").strip().lower()
-            if overwrite == 'n':
-                print("Configuration cancelled")
-                return False
-
-        with open(env_file, 'r', encoding='utf-8') as f:
-            for line in f:
-                line = line.strip()
-                if line and not line.startswith('#') and '=' in line:
-                    key, value = line.split('=', 1)
-                    existing_config[key.strip()] = value.strip()
-
-    # Update configuration
-    existing_config['VL_API_URL'] = api_url
-    existing_config['VL_TOKEN'] = token
-
-    # Write to .env file
-    try:
-        with open(env_file, 'w', encoding='utf-8') as f:
-            # Write header
-            f.write("# PaddleOCR Skills Configuration\n")
-            f.write("# Generated by configuration wizard\n")
-            f.write("\n")
-
-            # Group PP-OCRv5 configs
-            if any(k.startswith('API_URL') or k.startswith('PADDLE_OCR_TOKEN') or k.startswith('PPOCRV5') for k in existing_config.keys()):
-                f.write("# ========================================\n")
-                f.write("# PP-OCRv5 Configuration\n")
-                f.write("# ========================================\n")
-                for key in ['API_URL', 'PADDLE_OCR_TOKEN', 'PPOCRV5_API_URL', 'PPOCRV5_TOKEN']:
-                    if key in existing_config:
-                        f.write(f"{key}={existing_config[key]}\n")
-                f.write("\n")
-
-            # Group PaddleOCR-VL configs
-            f.write("# ========================================\n")
-            f.write("# PaddleOCR-VL Configuration\n")
-            f.write("# ========================================\n")
-            f.write(f"VL_API_URL={existing_config['VL_API_URL']}\n")
-            f.write(f"VL_TOKEN={existing_config['VL_TOKEN']}\n")
-            f.write("\n")
-
-            # Write other configs
-            other_keys = [k for k in existing_config.keys()
-                         if k not in ['API_URL', 'PADDLE_OCR_TOKEN', 'PPOCRV5_API_URL',
-                                      'PPOCRV5_TOKEN', 'VL_API_URL', 'VL_TOKEN']]
-            if other_keys:
-                f.write("# ========================================\n")
-                f.write("# Other Configuration\n")
-                f.write("# ========================================\n")
-                for key in other_keys:
-                    f.write(f"{key}={existing_config[key]}\n")
-
-        if not quiet:
-            print(f"✓ Configuration saved to {env_file}")
-        return True
-
-    except Exception as e:
-        print(f"✗ Failed to save configuration: {e}")
-        return False
-
-
-def main():
-    # Parse command-line arguments
-    parser = argparse.ArgumentParser(
-        description='PaddleOCR-VL API Configuration Tool',
-        formatter_class=argparse.RawDescriptionHelpFormatter,
-        epilog="""
-Examples:
-  # Interactive mode
-  python configure.py
-
-  # CLI mode (non-interactive)
-  python configure.py --api-url "https://your-service.com/v1" --token "your_token"
-        """
-    )
-    parser.add_argument('--api-url', help='VL API URL (non-interactive mode)')
-    parser.add_argument('--token', help='VL access token (non-interactive mode)')
-    parser.add_argument('--quiet', action='store_true', help='Suppress output messages')
-
-    args = parser.parse_args()
-
-    # Find .env file location (project root, 2 levels up from script)
-    project_root = Path(__file__).parent.parent.parent
-
-    # ========================================
-    # CLI Mode (non-interactive)
-    # ========================================
-    if args.api_url and args.token:
-        try:
-            api_url = args.api_url.strip()
-            token = args.token.strip()
-
-            # Validate URL format
-            if not api_url.startswith(("http://", "https://")):
-                api_url = f"https://{api_url}"
-
-            # Validate token
-            if len(token) < 16:
-                print("Error: Token seems too short. Please check and try again.")
-                sys.exit(1)
-
-            # Save configuration
-            if save_config(api_url, token, project_root, quiet=args.quiet):
-                if not args.quiet:
-                    masked_token = token[:8] + "..." + token[-4:] if len(token) > 12 else "***"
-                    print("\n✓ Configuration complete!")
-                    print(f"  VL_API_URL: {api_url}")
-                    print(f"  VL_TOKEN: {masked_token}")
-                sys.exit(0)
-            else:
-                sys.exit(1)
-
-        except Exception as e:
-            print(f"Error: {e}")
-            sys.exit(1)
-
-    elif args.api_url or args.token:
-        print("Error: Both --api-url and --token are required for CLI mode")
-        print("Run without arguments for interactive mode")
-        sys.exit(1)
-
-    # ========================================
-    # Interactive Mode
-    # ========================================
-    print("=" * 60)
-    print("PaddleOCR-VL Configuration Wizard")
-    print("=" * 60)
-    print()
-
-    env_file = project_root / ".env"
-    print(f"Configuration will be saved to: {env_file}")
-    print()
-
-    # Read existing .env if it exists
-    existing_config = {}
-    if env_file.exists():
-        print("Found existing .env file, loading current values...")
-        with open(env_file, 'r', encoding='utf-8') as f:
-            for line in f:
-                line = line.strip()
-                if line and not line.startswith('#') and '=' in line:
-                    key, value = line.split('=', 1)
-                    existing_config[key.strip()] = value.strip()
-        print()
-
-    # Get current values
-    current_api_url = existing_config.get('VL_API_URL', '')
-    current_token = existing_config.get('VL_TOKEN', '')
-
-    print("Please provide your PaddleOCR-VL API credentials:")
-    print("(Press Enter to keep current value)")
-    print()
-
-    # Prompt for API URL
-    print("1. VL_API_URL - PaddleOCR-VL API endpoint")
-    print("   Example: https://your-service.com/v1")
-    if current_api_url:
-        print(f"   Current: {current_api_url}")
-
-    api_url_input = input("   Enter VL_API_URL: ").strip()
-    new_api_url = api_url_input if api_url_input else current_api_url
-
-    if not new_api_url:
-        print()
-        print("ERROR: VL_API_URL is required.")
-        print("Please run this wizard again and provide a valid API URL.")
-        sys.exit(1)
-
-    print()
-
-    # Prompt for Token
-    print("2. VL_TOKEN - Your access token")
-    if current_token:
-        masked_token = current_token[:8] + "..." + current_token[-4:] if len(current_token) > 12 else "***"
-        print(f"   Current: {masked_token}")
-
-    token_input = input("   Enter VL_TOKEN: ").strip()
-    new_token = token_input if token_input else current_token
-
-    if not new_token:
-        print()
-        print("ERROR: VL_TOKEN is required.")
-        print("Please run this wizard again and provide a valid token.")
-        sys.exit(1)
-
-    print()
-
-    # Save configuration
-    print("Saving configuration...")
-
-    if not save_config(new_api_url, new_token, project_root):
-        sys.exit(1)
-
-    print()
-
-    # Verify configuration
-    print("Verifying configuration...")
-    try:
-        sys.path.insert(0, str(Path(__file__).parent))
-        from _lib import Config
-
-        Config.load_env()
-
-        test_url = Config.get_vl_api_url()
-        test_token = Config.get_vl_token()
-
-        print("✓ VL_API_URL loaded successfully")
-        print("✓ VL_TOKEN loaded successfully")
-        print()
-
-    except Exception as e:
-        print(f"✗ Configuration verification failed: {e}")
-        print()
-        sys.exit(1)
-
-    # Next steps
-    print("=" * 60)
-    print("Configuration Complete!")
-    print("=" * 60)
-    print()
-    print("Next steps:")
-    print("  1. Test the configuration:")
-    print("     python scripts/paddleocr-vl/smoke_test.py")
-    print()
-    print("  2. Try parsing a document:")
-    print("     python scripts/paddleocr-vl/vl_caller.py --file-url \"URL\"")
-    print()
-
-
-if __name__ == '__main__':
-    main()
+#!/usr/bin/env python3
+"""
+PaddleOCR-VL 1.5 Configuration Wizard
+
+Supports two modes:
+1. Interactive mode (default): python configure.py
+2. CLI mode: python configure.py --api-url URL --token TOKEN
+
+Interactive configuration for PaddleOCR-VL 1.5 API credentials.
+Saves configuration to .env file in project root.
+"""
+
+import argparse
+import os
+import sys
+from pathlib import Path
+
+
+def save_config(api_url: str, token: str, project_root: Path, quiet: bool = False) -> bool:
+    """
+    Save configuration to .env file
+
+    Args:
+        api_url: VL API URL
+        token: VL access token
+        project_root: Project root directory
+        quiet: If True, suppress output messages
+
+    Returns:
+        True if successful, False otherwise
+    """
+    env_file = project_root / ".env"
+
+    # Read existing .env if it exists
+    existing_config = {}
+    if env_file.exists():
+        if not quiet:
+            print(f"Found existing .env file: {env_file}")
+            overwrite = input("Overwrite? [Y/n]: ").strip().lower()
+            if overwrite == 'n':
+                print("Configuration cancelled")
+                return False
+
+        with open(env_file, 'r', encoding='utf-8') as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith('#') and '=' in line:
+                    key, value = line.split('=', 1)
+                    existing_config[key.strip()] = value.strip()
+
+    # Update configuration
+    existing_config['VL_API_URL'] = api_url
+    existing_config['VL_TOKEN'] = token
+
+    # Write to .env file
+    try:
+        with open(env_file, 'w', encoding='utf-8') as f:
+            # Write header
+            f.write("# PaddleOCR Skills Configuration\n")
+            f.write("# Generated by configuration wizard\n")
+            f.write("\n")
+
+            # Group PP-OCRv5 configs
+            if any(k.startswith('API_URL') or k.startswith('PADDLE_OCR_TOKEN') or k.startswith('PPOCRV5') for k in existing_config.keys()):
+                f.write("# ========================================\n")
+                f.write("# PP-OCRv5 Configuration\n")
+                f.write("# ========================================\n")
+                for key in ['API_URL', 'PADDLE_OCR_TOKEN', 'PPOCRV5_API_URL', 'PPOCRV5_TOKEN']:
+                    if key in existing_config:
+                        f.write(f"{key}={existing_config[key]}\n")
+                f.write("\n")
+
+            # Group PaddleOCR-VL 1.5 configs
+            f.write("# ========================================\n")
+            f.write("# PaddleOCR-VL 1.5 Configuration\n")
+            f.write("# ========================================\n")
+            f.write(f"VL_API_URL={existing_config['VL_API_URL']}\n")
+            f.write(f"VL_TOKEN={existing_config['VL_TOKEN']}\n")
+            f.write("\n")
+
+            # Write other configs
+            other_keys = [k for k in existing_config.keys()
+                         if k not in ['API_URL', 'PADDLE_OCR_TOKEN', 'PPOCRV5_API_URL',
+                                      'PPOCRV5_TOKEN', 'VL_API_URL', 'VL_TOKEN']]
+            if other_keys:
+                f.write("# ========================================\n")
+                f.write("# Other Configuration\n")
+                f.write("# ========================================\n")
+                for key in other_keys:
+                    f.write(f"{key}={existing_config[key]}\n")
+
+        if not quiet:
+            print(f"✓ Configuration saved to {env_file}")
+        return True
+
+    except Exception as e:
+        print(f"✗ Failed to save configuration: {e}")
+        return False
+
+
+def main():
+    # Parse command-line arguments
+    parser = argparse.ArgumentParser(
+        description='PaddleOCR-VL 1.5 API Configuration Tool',
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  # Interactive mode
+  python configure.py
+
+  # CLI mode (non-interactive)
+  python configure.py --api-url "https://your-service.com/v1" --token "your_token"
+        """
+    )
+    parser.add_argument('--api-url', help='VL API URL (non-interactive mode)')
+    parser.add_argument('--token', help='VL access token (non-interactive mode)')
+    parser.add_argument('--quiet', action='store_true', help='Suppress output messages')
+
+    args = parser.parse_args()
+
+    # Find .env file location (project root, 2 levels up from script)
+    project_root = Path(__file__).parent.parent.parent
+
+    # ========================================
+    # CLI Mode (non-interactive)
+    # ========================================
+    if args.api_url and args.token:
+        try:
+            api_url = args.api_url.strip()
+            token = args.token.strip()
+
+            # Validate URL format
+            if not api_url.startswith(("http://", "https://")):
+                api_url = f"https://{api_url}"
+
+            # Validate token
+            if len(token) < 16:
+                print("Error: Token seems too short. Please check and try again.")
+                sys.exit(1)
+
+            # Save configuration
+            if save_config(api_url, token, project_root, quiet=args.quiet):
+                if not args.quiet:
+                    masked_token = token[:8] + "..." + token[-4:] if len(token) > 12 else "***"
+                    print("\n✓ Configuration complete!")
+                    print(f"  VL_API_URL: {api_url}")
+                    print(f"  VL_TOKEN: {masked_token}")
+                sys.exit(0)
+            else:
+                sys.exit(1)
+
+        except Exception as e:
+            print(f"Error: {e}")
+            sys.exit(1)
+
+    elif args.api_url or args.token:
+        print("Error: Both --api-url and --token are required for CLI mode")
+        print("Run without arguments for interactive mode")
+        sys.exit(1)
+
+    # ========================================
+    # Interactive Mode
+    # ========================================
+    print("=" * 60)
+    print("PaddleOCR-VL 1.5 Configuration Wizard")
+    print("=" * 60)
+    print()
+
+    env_file = project_root / ".env"
+    print(f"Configuration will be saved to: {env_file}")
+    print()
+
+    # Read existing .env if it exists
+    existing_config = {}
+    if env_file.exists():
+        print("Found existing .env file, loading current values...")
+        with open(env_file, 'r', encoding='utf-8') as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith('#') and '=' in line:
+                    key, value = line.split('=', 1)
+                    existing_config[key.strip()] = value.strip()
+        print()
+
+    # Get current values
+    current_api_url = existing_config.get('VL_API_URL', '')
+    current_token = existing_config.get('VL_TOKEN', '')
+
+    print("Please provide your PaddleOCR-VL 1.5 API credentials:")
+    print("(Press Enter to keep current value)")
+    print()
+
+    # Prompt for API URL
+    print("1. VL_API_URL - PaddleOCR-VL 1.5 API endpoint")
+    print("   Example: https://your-service.com/v1")
+    if current_api_url:
+        print(f"   Current: {current_api_url}")
+
+    api_url_input = input("   Enter VL_API_URL: ").strip()
+    new_api_url = api_url_input if api_url_input else current_api_url
+
+    if not new_api_url:
+        print()
+        print("ERROR: VL_API_URL is required.")
+        print("Please run this wizard again and provide a valid API URL.")
+        sys.exit(1)
+
+    print()
+
+    # Prompt for Token
+    print("2. VL_TOKEN - Your access token")
+    if current_token:
+        masked_token = current_token[:8] + "..." + current_token[-4:] if len(current_token) > 12 else "***"
+        print(f"   Current: {masked_token}")
+
+    token_input = input("   Enter VL_TOKEN: ").strip()
+    new_token = token_input if token_input else current_token
+
+    if not new_token:
+        print()
+        print("ERROR: VL_TOKEN is required.")
+        print("Please run this wizard again and provide a valid token.")
+        sys.exit(1)
+
+    print()
+
+    # Save configuration
+    print("Saving configuration...")
+
+    if not save_config(new_api_url, new_token, project_root):
+        sys.exit(1)
+
+    print()
+
+    # Verify configuration
+    print("Verifying configuration...")
+    try:
+        sys.path.insert(0, str(Path(__file__).parent))
+        from _lib import Config
+
+        Config.load_env()
+
+        test_url = Config.get_vl_api_url()
+        test_token = Config.get_vl_token()
+
+        print("✓ VL_API_URL loaded successfully")
+        print("✓ VL_TOKEN loaded successfully")
+        print()
+
+    except Exception as e:
+        print(f"✗ Configuration verification failed: {e}")
+        print()
+        sys.exit(1)
+
+    # Next steps
+    print("=" * 60)
+    print("Configuration Complete!")
+    print("=" * 60)
+    print()
+    print("Next steps:")
+    print("  1. Test the configuration:")
+    print("     python scripts/paddleocr-vl-1.5/smoke_test.py")
+    print()
+    print("  2. Try parsing a document:")
+    print("     python scripts/paddleocr-vl-1.5/vl_caller.py --file-url \"URL\"")
+    print()
+
+
+if __name__ == '__main__':
+    main()