Update app.py

app.py

@@ -224,6 +224,9 @@ class SuperSmartAgent:
         builder = StateGraph(AgentState)
     
         # Nodes
+        builder.add_node("check_special_tools", check_special_tools)
+        builder.add_node("use_web_search", use_web_search)
+        builder.add_node("use_video_analysis", use_video_analysis)
         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)
@@ -232,16 +235,10 @@ class SuperSmartAgent:
         builder.add_node("generate_code", generate_code)
         builder.add_node("fallback", fallback)
         
-        # Edges
-        #####################################################
         # Entry point: special tool detection first
         builder.set_entry_point("check_special_tools")
 
-        # Special tool path (web/video) → END
-        builder.add_node("check_special_tools", check_special_tools)
-        builder.add_node("use_web_search", use_web_search)
-        builder.add_node("use_video_analysis", use_video_analysis)
-        
+        # Tool-based branching
         builder.add_conditional_edges(
             "check_special_tools",
             lambda s: "use_web_search" if s.get("use_tool") == "web"
@@ -251,23 +248,26 @@ class SuperSmartAgent:
         builder.add_edge("use_web_search", END)
         builder.add_edge("use_video_analysis", END)
         
-        # Reasoning pipeline (for non-tool questions)
+        # Reasoning pipeline (non-tool questions)
         builder.add_edge("check_reversed", "fix_question")
         builder.add_edge("fix_question", "check_riddle_or_trick")
         
         builder.add_conditional_edges(
             "check_riddle_or_trick",
-            lambda s: "solve_riddle" if s["is_riddle"] else "check_python_suitability"
+            lambda s: "solve_riddle" if s.get("is_riddle") else "check_python_suitability"
         )
         builder.add_edge("solve_riddle", END)
         
         builder.add_conditional_edges(
             "check_python_suitability",
-            lambda s: "generate_code" if s["is_python"] else "fallback"
+            lambda s: "generate_code" if s.get("is_python") else "fallback"
         )
         builder.add_edge("generate_code", END)
         builder.add_edge("fallback", END)
 
+        
+  
+
             
         graph = builder.compile()
         return graph