Update app.py

app.py

@@ -180,16 +180,17 @@ def escape_markdown(text):
 
 # Synonym mapping for flexible query understanding
 COLUMN_SYNONYMS = {
-    "job_title": ["job title", "job role", "role", "designation", "position", "job responsibility"],
-    "experience_level": ["experience level", "seniority", "experience", "career stage"],
-    "employment_type": ["employment type", "job type", "contract type"],
-    "salary_in_usd": ["salary", "income", "earnings", "pay", "wage"],
-    "remote_ratio": ["remote work", "work from home", "remote ratio", "remote"],
-    "company_size": ["company size", "organization size", "business size"],
-    "employee_residence": ["country", "residence", "location", "employee location"],
-    "company_location": ["company location", "office location", "company country"],
+    "job_title": ["job title", "job role", "role", "designation", "position", "job responsibility", "occupation"],
+    "experience_level": ["experience level", "seniority", "experience", "career stage", "years of experience"],
+    "employment_type": ["employment type", "job type", "contract type", "employment status"],
+    "salary_in_usd": ["salary", "income", "earnings", "pay", "wage", "compensation"],
+    "remote_ratio": ["remote work", "work from home", "remote ratio", "remote", "telecommute"],
+    "company_size": ["company size", "organization size", "business size", "firm size"],
+    "employee_residence": ["country", "residence", "location", "employee location", "home country"],
+    "company_location": ["company location", "office location", "company country", "headquarters", "location", "located", "area"],
 }
 
+
 # Helper function to map user query terms to dataset columns
 def map_query_to_column(query):
     for col, synonyms in COLUMN_SYNONYMS.items():