dice.py

# Import necessary modules
import random
from dice_art import DiceArt

# ~~ Rolling the dices ~~

# Return a list of integers with lenght num_dice
def roll_dice(num_dice):
    roll_results = []
    for _ in range(num_dice):
        roll = random.randint(1, 6)
        roll_results.append(roll)
    return roll_results

# ~~ App's main code block ~~

# Ask the user until provides a valid number or exit the program
def num_dice():
    while True:
        try:
            print("\nType 0 to exit the program.\n")
            user_input = int(input("How many dices would you like to roll? "))
            
            # Exit the program if the user inputs 0
            if user_input == 0:
                print("\nExiting the program. Goodbye!\n")
                exit()

            # Validate user's input is a number
            elif 1 <= user_input:
                return user_input
            
        # Raise an error if the input is not a number
        except ValueError:
            print("\nInvalid input. Please enter a valid number.\n")

# Play the game
while True:
    roll = roll_dice(num_dice())
    dice_diagram = DiceArt()
    result = dice_diagram.generate_dice_faces_diagram(roll)
    print(result)