from app import app, db
from models import User
from werkzeug.security import generate_password_hash

def create_users():
    with app.app_context():
        # Delete existing users
        db.session.query(User).delete()
        db.session.commit()

        # Create admin user
        admin = User(
            username='admin',
            password=generate_password_hash('adminpass'),
            is_admin=True
        )
        db.session.add(admin)

        # Create gallery users
        for i in range(1, 6):
            gallery_user = User(
                username=f'gallery{i}_user',
                password=generate_password_hash('pass'),
                is_admin=False,
                gallery_id=i
            )
            db.session.add(gallery_user)

        db.session.commit()
        print('Users created successfully')

if __name__ == '__main__':
    create_users()