Update app.py

app.py

@@ -20,28 +20,29 @@ class AgentState(TypedDict):
 
 class SuperSmartAgent:
     def __init__(self):
-        self.graph = self._build_graph() # Build graph after initializing wiki_wiki
+        self.graph = self._build_graph()
 
     def _build_graph(self):
         # Define the graph
         workflow = StateGraph(AgentState)
-    
-        # Only keep relevant nodes
+
+        # Add nodes
         workflow.add_node("check_reversed", self.check_reversed)
         workflow.add_node("fix_question", self.fix_question)
-    
-        # Entry point
+        workflow.add_node("generate_response", self.generate_response)  # New node for response generation
+
+        # Set entry point
         workflow.set_entry_point("check_reversed")
-    
-        # Transitions
+
+        # Add edges
         workflow.add_conditional_edges(
             "check_reversed",
-            lambda state: "fix_question" if state["is_reversed"] else END,
+            lambda state: "fix_question" if state["is_reversed"] else "generate_response",
         )
-        workflow.add_edge("fix_question", END)
-    
-        return workflow.compile()
+        workflow.add_edge("fix_question", "generate_response")
+        workflow.add_edge("generate_response", END)
 
+        return workflow.compile()
 
     def __call__(self, question: str) -> str:
         """
@@ -51,11 +52,7 @@ class SuperSmartAgent:
         initial_state = AgentState(
             question=question,
             response="",
-            is_reversed=False,
-            is_riddle=False,
-            is_python=False,
-            needs_reasoning=False,
-            is_wiki=False
+            is_reversed=False
         )
         final_state = self.graph.invoke(initial_state)
         return final_state["response"]
@@ -86,6 +83,29 @@ class SuperSmartAgent:
             state["question"] = state["question"][::-1]
         return state
 
+    def generate_response(self, state):
+        """
+        Generate a response to the question.
+        This is a simplified version that just echoes the question for demonstration.
+        """
+        question = state["question"]
+
+        # Simple response generation logic
+        if not question.strip():
+            state["response"] = "No question detected."
+        else:
+            # For reversed questions, we've already fixed them, so now we can answer
+            if "what is" in question.lower():
+                state["response"] = "This is a generic answer to a 'what is' question."
+            elif "how to" in question.lower():
+                state["response"] = "This is a generic answer to a 'how to' question."
+            elif "?" in question:  # Simple check if it's a question
+                state["response"] = f"I understand you're asking: '{question}'. Here's a generic response."
+            else:
+                state["response"] = f"I've processed your statement: '{question}'. Here's a generic acknowledgment."
+
+        return state
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):