Задача: Тренажер таблицы умножения

import random


# функция вывода таблицы умножения
def print_mult_table():
    t1 = f"{'-'*4}+{'-'*38}"
    print(t1, end='\n')
    print(f"| x |   1   2   3   4   5   6   7   8   9 |", end='\n')
    print(t1, end='\n')
    for i in range(1, 10):
        print(f"| {i} |", end='')
        for j in range(1, 10):
            print(f"  {j*i:>2}", end='')
        print(' |', end='\n')
    print(t1, end='\n')


# функция тренажера таблицы умножения
def trainer():
    while True:
        n1 = random.choice(range(1, 10))
        n2 = random.choice(range(1, 10))
        res = int(input(f"{n1} * {n2} = "))
        if res == n1 * n2:
            print("Верно!")
        else:
            print("Вы ошиблись!")
            print(f"{n1} * {n2} = {n1 * n2}")

        key = input("Продолжить / Меню (any_key / m): ")
        if key == 'm':
            return


def main():
    print("Тренажер таблицы умножения, версия 1.0.")
    print()
    while True:
        print("Вывести таблицу умножения: t")
        print("Запустить тренажер: r")
        print("Выйти из программы: q")
        menu = input("Выберите действие: ")
        if menu == 't':
            print_mult_table()
        elif menu == 'r':
            trainer()
        elif menu == 'q':
            return


if __name__ == '__main__':
    main()