Update app.py

app.py

@@ -27,7 +27,7 @@ class State(TypedDict):
     name: str
     birthday: str
     input : str
-    code = str
+    code : str
  
 # LLM
 llm = ChatGroq(model="deepseek-r1-distill-llama-70b")
@@ -80,11 +80,13 @@ def router(state: State):
 builder = StateGraph(State)
 builder.add_node("AI_Assistance", ai_assistance)
 builder.add_node("Generate_Code", generate_code)
-builder.add_node("Router", router)
-
+builder.add_conditional_edges("Router", router, {
+    "AI_Assistance": "AI_Assistance",
+    "Generate_Code": "Generate_Code",
+})
 builder.set_entry_point("Router")
-builder.add_edge("Router", "AI_Assistance")
-builder.add_edge("Router", "Generate_Code")
+# builder.add_edge("Router", "AI_Assistance")
+# builder.add_edge("Router", "Generate_Code")
 builder.add_edge("AI_Assistance", END)
 builder.add_edge("Generate_Code", END)