import os
from src.graph import create_graph
from src.state import PlanningState

def main() -> None:
    # Ensure the OpenAI API key is set
    if "OPENAI_API_KEY" not in os.environ:
        raise RuntimeError("Please set the OPENAI_API_KEY environment variable.")

    task = "Compare Python and JavaScript"
    initial_state: PlanningState = {
        "task": task,
        "plan": None,
        "current_step": 0,
        "results": []
    }

    graph = create_graph()
    final_state = graph.invoke(initial_state)

    print("\n=== Plan ===")
    for i, step in enumerate(final_state["plan"], 1):
        print(f"{i}. {step}")

    print("\n=== Results ===")
    for i, res in enumerate(final_state["results"], 1):
        print(f"[Step {i}] {res}")

    print("\n=== Final Summary ===")
    summary = "\n".join(final_state["results"])
    print(summary)

if __name__ == "__main__":
    main()