import turtle
import pandas

FONT = ("Arial", 15, "normal")

screen = turtle.Screen()
screen.title("U.S States Game")
image = "blank_states_img.gif"
screen.addshape(image)

turtle.shape(image)

data = pandas.read_csv("50_states.csv")
state_list = data["state"].tolist()

answer_list = []

while len(answer_list) < 50:
    answer_state = screen.textinput(title=f"{len(answer_list)}/50 States correct", prompt="What's another state's name?")
    answer_title = answer_state.title()
    if answer_title == "Exit":
        for state in answer_list:
            if state in state_list:
                state_list.remove(state)
        # data_dict = {
        #     "state": state_list
        # }
        df = pandas.DataFrame(state_list)
        df.to_csv("states_to_learn.csv")
        break

    if answer_title in state_list:
        answer_list.append(answer_title)
        cor = data[data["state"] == answer_title]
        # cor_list = cor.values.tolist()
        # xy_cor = (cor_list[0][1], cor_list[0][2])
        state = turtle.Turtle()
        state.penup()
        state.hideturtle()
        # state.setpos(xy_cor)
        state.goto(int(cor.x), int(cor.y))
        state.write(answer_title, align="center", font=FONT)
        score = len(answer_list)