fix_ai_loans.py

import sys, os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from app import app, db
from models.loan_model import Loan
from models.user_model import User
from services.ai_service import predict_risk

with app.app_context():
    loans = db.session.query(Loan).filter(Loan.risk_score == None).all()
    print(f"Found {len(loans)} loans with no AI assessment")
    
    for loan in loans:
        user = db.session.query(User).filter(User.id == loan.user_id).first()
        if not user:
            print(f"Loan #{loan.id}: No user found, skipping")
            continue
        
        print(f"Running AI assessment for Loan #{loan.id} (Amount: {loan.amount}, Income: {user.income})...")
        ai_result = predict_risk(
            loan_amount=loan.amount,
            tenure=loan.tenure,
            interest_rate=loan.interest_rate,
            income=user.income
        )
        
        if ai_result.get("success"):
            loan.risk_score = ai_result.get("risk_score")
            loan.risk_category = ai_result.get("risk_category")
            loan.ai_recommendation = ai_result.get("ai_recommendation")
            loan.ai_confidence = ai_result.get("ai_confidence")
            loan.fraud_risk = ai_result.get("fraud_risk")
            loan.fraud_confidence = ai_result.get("fraud_confidence")
            print(f"  -> Risk: {loan.risk_score} ({loan.risk_category}), AI: {loan.ai_recommendation}, Fraud: {loan.fraud_risk}")
        else:
            print(f"  -> AI assessment failed: {ai_result}")
    
    db.session.commit()
    print("\nAll loans updated with AI assessment!")