import sys
import logging
from langgraph.graph import START, END, StateGraph
from exception import MyException
import sys
# ------------------- Schemas -------------------
from src.Agents.models.Resume_model import ResumeState
# ------------------- LLM -------------------
from src.Agents.nodes.resume_builder import resume_maker
graph = StateGraph(state_schema=ResumeState)
graph.add_node("resume_maker", resume_maker)
graph.add_edge(START, "resume_maker")
graph.add_edge("resume_maker", END)
graph = graph.compile()

try:
    with open("resume_graph.png","wb") as f:
        f.write(graph.get_graph().draw_mermaid_png())
except Exception as e:
    raise MyException(e,sys)
        

# ------------------- Main -------------------
async def create_resume_schema(userDetails: str):
    logging.info("Entering create_resume_schema")
    try:
        final_state = await graph.ainvoke({"userDetails": userDetails})
        logging.info("Exiting create_resume_schema")
        return final_state
    except Exception as e:
        raise MyException(e, sys)