import streamlit as st
import random


if "score" not in st.session_state:
    st.session_state.score = 0
if "question" not in st.session_state:
    st.session_state.question = None
if "answer" not in st.session_state:
    st.session_state.answer = None


def generate_question():
    num1 = random.randint(1, 10)
    num2 = random.randint(1, 10)
    operation = random.choice(["+", "-", "*"])

    if operation == "+":
        answer = num1 + num2
    elif operation == "-":
        answer = num1 - num2
    else:
        answer = num1 * num2

    question = f"{num1} {operation} {num2}"
    return question, answer


if st.session_state.question is None:
    st.session_state.question, st.session_state.answer = generate_question()

st.title("Math Quiz Game")
st.write("Test your math skills! Answer correctly to earn points.")


st.write(f"Question: {st.session_state.question}")


user_answer = st.text_input("Your Answer", "")

if st.button("Submit Answer"):
    try:
        user_answer = int(user_answer)
        if user_answer == st.session_state.answer:
            st.success("Correct! Well done.")
            st.session_state.score += 1
        else:
            st.error(f"Incorrect. The correct answer was {st.session_state.answer}.")

        
        st.session_state.question, st.session_state.answer = generate_question()
    except ValueError:
        st.error("Please enter a valid number.")


st.write(f"Your Score: {st.session_state.score}")