import time
import os

def limpar_console():
    os.system('cls' if os.name == 'nt' else 'clear')

def simular_votacao(partidos, maioria_necessaria, descricao):
    votacao_total_favor = 0
    votacao_total_contra = 0
    votacao_abstencao = 0

    for partido, votos in partidos.items():
        while True:
            voto = input(f"Votação do {partido.upper()}: Favor [F], Contra [C] ou Abstenção [A]: ").lower()
            if voto == 'f':
                votacao_total_favor += votos
                break
            elif voto == 'c':
                votacao_total_contra += votos
                break
            elif voto == 'a':
                votacao_abstencao += votos
                break
            else:
                print("\n Escolha não válida. Tente outra vez.")

    print(f"\nResultado da Votação para: {descricao}")
    if votacao_total_favor >= maioria_necessaria:
        print(f"\033[32mA proposta foi APROVADA com {votacao_total_favor} a favor, "
              f"{votacao_total_contra} contra e {votacao_abstencao} abstenção.\033[m")
    else:
        print(f"\033[31mA proposta foi REJEITADA com {votacao_total_favor} a favor, "
              f"{votacao_total_contra} contra e {votacao_abstencao} abstenção.\033[m")
    time.sleep(3)

# Dicionário de partidos (definido no início para estar disponível em todas as opções)
partidos = {
    "Partido Socialista": 58,
    "Aliança Democrática": 89,
    "Iniciativa Liberal": 9,
    "Partido Comunista Português": 3,
    "Bloco de Esquerda": 1,
    "Pessoas Animais e Natureza": 1,
    "Livre": 6,
    "CHEGA": 58,
    "Juntos pelo Povo": 1
}

escolha = True
while escolha:
    limpar_console()
    print("Simulação Provisória da Votação de Propostas Legislativas - 18 de Maio de 2025")
    print("""
    1. Simular Proposta Normal 
    2. Mudar a Constituição 
    0. Sair
    """)
    escolha = input("Escolha uma opção:  ")

    if escolha == "1":
        simular_votacao(partidos, maioria_necessaria=116, descricao="Proposta Legislativa Normal")

    elif escolha == "2":
        simular_votacao(partidos, maioria_necessaria=154, descricao="Mudança da Constituição")

    elif escolha == "0":
        print("\nAdeus")
        escolha = False

    else:
        print("\nEscolha não válida. Tente outra vez.")
        time.sleep(2)