Update app.py

app.py

@@ -192,13 +192,27 @@ COLUMN_SYNONYMS = {
 
 
 # Helper function to map user query terms to dataset columns
+#def map_query_to_column(query):
+#    for col, synonyms in COLUMN_SYNONYMS.items():
+#        for term in synonyms:
+#            if term in query:
+#                return col
+#    return None
+
 def map_query_to_column(query):
-    for col, synonyms in COLUMN_SYNONYMS.items():
-        for term in synonyms:
-            if term in query:
+    query = query.lower()
+    all_synonyms = {synonym: col for col, synonyms in COLUMN_SYNONYMS.items() for synonym in synonyms}
+    matches = get_close_matches(query, all_synonyms.keys(), n=1, cutoff=0.6)
+    
+    if matches:
+        return all_synonyms[matches[0]]
+    else:
+        for col, synonyms in COLUMN_SYNONYMS.items():
+            if any(term in query for term in synonyms):
                 return col
     return None
 
+
 # Visualization generator with synonym handling
 def generate_visual_from_query(query, df):
     try: