Update app.py

app.py

@@ -100,7 +100,7 @@ def generate_sql(question: str, schema: str) -> str:
     if re.search(r'show.*(all|every).*row|all.*row|select all', q):
         return f'SELECT * FROM {quoted} LIMIT 50'
     if re.search(r'average|avg', q) and col_match:
-        num_col = next((c for c in col_match if re.search(r'num|price|val|amt|count|qty|sal|rev|cost|pm|aqi|no|co|so|o3', c, re.I)), col_match[1] if len(col_match) > 1 else col_match[0])
+        num_col = next((c for c in col_match if re.search(r'pm|aqi|no|co|so|o3|benzene|toluene|xylene', c, re.I)), col_match[2])
         return f'SELECT AVG("{num_col}") FROM {quoted}'
     if re.search(r'unique|distinct', q) and col_match:
         return f'SELECT COUNT(DISTINCT "{col_match[0]}") FROM {quoted}'