Update app.py

app.py

@@ -12,47 +12,49 @@ import string
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # Define the State for the graph
+import string
+from langgraph.graph import StateGraph, END
+from typing import TypedDict
+
 class AgentState(TypedDict):
     question: str
     response: str
     is_reversed: bool
-
+    is_riddle: bool
 
 class SuperSmartAgent:
     def __init__(self):
         self.graph = self._build_graph()
 
     def _build_graph(self):
-        # Define the graph
         workflow = StateGraph(AgentState)
 
-        # Add nodes
         workflow.add_node("check_reversed", self.check_reversed)
         workflow.add_node("fix_question", self.fix_question)
-        workflow.add_node("generate_response", self.generate_response)
+        workflow.add_node("check_riddle_or_trick", self.check_riddle_or_trick)
+        workflow.add_node("solve_riddle", self.solve_riddle)
 
-        # Set entry point
         workflow.set_entry_point("check_reversed")
 
-        # Add edges
         workflow.add_conditional_edges(
             "check_reversed",
-            lambda state: "fix_question" if state["is_reversed"] else "generate_response",
+            lambda state: "fix_question" if state["is_reversed"] else "check_riddle_or_trick",
+        )
+        workflow.add_conditional_edges(
+            "check_riddle_or_trick",
+            lambda state: "solve_riddle" if state["is_riddle"] else END,
         )
-        workflow.add_edge("fix_question", "generate_response")
-        workflow.add_edge("generate_response", END)
+        workflow.add_edge("fix_question", "check_riddle_or_trick")
+        workflow.add_edge("solve_riddle", END)
 
         return workflow.compile()
 
     def __call__(self, question: str) -> str:
-        """
-        Runs the agent's graph with the given question.
-        """
-        # Initialize state for each new question
         initial_state = AgentState(
             question=question,
             response="",
-            is_reversed=False
+            is_reversed=False,
+            is_riddle=False
         )
         final_state = self.graph.invoke(initial_state)
         return final_state["response"]
@@ -83,47 +85,31 @@ class SuperSmartAgent:
             state["question"] = state["question"][::-1]
         return state
 
-    def generate_response(self, state):
-        """
-        Generate a response to the question.
-        """
-        question = state["question"]
-    
-        # Check if the question is about finding the opposite of a word
-        if "opposite of the word" in question.lower():
-            # Extract the word whose opposite is to be found
-            parts = question.split()
-            try:
-                # Find the index of "the" before "word"
-                the_index = parts.index("the")
-                word_index = parts.index("word")
-                if word_index - the_index == 1:  # Ensure "the" and "word" are consecutive
-                    word = parts[word_index + 1].strip('."\'')
-                    # Determine the opposite
-                    opposites = {
-                        "left": "right",
-                        "right": "left",
-                        "up": "down",
-                        "down": "up",
-                        "hot": "cold",
-                        "cold": "hot"
-                    }
-                    opposite = opposites.get(word.lower(), "unknown")
-                    state["response"] = opposite
-            except (ValueError, IndexError):
-                state["response"] = "Could not determine the opposite."
-        elif "how many" in question.lower():
-            state["response"] = "I don't have the capability to search for specific numbers."
-        elif "what is the highest number" in question.lower():
-            state["response"] = "I don't have the capability to search for specific numbers."
+    def check_riddle_or_trick(self, state):
+        q = state["question"].lower()
+        keywords = ["opposite of", "if you understand", "riddle", "trick question", "what comes next", "i speak without"]
+        state["is_riddle"] = any(kw in q for kw in keywords)
+        return state
+
+    def solve_riddle(self, 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"] = f"I've processed your statement: '{question}'. Here's a generic acknowledgment."
-    
+            state["response"] = "Could not solve riddle."
         return state
 
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,