Untitled

import time
#Dicionário com o "cardápio"
menu = {
    'codigos': {
        'TR': {
            'descricao': 'Sabor Tradicional',
            'P': 6.00,
            'M': 10.00,
            'G': 18.00
        },
        'ES': {
            'descricao': 'Sabor Especial',
            'P': 7.00,
            'M': 12.00,
            'G': 21.00
        },
        'PR': {
            'descricao': 'Sabor Premium',
            'P': 8.00,
            'M': 14.00,
            'G': 24.00
        },
    }
}
# Boas-vindas
print('\n[!] Bem-vindo(a) à sorveteria Marques. (RU: )')
print(
    '''
╔════════╦══════════════════════╦══════════════════╦══════════════════╦════════════════════╗
║ Código ║      Descrição       ║ Tamanho P(500ml) ║ Tamanho M(500ml) ║ Tamanho G (2000ml) ║
╠════════╬══════════════════════╬══════════════════╬══════════════════╬════════════════════╣
║   TR   ║ Sabores Tradicionais ║   R$6,00         ║   R$10,00        ║    R$18,00         ║
║   ES   ║ Sabores Especiais    ║   R$7,00         ║   R$12,00        ║    R$21,00         ║
║   PR   ║ Sabores Premium      ║   R$8,00         ║   R$14,00        ║    R$24,00         ║
╚════════╩══════════════════════╩══════════════════╩══════════════════╩════════════════════╝
    '''
)
#Iniciando o contador com 0
i = 0
# Enquanto a condição for verdadeira, executa o código abaixo
while True:
    #Solicitando as informações, tais como sabores e tamanho
    pot_size = str(input('Digite o tamanho do pote desejado (P/M/G): ')).upper()
    flavor = str(input('Digite o código do sabor desejado (TR/ES/PR): ')).upper()
    #Validando se as informações fornecidas existem
    if (pot_size == 'P' or pot_size == 'M' or pot_size == 'G'):
        if(flavor == 'TR' or flavor == 'ES' or flavor == 'PR'):
            #Exibindo no console o pedido que foi concluido com sequências de escape ANSI
            print('\033[32m' + '[+] Você pediu com sucesso um {} {} de R${:.2f}'.format(menu['codigos'][flavor]['descricao'], pot_size, menu['codigos'][flavor][pot_size]))
            # Atribuindo o valor do pedido ao contador
            i = i + menu['codigos'][flavor][pot_size]
            # Perguntando se o cliente quer efetuar mais algum pedido. Se sim, a execução continua. Se não, mostra um resumo do pedido e um break encerra a execução do While.
            proceed = str(input('\033[0;0m' + '[!] Deseja pedir mais alguma coisa? (S/N): ')).upper()
            if proceed == 'N':
                print('\033[35m' + 'Valor a ser pago: R$', i)
                break
            else:
                continue
        #Se o código está incorreto
        else:
            print('TAMANHO ou CÓDIGO inválido(s).')
    #Se o tamanho está incorreto
    else:
        print('TAMANHO ou CÓDIGO inválido(s).')