Update conversation_logic.py

conversation_logic.py

@@ -117,27 +117,42 @@ def _normalize_classified_topic(
     q = (question_text or "").lower()
     c = (category or "").strip()
 
-    if t in {"general_quant", "general", "unknown", ""}:
+    if t not in {"general_quant", "general", "unknown", ""}:
+        return topic
+
+    if "%" in q or "percent" in q:
+        return "percent"
+
+    if "ratio" in q or ":" in q:
+        return "ratio"
+
+    if "probability" in q or "chosen at random" in q:
+        return "probability"
+
+    if "divisible" in q or "remainder" in q or "prime" in q or "factor" in q:
+        return "number_theory"
+
+    if "circle" in q or "triangle" in q or "perimeter" in q or "area" in q or "circumference" in q:
+        return "geometry"
+
+    if "mean" in q or "median" in q or "average" in q or "sales" in q or "revenue" in q:
         if c == "Quantitative":
-            if "%" in q or "percent" in q:
-                return "percent"
-            if "ratio" in q:
-                return "ratio"
-            if "probability" in q or "chosen at random" in q:
-                return "probability"
-            if "divisible" in q or "remainder" in q or "prime" in q:
-                return "number_theory"
-            if "circle" in q or "triangle" in q or "perimeter" in q or "area" in q:
-                return "geometry"
-            if "=" in q or "what is x" in q or "what is y" in q:
-                return "algebra"
-            return "quant"
-
-        if c == "DataInsight":
-            return "data"
-
-        if c == "Verbal":
-            return "verbal"
+            return "statistics"
+        return "data"
+
+    if "=" in q or "what is x" in q or "what is y" in q or "integer" in q:
+        return "algebra"
+
+    if c == "DataInsight":
+        return "data"
+
+    if c == "Verbal":
+        return "verbal"
+
+    if c == "Quantitative":
+        return "quant"
+
+    return "general"
 
     return topic
 
@@ -450,12 +465,48 @@ class ConversationEngine:
             question_text=solver_input,
             category=category,
         )
-        inferred_category = classification.get("category") or category
+                inferred_category = classification.get("category") or category
+
+        if not inferred_category:
+            q = solver_input.lower()
+            if any(
+                k in q
+                for k in [
+                    "percent",
+                    "%",
+                    "ratio",
+                    "divisible",
+                    "remainder",
+                    "probability",
+                    "circle",
+                    "triangle",
+                    "=",
+                ]
+            ):
+                inferred_category = "Quantitative"
+            elif any(
+                k in q
+                for k in [
+                    "sales",
+                    "revenue",
+                    "median",
+                    "mean",
+                    "chart",
+                    "table",
+                    "scatter",
+                    "distribution",
+                ]
+            ):
+                inferred_category = "DataInsight"
+            else:
+                inferred_category = "General"
+
         question_topic = _normalize_classified_topic(
             classification.get("topic"),
             inferred_category,
             solver_input,
         )
+        )
         question_type = classification.get("type")
 
         resolved_intent = intent or detect_intent(user_text, help_mode)