Update app.py

app.py

@@ -44,6 +44,9 @@ def validate_sql(query, valid_columns):
     valid_columns_lower = [col.lower() for col in valid_columns]
 
     for column in columns_in_query:
+        # Skip functions like LOWER()
+        if '(' in column or ')' in column:
+            continue
         if column.lower() not in valid_columns_lower:
             st.write(f"Invalid column detected: {column}")
             return False
@@ -58,6 +61,8 @@ def validate_sql_with_sqlparse(query):
 template = """
 You are an expert data scientist. Given a natural language question, the name of the table, and a list of valid columns, generate a valid SQL query that answers the question.
 
+Ensure that string comparisons in the WHERE clause are case-insensitive by using 'COLLATE NOCASE' or the LOWER() function.
+
 Question: {question}
 
 Table name: {table_name}