setup.py

"""
One-click setup script for Student Performance Tracker.
Generates synthetic Indian student data, loads it into SQLite, and trains the ML model.

Usage:
    python setup.py
"""

import subprocess
import sys
import os

def main():
    base_dir = os.path.dirname(os.path.abspath(__file__))
    src_dir = os.path.join(base_dir, 'src')
    
    steps = [
        ("Step 1/3: Generating synthetic Indian student data...", [sys.executable, os.path.join(src_dir, 'generate_data.py')]),
        ("Step 2/3: Loading data into SQLite database...", [sys.executable, os.path.join(src_dir, 'db.py')]),
        ("Step 3/3: Training ML model + SHAP explainer...", [sys.executable, os.path.join(src_dir, 'train_model.py')]),
    ]
    
    print("=" * 60)
    print("  Student Performance Tracker — Setup")
    print("=" * 60)
    
    for msg, cmd in steps:
        print(f"\n{msg}")
        result = subprocess.run(cmd, cwd=base_dir)
        if result.returncode != 0:
            print(f"❌ Failed at: {msg}")
            sys.exit(1)
    
    print("\n" + "=" * 60)
    print("  [OK] Setup complete! Run the dashboard with:")
    print("       streamlit run app.py")
    print("=" * 60)

if __name__ == "__main__":
    main()