Update conversation_logic.py

conversation_logic.py

@@ -74,12 +74,24 @@ def _normalize_classified_topic(topic: Optional[str], category: Optional[str], q
     q = (question_text or "").lower()
     c = normalize_category(category)
 
+    has_ratio_form = bool(re.search(r"\b\d+\s*:\s*\d+\b", q))
+    has_algebra_form = (
+        "=" in q
+        or bool(re.search(r"\b[xyz]\b", q))
+        or bool(re.search(r"\d+[a-z]\b", q))
+        or bool(re.search(r"\b[a-z]\s*[\+\-\*/=]", q))
+    )
+
+    # override suspicious ratio classifications caused by prompt punctuation like "Solve:"
+    if t == "ratio" and not has_ratio_form and has_algebra_form:
+        t = "algebra"
+
     if t not in {"general_quant", "general", "unknown", ""}:
         return t
 
     if "%" in q or "percent" in q:
         return "percent"
-    if "ratio" in q or re.search(r"\b\d+\s*:\s*\d+\b", q):
+    if "ratio" in q or has_ratio_form:
         return "ratio"
     if "probability" in q or "chosen at random" in q:
         return "probability"
@@ -89,7 +101,7 @@ def _normalize_classified_topic(topic: Optional[str], category: Optional[str], q
         return "geometry"
     if any(k in q for k in ["mean", "median", "average", "sales", "revenue"]):
         return "statistics" if c == "Quantitative" else "data"
-    if "=" in q or "what is x" in q or "what is y" in q or "integer" in q:
+    if has_algebra_form or "what is x" in q or "what is y" in q or "integer" in q:
         return "algebra"
 
     if c == "DataInsight":