#!/usr/bin/env python3
"""
NZ Legislation Loophole Analysis Streamlit App Runner

This script runs the modern Streamlit application for analyzing New Zealand legislation
to identify potential loopholes, ambiguities, and unintended consequences using AI.

Features:
- Advanced UI with multi-page layout
- Context memory cache system for improved performance
- Real-time progress monitoring
- Interactive results visualization
- Batch processing capabilities
- Comprehensive configuration management

Usage:
    python run_streamlit_app.py

Requirements:
    - All dependencies from requirements.txt must be installed
    - Run from the project root directory
"""

import os
import sys
import subprocess
from pathlib import Path

def check_requirements():
    """Check if all required packages are installed"""
    required_packages = [
        'streamlit',
        'pandas',
        'plotly',
        'llama-cpp-python',
        'psutil',
        'numpy'
    ]

    missing_packages = []

    for package in required_packages:
        try:
            __import__(package.replace('-', '_'))
        except ImportError:
            missing_packages.append(package)

    if missing_packages:
        print("❌ Missing required packages:")
        for package in missing_packages:
            print(f"   - {package}")

        print("\n📦 Installing missing packages...")
        try:
            subprocess.check_call([
                sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt'
            ])
            print("✅ All packages installed successfully!")
        except subprocess.CalledProcessError:
            print("❌ Failed to install packages. Please install manually:")
            print("   pip install -r requirements.txt")
            return False

    return True

def check_app_structure():
    """Check if the app structure is correct"""
    app_dir = Path('streamlit_app')
    required_files = [
        'app.py',
        'core/cache_manager.py',
        'core/text_processor.py',
        'core/llm_analyzer.py',
        'core/dataset_builder.py',
        'utils/config.py',
        'utils/performance.py',
        'utils/ui_helpers.py'
    ]

    missing_files = []

    for file_path in required_files:
        full_path = app_dir / file_path
        if not full_path.exists():
            missing_files.append(str(full_path))

    if missing_files:
        print("❌ Missing app files:")
        for file_path in missing_files:
            print(f"   - {file_path}")
        return False

    print("✅ App structure is complete!")
    return True

def create_directories():
    """Create necessary directories"""
    directories = [
        'streamlit_app/cache',
        'streamlit_app/config',
        'streamlit_app/datasets',
        'streamlit_app/logs'
    ]

    for dir_path in directories:
        Path(dir_path).mkdir(parents=True, exist_ok=True)
        print(f"📁 Created directory: {dir_path}")

def setup_environment():
    """Setup environment variables and configuration"""
    # Add current directory to Python path for imports
    current_dir = os.path.dirname(os.path.abspath(__file__))
    if current_dir not in sys.path:
        sys.path.insert(0, current_dir)

    # Set environment variables
    os.environ.setdefault('STREAMLIT_SERVER_HEADLESS', 'true')
    os.environ.setdefault('STREAMLIT_BROWSER_GATHER_USAGE_STATS', 'false')

    print("🔧 Environment setup complete!")

def run_app():
    """Run the Streamlit application"""
    print("\n🚀 Starting NZ Legislation Loophole Analyzer...")
    print("=" * 60)
    print("📱 Access the app at: http://localhost:8501")
    print("🛑 Press Ctrl+C to stop the application")
    print("=" * 60)

    try:
        # Change to app directory
        os.chdir('streamlit_app')

        # Run Streamlit
        subprocess.run([
            sys.executable, '-m', 'streamlit', 'run', 'app.py',
            '--server.port', '8501',
            '--server.address', '0.0.0.0',
            '--theme.base', 'light'
        ])

    except KeyboardInterrupt:
        print("\n\n👋 Application stopped by user")
    except Exception as e:
        print(f"\n❌ Error running application: {e}")
        return False

    return True

def main():
    """Main function"""
    print("🏛️  NZ Legislation Loophole Analysis Streamlit App")
    print("=" * 60)

    # Check requirements
    if not check_requirements():
        return 1

    # Check app structure
    if not check_app_structure():
        return 1

    # Create directories
    create_directories()

    # Setup environment
    setup_environment()

    # Run the app
    if not run_app():
        return 1

    return 0

if __name__ == "__main__":
    sys.exit(main())