import pandas


# functions go here

# Checks user has entered yes / no to a question
def yes_no(question):
    while True:
        response = input(question).lower()

        if response == "yes" or response == "y":
            return "yes"

        elif response == "no" or response == "n":
            return "no"

        else:
            print("Please enter yes or no")


# checks that user response is not blank
def not_blank(question):
    while True:
        response = input(question)
        if response == "":
            print("Sorry this can't be blank. Please try again")
        else:
            return response


# checks users enter an integer to given question
def num_check(question, low, high):
    while True:
        error = f"Enter a number between {low} and {high}"

        try:
            response = int(input(question))
            if low <= response <= high:
                return response
            else:
                print(error)

        except ValueError:
            print("Please enter an integer.")

# Currency formatting function
def currency(x):
    return "${:.2f}".format(x)

# Calculate the room price based on room type
def calc_room_price(var_type):
        # room is $30 for a single
        if var_type == 1:
            price = 30

        # room is $35 for two beds
        elif var_type == 2:
            price = 35

        # room is $45 for queen
        elif var_type == 3:
            price = 45

        # room is $50 for king
        elif var_type == 4:
            price = 50

        # room is $65 for family room
        elif var_type == 5:
            price = 65

        return price


# main routine starts here
print("Welcome to the Melia Hotel, thank you for choosing to stay with us.\n"
      "We have everything from an indoor pool, a 24/7 gym, and even a kids club.")


while True:

# set maximum number of rooms below
    MAX_ROOMS = 30
    rooms_sold = 0

    yes_no_list = ["yes", "no"]
    payment_list = ["cash", "credit"]
    user_room = []
    user_stay = []
    user_booking_total = []

# loop to sell rooms
    name = not_blank("Enter your name (or 'xxx' to quit) ")

# age checker
    age = num_check("How old are you? ", 18, 120)

# age checker
    if 18 <= age <= 120:
        pass
    elif age < 18:
        print("Sorry you are too young to book a room")
    elif age > 120:
        print("?? That looks like a typo, please try again.")

# Dictionaries to hold room details
    all_names = ["Single","Double", "Queen", "King", "Family's",]
    all_room_costs = [30, 35, 45, 50, 65]

    hotel_reservation_dict = {
        "Room": all_names,
        "Room Prices": all_room_costs
    }

    hotel_reservation_frame = pandas.DataFrame(hotel_reservation_dict)

# Output table with room data
    print(hotel_reservation_frame)

    room_selection = num_check("Which room do you select? ", 0, 4)
    room_stay = num_check("How long would you like to stay? ", 1, 60)
    room_selection_names = {
        0: "Single room",
        1: "Double room",
        2: "Queen room",
        3: "King room",
        4: "Family's room"
    }
    if room_selection in room_selection_names:
        print(f'{name} you have selected a', (room_selection_names[room_selection]), 'for', room_stay, 'days')

    room_total = calc_room_price(room_selection + 1)
    user_room.append((room_selection_names[room_selection]))
    booking_price = room_stay*room_total
    user_booking_total.append(booking_price)
    user_stay.append(room_stay)
    rooms_sold += 1

    print("")


    booking_reservation_dict = {
            "Room": user_room,
            "Days booked": user_stay,
            "Room Prices": user_booking_total
        }

    booking_reservation_frame = pandas.DataFrame(booking_reservation_dict)

# Output table with room data
    print(booking_reservation_frame)

    # calculate room cost
    print("Room price: ${:.2f}, Booked days: {}".format(room_total, room_stay))
    print("Your total cost will be: ${:.2f}".format(booking_price))

    # Ask the user for their payment method
    payment_method = input("Would you like to pay with cash or credit? ")

    # Check the payment method and calculate the total
    if payment_method.lower() == "cash" or payment_method.lower() == "ca":
        total = booking_price  # Example total without surcharge for cash payment
        print("Total amount: $", total)
    elif payment_method.lower() == "credit" or payment_method.lower() == "cr":
        total = booking_price  # Example total without surcharge for credit payment
        surcharge = 0.90
        total_with_surcharge = total + surcharge
        print("Total amount with surcharge: $", total_with_surcharge)
    else:
        print("Invalid payment method. Please choose 'cash' or 'credit'.")

    print()
    user_choice = yes_no("Would you like to confirm the booking?")

    if user_choice.lower() == "yes":
        print("Booking confirmed. Thank you!\n"
              "We hope you have a great stay.")
        new_booking = yes_no("Do you want to make another booking?")
        if new_booking == "yes":
            continue
        else:
            break
    else:
        break
print("Program has ended.")