demo.py

import sys

from dotenv import load_dotenv
from eval.game import Game, Player1, Player2
from loguru import logger

logger.remove()
logger.add(sys.stdout, level="INFO")

load_dotenv()


def main():
    # Environment Settings

    game = Game(
        render=True,
        save_game=True,
        player_1=Player1(
            nickname="Baby",
            model="mistral:mistral-small-latest",
        ),
        player_2=Player2(
            nickname="Daddy",
            model="mistral:mistral-small-latest",
        ),
    )

    game.run()
    return 0


if __name__ == "__main__":
    main()