Update app.py

app.py

@@ -19,8 +19,12 @@ default_db_config = {
     'port': int(os.getenv('DB_PORT', 3306))
 }
 
-# Groq API configuration
-groq_client = Groq(api_key=os.getenv('GROQ_API_KEY'))
+# Groq API configuration with error handling
+try:
+    groq_client = Groq(api_key=os.getenv('GROQ_API_KEY'))
+except Exception as e:
+    groq_client = None
+    print(f"Failed to initialize Groq client: {str(e)}")
 
 # Temporary storage for current database name and schema
 current_db_name = None
@@ -215,6 +219,9 @@ def load_sql_file(file):
 
 def generate_sql_query(question, schema):
     """Generate SQL query using Groq API with user-friendly aliases."""
+    if not groq_client:
+        return "ERROR: Groq client not initialized. Check API key and try again."
+
     schema_text = "\n".join([f"Table: {table}\nColumns: {', '.join(columns)}" for table, columns in schema.items()])
     prompt = f"""
 You are a SQL expert. Based on the following database schema, generate a valid MySQL query for the user's question. Only use tables and columns that exist in the schema. Use user-friendly aliases for column names (e.g., 'cust_id' becomes 'Customer ID', 'admission_date' becomes 'Admission Date'). Return ONLY the SQL query, without explanations, markdown, or code block formatting (e.g., no ```). If the question references non-existent tables or columns, return an error message starting with 'ERROR:'. Do not use GROUP BY or aggregation functions (e.g., SUM, COUNT, AVG) unless the question explicitly requests aggregation (e.g., 'sum of all bills', 'average cost', 'count of patients'). Treat 'total bill amount' as the individual bill amount (e.g., bill.amount) unless aggregation is clearly specified. For names, concatenate first_name and last_name if applicable (e.g., CONCAT(first_name, ' ', last_name) AS 'Full Name'). Use direct JOINs with correct foreign key relationships. Avoid subqueries unless absolutely necessary. Place filtering conditions (e.g., department name, status) in the WHERE clause, not JOIN clauses. Handle case sensitivity in string comparisons by using LOWER() for status fields (e.g., LOWER(status) = 'unpaid'). Verify table relationships before joining.
@@ -262,6 +269,9 @@ def index():
     results = None
     generated_query = None
 
+    if not groq_client:
+        error = "Groq client not initialized. Please check GROQ_API_KEY and restart the app."
+
     if request.method == 'POST':
         if 'sql_file' in request.files:
             file = request.files['sql_file']