"""
Felix Muzny
DS 2000
Lecture 4
September 16, 2024

A program to visualize someone's croissant habits
"""
import matplotlib.pyplot as plt

def main():
    print("lec 4 - croissants w/ input")

    # step 1 - gather data
    rating1 = float(input("How good is the croissant (scale of 1 - 10)? "))
    time1 = float(input("How long are you willing to wait (minutes)? "))
    rating2 = float(input("How good is the croissant (scale of 1 - 10)? "))
    time2 = float(input("How long are you willing to wait (minutes)? "))
    rating3 = float(input("How good is the croissant (scale of 1 - 10)? "))
    time3 = float(input("How long are you willing to wait (minutes)? "))

    # let's also ask the user what color the points should be!
    user_color = input("what color should the points be? ")

    # step 2 - any computations?
    # none! :)

    # step 3 - communication - give the user information!
    plt.plot(rating1, time1, "o", color=user_color)
    plt.plot(rating2, time2, "h", color=user_color)
    plt.plot(rating3, time3, "*", color=user_color)

    plt.title("Croissant goodness vs. time willing to wait")
    plt.xlabel("Goodness (scale of 1 - 10)")
    plt.ylabel("Time (minutes)")
    plt.show()

main()