Update app.py

app.py

@@ -48,6 +48,29 @@ class SuperSmartAgent:
                 print(f"Reversing question: {state['question']}")
                 state["question"] = state["question"][::-1]
             return state
+
+        def check_riddle_or_trick(state):
+            q = state["question"].lower()
+            if "opposite of the word" in q:
+                state["is_riddle"] = True
+            else:
+                state["is_riddle"] = False
+            return state
+
+        def solve_riddle(state):
+            q = state["question"].lower()
+            if "opposite of the word" in q:
+                if "left" in q:
+                    state["response"] = "right"
+                elif "up" in q:
+                    state["response"] = "down"
+                elif "hot" in q:
+                    state["response"] = "cold"
+                else:
+                    state["response"] = "Unknown opposite."
+            else:
+                state["response"] = "Could not solve riddle."
+            return state
         
         def check_python_suitability(state):
             question = state["question"].lower()
@@ -85,14 +108,25 @@ class SuperSmartAgent:
         # Nodes
         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_python_suitability", check_python_suitability)
         builder.add_node("generate_code", generate_code)
         builder.add_node("fallback", fallback)
+    
         
         # Edges
         builder.set_entry_point("check_reversed")
         builder.add_edge("check_reversed", "fix_question")
-        builder.add_edge("fix_question", "check_python_suitability")
+        
+        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"
+        )
+        builder.add_edge("solve_riddle", END)
+        
+        builder.add_node("check_python_suitability", check_python_suitability)
         builder.add_conditional_edges(
             "check_python_suitability",
             lambda s: "generate_code" if s["is_python"] else "fallback"