Verifica se uma data é um ambigrama

ambigram.py

representations = {
    '0': ('---', '- -', '- -', '- -', '---'),
    '1': (' - ', ' - ', ' - ', ' - ', ' - '),
    '2': ('---', '  -', '---', '-  ', '---'),
    '3': ('---', '  -', '---', '  -', '---'),
    '4': ('- -', '- -', '---', '  -', '  -'),
    '5': ('---', '-  ', '---', '  -', '---'),
    '6': ('---', '-  ', '---', '- -', '---'),
    '7': ('---', '  -', '  -', '  -', '  -'),
    '8': ('---', '- -', '---', '- -', '---'),
    '9': ('---', '- -', '---', '  -', '---'),
}

# method taken from https://stackoverflow.com/a/58721716/18081350
def seven_segment(number):
    digits = [representations[digit] for digit in str(number)]
    outpewt=''
    for i in range(5):
        outpewt+=("  ".join(segment[i] for segment in digits) + "\n")
    return outpewt

def ambigrammed_segment(number):
    digits = [representations[digit] for digit in str(number)]
    block=''
    for i in range(4,-1,-1):
        for idx, segment in enumerate(digits):
            if segment[i] == '  -':
                block += '-  '
            elif segment[i] == '-  ':
                block += '  -'
            else:
                block += segment[i]
            if (idx < len(str(number)) - 1):
                block += '  '
        block+='\n'
    return block

inpewt = input('Digite uma data (DDMMAAAA): ')
original = seven_segment(inpewt)
reversed = ambigrammed_segment(inpewt)
print(original)
print(reversed)

if original == reversed:
    print('Ambigrama(x)')
else:
    print('Ambigrama( )')