Update app.py

app.py

@@ -2,6 +2,14 @@ import streamlit as st
 from google import genai
 from io import StringIO
 import prompts # Assuming prompts.py is in the same directory
+import os  # Add this at the top of your imports
+
+# Replace the st.secrets line with this:
+api_key = os.environ.get("GOOGLE_API_KEY")
+
+if not api_key:
+    st.error("API Key not found. Please ensure GOOGLE_API_KEY is set in Hugging Face Secrets.")
+    st.stop()
 
 st.set_page_config(page_title="SQL AI Assistant", layout="wide")