main.py

#!/usr/bin/env python
"""
Jogo de Damas usando modelos LLM e Pydantic

Este script inicia um jogo de damas entre dois modelos LLM locais (Ollama).
"""

import argparse
import sys
from engine import MotorJogo

def obter_argumentos():
    """Processa os argumentos de linha de comando"""
    parser = argparse.ArgumentParser(description="Jogo de damas entre modelos LLM usando Pydantic")
    
    parser.add_argument("--modelo-brancas", default="llama3.1", help="Nome do modelo Ollama para peças brancas")
    parser.add_argument("--modelo-pretas", default="mistral", help="Nome do modelo Ollama para peças pretas")
    parser.add_argument("--max-jogadas", type=int, default=10, help="Número máximo de jogadas")
    
    return parser.parse_args()

def main():
    """Função principal"""
    args = obter_argumentos()
    
    try:
        # Mostra configuração
        print("Jogo de Damas com modelos LLM")
        print(f"Modelos Ollama disponíveis:")
        print(f"- {args.modelo_brancas} (Peças Brancas)")
        print(f"- {args.modelo_pretas} (Peças Pretas)")
        print(f"Limite de jogadas: {args.max_jogadas}")
        
        # Inicia o motor do jogo
        motor = MotorJogo(
            modelo_brancas=args.modelo_brancas,
            modelo_pretas=args.modelo_pretas,
            max_jogadas=args.max_jogadas
        )
        
        # Inicia o jogo
        tabuleiro_final, historico, arquivo_relatorio = motor.iniciar_jogo()
        
        print(f"\nJogo concluído! Confira o relatório completo em: {arquivo_relatorio}")
        
    except KeyboardInterrupt:
        print("\nJogo interrompido pelo usuário.")
        sys.exit(1)
    except Exception as e:
        print(f"Erro ao executar o jogo: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()