Shrook21/Code-Assistant

1from langgraph.graph import StateGraph, END
2from agents.state import StateAgent
3from agents.nodes import chat, generate_code, explain_code, fallback, router
4
5def get_app():
6    graph = StateGraph(StateAgent)
7    graph.add_node('chat', chat)
8    graph.add_node('generate_code', generate_code)
9    graph.add_node('explain_code', explain_code)
10    graph.add_node('fallback', fallback)
11    
12    graph.set_entry_point('chat')
13    graph.add_conditional_edges('chat', router, {
14        'generate': 'generate_code',
15        'explain': 'explain_code',
16        'unclear': 'fallback'
17    })
18    graph.add_edge('generate_code', END)
19    graph.add_edge('explain_code', END)
20    graph.add_edge('fallback', END)
21    
22    return graph.compile()
23