Update app.py

app.py

@@ -4,12 +4,17 @@ from google import genai
 from io import StringIO
 import prompts # Assuming prompts.py is in the same directory
 
-# 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()
+    # Fallback for local testing (if you have a local secrets.toml)
+    try:
+        api_key = st.secrets["GOOGLE_API_KEY"]
+    except:
+        st.error("Missing GOOGLE_API_KEY. Please add it to your Space Secrets.")
+        st.stop()
+
+client = genai.Client(api_key=api_key)
 
 st.set_page_config(page_title="SQL AI Assistant", layout="wide")