Update app.py

app.py

@@ -699,10 +699,21 @@ class SuperSmartAgent:
     
         builder = StateGraph(AgentState)
         
-        # Entry point remains the same
+      # Add all nodes to the builder
+        builder.add_node("check_reversed", check_reversed)
+        builder.add_node("fix_question", fix_question)
+        builder.add_node("check_riddle_or_trick", check_riddle_or_trick)
+        builder.add_node("solve_riddle", solve_riddle)
+        builder.add_node("check_wikipedia_suitability", check_wikipedia_suitability)
+        builder.add_node("check_reasoning_needed", check_reasoning_needed)
+        builder.add_node("general_reasoning_qa", general_reasoning_qa)
+        builder.add_node("search_wikipedia", search_wikipedia)
+        builder.add_node("check_python_suitability", check_python_suitability)
+        builder.add_node("generate_code", generate_code)
+        builder.add_node("fallback", fallback)
+
+        # Set entry point and define edges
         builder.set_entry_point("check_reversed")
-
-        # Edges - updated to match your current workflow
         builder.add_edge("check_reversed", "fix_question")
         builder.add_edge("fix_question", "check_riddle_or_trick")
         builder.add_conditional_edges(
@@ -721,14 +732,14 @@ class SuperSmartAgent:
             "check_python_suitability",
             lambda s: "generate_code" if s.get("is_python") else "fallback"
         )
-        
+    
         # Ending edges
         builder.add_edge("solve_riddle", END)
         builder.add_edge("search_wikipedia", END)
         builder.add_edge("general_reasoning_qa", END)
         builder.add_edge("generate_code", END)
         builder.add_edge("fallback", END)
-        
+    
         graph = builder.compile()
         return graph