"""
Simple setup script to test if required packages are available
Run this first to check your environment
"""

def check_dependencies():
    """Check if all required packages are available"""
    required_packages = [
        'pandas', 'numpy', 'sklearn', 'streamlit', 'plotly'
    ]
    
    missing_packages = []
    
    for package in required_packages:
        try:
            if package == 'sklearn':
                import sklearn
            else:
                __import__(package)
            print(f"✅ {package} is available")
        except ImportError:
            print(f"❌ {package} is missing")
            missing_packages.append(package)
    
    if missing_packages:
        print(f"\n📦 To install missing packages, run:")
        if 'sklearn' in missing_packages:
            missing_packages.remove('sklearn')
            missing_packages.append('scikit-learn')
        print(f"pip install {' '.join(missing_packages)}")
    else:
        print("\n🎉 All dependencies are available!")
        print("You can now run: python train_and_save_model.py")

if __name__ == "__main__":
    check_dependencies()