main.py

import datetime

cost = 0
extraCost = 0
done = False
shoppingList = []

print("-----| Coffee Menu |-----\n")
print("-------| Black |------------|")
print("1. Espresso..............2.50 EUR")
print("2. Americano.............3.25 EUR\n")
print("-------| White |------------|")
print("3. Cappuccino............3.00 EUR")
print("4. Latte.................3.20 EUR")
print("5. Flat White............3.50 EUR\n")
print("-------| Specialties |------------|")
print("6. Mocha.................3.75 EUR")
print("7. Caramel Macchiato.....4.00 EUR\n")

coffee_prices = {
    1: ("Espresso", 2.50),
    2: ("Americano", 3.25),
    3: ("Cappuccino", 3.00),
    4: ("Latte", 3.20),
    5: ("Flat White", 3.50),
    6: ("Mocha", 3.75),
    7: ("Caramel Macchiato", 4.00)
}

extra_prices = {
    "a": ("Extra Sugar", 0.25),
    "b": ("Extra Milk", 0.50)
}

while not done:
    choice = int(input("Make your choice (1-7): "))
    if choice in coffee_prices:
        drink, price = coffee_prices[choice]
        cost += price
        shoppingList.append(f"{drink}..............{price:.2f} EUR")

        extra_added = {"a": False, "b": False}
        extraDone = False
        while not extraDone:
            print("\n-------| Extras |------------|")
            if not extra_added["a"]:
                print("A. Extra Sugar.............0.25 EUR")
            if not extra_added["b"]:
                print("B. Extra Milk...............0.50 EUR\n")
            print("N. No extras\n")

            extra = input("Would you like to add extras (a/b)? (n for no extra): ").lower()

            if extra in extra_prices:
                extra_name, extra_price = extra_prices[extra]
                extraCost += extra_price
                shoppingList.append(f"- {extra_name}..............{extra_price:.2f} EUR")
                extra_added[extra] = True
            elif extra == "n":
                extraDone = True
            else:
                print("Invalid choice, please try again")

        more = input("Would you like to order another drink (y/n): ").lower()
        if more == "n":
            done = True

    else:
        print("Invalid choice, please try again")

print("\n-----| Your Order Summary |-----")
for item in shoppingList:
    print(item)
print(f"\nTotal: {cost + extraCost:.2f} EUR")

while True:
    good = input("Is your order correct (y/n)? ").lower()

    if good == "y":
        print("\n-----| Receipt |-----\n")
        print("     Receipt of sale       ")
        print("       Coffee Shop      \n")  
        print("-------------------------")
        print("Date: ", datetime.datetime.now())
        print("-------------------------\n")

        for item in shoppingList:
            print(item)
        total = cost + extraCost
        print("\n-------------------------")
        print(f"\nTotal: {total:.2f} EUR")
        print("-------------------------")
        print("Thank you for your purchase!")
        break 
    elif good == "n":
        remove_item = input("Would you like to remove an item (y/n)? ").lower()

        if remove_item == "y":
            print("Which item would you like to remove? (Enter the number corresponding to the item) ")
            for idx, item in enumerate(shoppingList, 1):
                print(f"{idx}. {item}")
        
            try:
                item_to_remove = int(input("Enter the number of the item you would like to remove: "))
                if 1 <= item_to_remove <= len(shoppingList):
                    shoppingList.pop(item_to_remove - 1)
                    print("\n-----| Updated Order Summary |-----")
                    for item in shoppingList:
                        print(item)
                    print(f"\nTotal: {cost + extraCost:.2f} EUR")
                else:
                    print("Invalid choice, please try again")
            except ValueError:
                print("Invalid choice, please try again")
        elif remove_item == "n":
            print("No items were removed.")
    else:
        print("Invalid choice, please try again")