Update app.py

app.py

@@ -5,6 +5,7 @@ import inspect
 import pandas as pd
 
 from langgraph.graph import StateGraph, END
+from typing import TypedDict
 
 
 # (Keep Constants as is)
@@ -67,15 +68,13 @@ class SuperSmartAgent:
             state["response"] = "This question doesn't require Python or is unclear."
             return state
         
-        # State schema
-        state_schema = {
-            "question": str,
-            "is_reversed": bool,
-            "is_python": bool,
-            "response": str,
-        }
-    
-        builder = StateGraph(state_schema)
+        class AgentState(TypedDict):
+            question: str
+            is_reversed: bool
+            is_python: bool
+            response: str
+        
+        builder = StateGraph(AgentState)
     
         # Nodes
         builder.add_node("check_reversed", check_reversed)