verify_setup.py

"""
Quick setup verification script.
"""

import sys
from pathlib import Path

def verify_setup():
    """Verify that all components are set up correctly."""
    print("Verifying setup...")
    print("-" * 60)
    
    errors = []
    
    # Check Python packages
    print("Checking Python packages...")
    try:
        import pdfplumber
        print("  ✓ pdfplumber")
    except ImportError:
        errors.append("pdfplumber not installed")
        print("  ✗ pdfplumber")
    
    try:
        import openpyxl
        print("  ✓ openpyxl")
    except ImportError:
        errors.append("openpyxl not installed")
        print("  ✗ openpyxl")
    
    try:
        import ollama
        print("  ✓ ollama")
    except ImportError:
        errors.append("ollama not installed")
        print("  ✗ ollama")
    
    try:
        import pandas
        print("  ✓ pandas")
    except ImportError:
        errors.append("pandas not installed")
        print("  ✗ pandas")
    
    # Check Ollama model
    print("\nChecking Ollama model...")
    try:
        import ollama
        models_response = ollama.list()
        # Handle different response formats
        if isinstance(models_response, dict) and 'models' in models_response:
            model_names = [model.get('name', '') for model in models_response['models']]
        elif isinstance(models_response, list):
            model_names = [model.get('name', '') for model in models_response]
        else:
            model_names = []
        
        if 'llama3.1:8b' in model_names:
            print("  ✓ llama3.1:8b model available")
        else:
            # Not a critical error - user can pull it later
            print(f"  ⚠ llama3.1:8b not found. Run: ollama pull llama3.1:8b")
            print(f"    Available models: {model_names[:3]}...")
    except Exception as e:
        print(f"  ⚠ Could not verify Ollama model: {e}")
        print("    Make sure Ollama is running: ollama serve")
    
    # Check files
    print("\nChecking project files...")
    required_files = [
        'main.py',
        'pdf_extractor.py',
        'ai_parser.py',
        'excel_writer.py',
        'config.json',
        'requirements.txt',
        'template.xlsx'
    ]
    
    for file in required_files:
        if Path(file).exists():
            print(f"  ✓ {file}")
        else:
            errors.append(f"{file} not found")
            print(f"  ✗ {file}")
    
    # Check directories
    print("\nChecking directories...")
    required_dirs = ['statements', 'output']
    for dir_name in required_dirs:
        if Path(dir_name).exists():
            print(f"  ✓ {dir_name}/")
        else:
            print(f"  ✗ {dir_name}/ (will be created automatically)")
    
    # Summary
    print("\n" + "=" * 60)
    if errors:
        print("SETUP INCOMPLETE")
        print("=" * 60)
        print("Errors found:")
        for error in errors:
            print(f"  - {error}")
        print("\nPlease fix the errors above and try again.")
        return False
    else:
        print("SETUP COMPLETE ✓")
        print("=" * 60)
        print("\nYou're ready to use LedgerLens!")
        print("\nNext steps:")
        print("1. Place your PDF statements in the 'statements/' folder")
        print("2. Ensure your Excel template is at 'template.xlsx' (or update config.json)")
        print("3. Run: python main.py")
        return True

if __name__ == "__main__":
    success = verify_setup()
    sys.exit(0 if success else 1)