Main.py

# main.py

from Nominees import view_all_nominees, edit_nominee_name
from VoterRegistration import create_new_voter, view_all_voters, view_specific_voter
from VoteCasting import cast_vote
from VoteCounting import count_votes, view_results, primary_winner, preferential_winner

def main():
    while True:
        print("\n====== STUDENT UNION ELECTION SYSTEM ======")
        print("1. Manage Nominees")
        print("2. Voter Registration")
        print("3. Cast Vote")
        print("4. Counting")
        print("5. Exit")

        try:
            choice = int(input("Enter your choice: "))
        except ValueError:
            print("❌ Please enter a number.")
            continue

        if choice == 1:
            print("\n--- Nominees Menu ---")
            print("1. View all nominees")
            print("2. Edit nominee name")
            sub = int(input("Enter choice: "))
            if sub == 1:
                view_all_nominees()
            elif sub == 2:
                edit_nominee_name()

        elif choice == 2:
            print("\n--- Voter Registration Menu ---")
            print("1. Create voter")
            print("2. View all voters")
            print("3. Search voter")
            sub = int(input("Enter choice: "))
            if sub == 1:
                create_new_voter()
            elif sub == 2:
                view_all_voters()
            elif sub == 3:
                view_specific_voter()

        elif choice == 3:
            cast_vote()

        elif choice == 4:
            results = count_votes()
            view_results(results)
            w_ballot, w_name = primary_winner(results)
            if w_name:
                print(f"🎉 Primary Winner: {w_name} (Ballot {w_ballot})")
            else:
                b, n = preferential_winner()
                print(f"🎉 Preferential Winner: {n} (Ballot {b})")

        elif choice == 5:
            print("👋 Exiting...")
            break
        else:
            print("❌ Enter a valid choice.")

if __name__ == "__main__":
    main()