Update app.py

app.py

@@ -7,6 +7,7 @@ import pandas as pd
 from langgraph.graph import StateGraph, END
 from typing import TypedDict
 import re
+import string
 
 
 # (Keep Constants as is)
@@ -30,15 +31,34 @@ class SuperSmartAgent:
 
     def _build_graph(self):
         ############## NODES ###################
+        def score_text(text):
+            # Count alphanumeric characters
+            alnum_count = sum(c.isalnum() for c in text)
+            # Count spaces
+            space_count = text.count(' ')
+            # Count punctuation marks
+            punctuation_count = sum(c in string.punctuation for c in text)
+            # Check if text ends with sentence punctuation (., !, ?)
+            ends_properly = text[-1] in '.!?'
+            
+            # Simple heuristic score: alnum + spaces + bonus if ends properly
+            score = alnum_count + space_count
+            if ends_properly:
+                score += 5  # give bonus if text ends with punctuation
+            
+            return score
+
         def check_reversed(state):
             question = state["question"]
             reversed_candidate = question[::-1]
-            
-            # Your original reversed_ratio logic is unreliable; improve it:
-            original_alpha_count = sum(c.isalpha() for c in question)
-            reversed_alpha_count = sum(c.isalpha() for c in reversed_candidate)
-            
-            if reversed_alpha_count > original_alpha_count:
+        
+            original_score = score_text(question)
+            reversed_score = score_text(reversed_candidate)
+        
+            print(f"Original text score: {original_score}")
+            print(f"Reversed text score: {reversed_score}")
+        
+            if reversed_score > original_score:
                 print("Detected reversed text. Marking as reversed.")
                 state["is_reversed"] = True
             else:
@@ -47,9 +67,10 @@ class SuperSmartAgent:
             return state
         
         def fix_question(state):
-            if state["is_reversed"]:
+            if state.get("is_reversed", False):
                 print(f"Fixing question by reversing: {state['question']}")
                 state["question"] = state["question"][::-1]
+                print(f"Fixed question: {state['question']}")
             return state
 
         def check_riddle_or_trick(state):
@@ -101,10 +122,11 @@ class SuperSmartAgent:
             state["response"] = "This question doesn't require Python or is unclear."
             return state
         
-        class AgentState(TypedDict):
+        class AgentState(TypedDict, total=False):
             question: str
             is_reversed: bool
             is_python: bool
+            is_riddle:bool
             response: str
         
         builder = StateGraph(AgentState)