scripts/train.py

"""
One-time training script.
Run: python scripts/train.py
"""
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from db.supabase_client import fetch_user_profiles
from ml.roommate_model import RoommateMatchingModel

def main():
    print("🚀 Starting model training...")
    profiles = fetch_user_profiles()

    if not profiles:
        print("❌ No user profiles found. Check your Supabase credentials.")
        return

    print(f"📊 Found {len(profiles)} user profiles")

    model = RoommateMatchingModel()
    model.train(profiles)

    os.makedirs("models", exist_ok=True)
    model.save("models/")

    print("✅ Training complete! Models saved to models/")
    print("   → models/gb_model.pkl")
    print("   → models/kmeans.pkl")
    print("   → models/scaler.pkl")
    print("\n▶ Now run: uvicorn main:app --reload")

if __name__ == "__main__":
    main()