Exercício - Vendas de frutas

import os

#sales_lastmonth = []
'''
This list will be used in next revision to store monthly sales 
'''

def menu_heading(heading):
    '''
    Define the menu Heading for each menu in the program.
    Input: Taken from defined text when the function is applied
    Output: Print the defined heading on the current menu
    '''
    print(heading)

def wrong_value():
    '''
    Warns the user input is invalid and give user another try
    Input: Called in 'def itens()' when user enter a string/bolean value
    Output: Make user aware of mistake ans return  to previous menu
    '''
    os.system('clear')
    input('Enter a whole number. Press Enter to try again\n')
    itens()


def itens():
    '''
    Menu used to collect info from user (amount of itens) and then calculate which entry has the highest value.
    Input: User
    Output: Which of the intens in the list had the highest sales
        In next revision will be added the functionality to append to a list to save the sales records
    '''
    menu_heading('Inventory Update\n')

    try:
        apples= int(input('Enter the amount of apples sold: \n'))
        bananas = int(input('Enter the amount of bananas sold: \n'))
            #total = {'Bananas':bananas, 'Apples':apples}
            #sales_lastmonth.append(total)
    except ValueError:
        wrong_value()
        
    
    if apples > bananas:
        print('Apples sold more than bananas')
    elif apples < bananas:
        print('Bananas sold more than Apples')
    else:
        print('Both itens sold the same amount')
    return

itens()