Update app.py

app.py

@@ -3,7 +3,6 @@ import streamlit as st
 import pandas as pd
 import sqlite3
 from langchain import OpenAI, LLMChain, PromptTemplate
-# Removed unused import: from langchain_community.utilities import SQLDatabase
 import sqlparse
 import logging
 from sql_metadata import Parser
@@ -40,8 +39,12 @@ def validate_sql(query, valid_columns):
     """Validates the SQL query by ensuring it references only valid columns."""
     parser = Parser(query)
     columns_in_query = parser.columns
+
+    # Convert valid columns to lowercase for case-insensitive comparison
+    valid_columns_lower = [col.lower() for col in valid_columns]
+
     for column in columns_in_query:
-        if column not in valid_columns:
+        if column.lower() not in valid_columns_lower:
             st.write(f"Invalid column detected: {column}")
             return False
     return True
@@ -86,9 +89,6 @@ def process_input():
                     'columns': columns
                 })
 
-                # Debug: Display generated SQL query for inspection
-                # st.write(f"Generated SQL Query:\n{generated_sql}")
-
                 # Validate SQL query
                 if not validate_sql_with_sqlparse(generated_sql):
                     assistant_response = "Generated SQL is not valid."