import asyncio
from motor.motor_asyncio import AsyncIOMotorClient

async def seed_database():
    # Connect to local MongoDB
    client = AsyncIOMotorClient("mongodb+srv://sudhanp2004:root@cluster0.wsmrt.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0")
    db = client.medical_adjudication
    
    mock_users = [
        # TC010: Cashless Network (Deepak)
        {"member_id": "EMP010", "name": "Deepak Shah", "policy_status": "ACTIVE", "opd_wallet_balance": 50000.00},
        
        # TC009: Excluded Treatment (Anita)
        {"member_id": "EMP009", "name": "Anita Desai", "policy_status": "ACTIVE", "opd_wallet_balance": 25000.00},
        
        # TC008: Fraud Velocity (Ravi)
        {"member_id": "EMP008", "name": "Ravi Menon", "policy_status": "ACTIVE", "opd_wallet_balance": 15000.00},
        
        # Gatekeeper Test 1: Empty Wallet
        {"member_id": "EMP001", "name": "Zero Balance", "policy_status": "ACTIVE", "opd_wallet_balance": 0.00},
        
        # Gatekeeper Test 2: Inactive Policy
        {"member_id": "EMP002", "name": "Expired Policy", "policy_status": "INACTIVE", "opd_wallet_balance": 10000.00},

        {"member_id": "EMP000", "name": "John Doe", "policy_status": "ACTIVE", "opd_wallet_balance": 50000.00},
    ]
    
    # Wipe old test data and insert fresh data
    await db.users.delete_many({})
    await db.claims.delete_many({}) 
    
    await db.users.insert_many(mock_users)
    print("MongoDB seeded successfully! Users and wallets are ready.")

if __name__ == "__main__":
    asyncio.run(seed_database())