#All Rights Reserved John Salguero
#Starts the backend to my Youtube stream

from problem_generator import generate_problem
from steps_generator import generate_steps
from sympy import init_printing

#define the entry point to the programs
def main():
    init_printing(order='lex')
    problem = generate_problem()
    steps = generate_steps(problem);

    print("Generated Problem:")
    print(problem)
    
    print("Steps:")
    pretty_print_steps(steps)

def pretty_print_steps(steps):
    print("\n" + "=" * 50)

    for i, step in enumerate(steps, start=1):
        print(f"\nStep {i}")
        print("-" * 50)
        print(f"Before: {step.get('before', '')}")
        print(f"After:  {step.get('after', '')}")

        for key in step:
            if key not in ("before", "after"):
                print(f"{key.capitalize()}: {step[key]}")

    print("\n" + "=" * 50)

#Starts the program
if __name__ == "__main__":
    main()