# Import the modules
import random
import time
import sympy
from sympy.parsing.sympy_parser import parse_expr
from sympy.solvers import solve
from sympy import Symbol
import winsound

# Define the alarm sound
sound = "alarm.wav"

# Define the difficulty level of the calculus problems
difficulty = 3 # 1: easy, 2: medium, 3: hard

# Define the number of problems to solve
num_problems = 3

# Define the time limit for each problem in seconds
time_limit = 30

# Define the score for each correct answer
score = 10

# Define the total score
total_score = 0

# Define the symbols for the calculus problems
x = Symbol('x')
y = Symbol('y')
z = Symbol('z')

# Define the list of possible calculus problems
problems = [
    # Easy problems
    [
        ("Find the derivative of f(x) = x^2 + 3x - 5", "2*x + 3"),
        ("Find the derivative of f(x) = sin(x) + cos(x)", "cos(x) - sin(x)"),
        ("Find the derivative of f(x) = e^x + ln(x)", "e^x + 1/x"),
        ("Find the integral of f(x) = x^2 + 3x - 5", "x^3/3 + 3*x^2/2 - 5*x"),
        ("Find the integral of f(x) = sin(x) + cos(x)", "-cos(x) + sin(x)"),
        ("Find the integral of f(x) = e^x + ln(x)", "e^x + x*ln(x) - x")
    ],
    # Medium problems
    [
        ("Find the derivative of f(x) = x^3 * sin(x)", "3*x^2 * sin(x) + x^3 * cos(x)"),
        ("Find the derivative of f(x) = ln(x^2 + 1)", "2*x / (x^2 + 1)"),
        ("Find the derivative of f(x) = e^(x^2)", "2*x * e^(x^2)"),
        ("Find the integral of f(x) = x^3 * sin(x)", "-x^3 * cos(x) + 3*x^2 * sin(x) - 6*x * cos(x) + 6 * sin(x)"),
        ("Find the integral of f(x) = ln(x^2 + 1)", "x * ln(x^2 + 1) - 2 * atan(x)"),
        ("Find the integral of f(x) = e^(x^2)", "sqrt(pi) * erf(x) / 2")
    ],
    # Hard problems
    [
        ("Find the derivative of f(x,y) = x^2 * y^3 + sin(x*y)", "2*x * y^3 + x^2 * 3*y^2 + y * cos(x*y)"),
        ("Find the derivative of f(x,y) = ln(x^2 + y^2)", "2*x / (x^2 + y^2)"),
        ("Find the derivative of f(x,y) = e^(x*y)", "y * e^(x*y)"),
        ("Find the integral of f(x,y) = x^2 * y^3 + sin(x*y)", "x^3 * y^3 / 3 - cos(x*y)"),
        ("Find the integral of f(x,y) = ln(x^2 + y^2)", "x * y * ln(x^2 + y^2) - 2 * x * y + 2 * atan(y/x)"),
        ("Find the integral of f(x,y) = e^(x*y)", "e^(x*y) / (x + y)")
    ]
]

# Define a function to check the answer
def check_answer(problem, answer):
    # Parse the problem and the answer into sympy expressions
    problem = parse_expr(problem)
    answer = parse_expr(answer)
    # Check if the problem is a derivative or an integral
    if problem.has(sympy.Derivative):
        # Solve the problem for the derivative
        solution = solve(problem, sympy.Derivative)
        # Check if the answer matches the solution
        return answer == solution[0]
    elif problem.has(sympy.Integral):
        # Solve the

# Solve the problem for the integral
        solution = solve(problem, sympy.Integral)
        # Check if the answer matches the solution
        return answer == solution[0]
    else:
        # Invalid problem
        return False

# Define a function to play the alarm sound
def play_alarm():
    # Play the sound in a loop
    while True:
        winsound.PlaySound(sound, winsound.SND_FILENAME)

# Define a function to stop the alarm sound
def stop_alarm():
    # Stop the sound
    winsound.PlaySound(None, winsound.SND_FILENAME)

# Define a function to start the alarm clock
def start_alarm():
    # Print a welcome message
    print("Welcome to the alarm clock that requires you to solve complex calculus problems in order to turn off the alarm.")
    print("You have to solve {} problems with a difficulty level of {} within {} seconds each.".format(num_problems, difficulty, time_limit))
    print("You will get {} points for each correct answer. Good luck!".format(score))
    # Play the alarm sound
    play_alarm()
    # Loop through the number of problems
    for i in range(num_problems):
        # Pick a random problem from the list
        problem = random.choice(problems[difficulty-1])
        # Print the problem
        print("Problem {}: {}".format(i+1, problem[0]))
        # Start a timer
        start_time = time.time()
        # Get the user input
        user_answer = input("Enter your answer: ")
        # Stop the timer
        end_time = time.time()
        # Calculate the elapsed time
        elapsed_time = end_time - start_time
        # Check if the user input is valid
        try:
            # Parse the user input into a sympy expression
            user_answer = parse_expr(user_answer)
            # Check if the user answer is correct
            if check_answer(problem[0], user_answer):
                # Check if the user answer is within the time limit
                if elapsed_time <= time_limit:
                    # Print a congratulatory message
                    print("Correct! You solved the problem in {:.2f} seconds.".format(elapsed_time))
                    # Update the total score
                    total_score += score
                else:
                    # Print a time out message
                    print("Time out! You took {:.2f} seconds to solve the problem.".format(elapsed_time))
            else:
                # Print a wrong answer message
                print("Wrong! The correct answer is {}.".format(problem[1]))
        except:
            # Print an invalid input message
            print("Invalid input! Please enter a valid mathematical expression.")
    # Stop the alarm sound
    stop_alarm()
    # Print the final score
    print("You have completed the alarm clock challenge. Your final score is {}/{}.".format(total_score, num_problems * score))

# Call the start alarm function
start_alarm()