Update app.py

app.py

@@ -45,19 +45,21 @@ FORBIDDEN_KEYWORDS = [
 # the sidebar lets you override every one of these and switch databases
 # without restarting the app)
 # -----------------------------
-
 load_dotenv()
 
-api_key = os.getenv("GOOGLE_API_KEY")
+api_key = os.getenv("GOOGLE_API_KEY", "")
 
 default_server = os.getenv("SERVER", "")
 default_database = os.getenv("DATABASE", "")
 default_driver = os.getenv("DRIVER", "ODBC Driver 17 for SQL Server")
 
-if not api_key:
-    st.error("Missing required environment variable: GOOGLE_API_KEY")
-    st.stop()
+client = None
 
+try:
+    if api_key:
+        client = genai.Client(api_key=api_key)
+except Exception as e:
+    st.error(f"Gemini Initialization Error: {e}")
 # -----------------------------
 # Gemini Client
 # -----------------------------
@@ -387,11 +389,14 @@ Question
     try:
 
         with st.spinner("Generating SQL..."):
-            response = client.models.generate_content(
-                model=GEMINI_MODEL,
-                contents=prompt
-            )
+           if client is None:
+    st.error("Gemini API key is missing or invalid.")
+    st.stop()
 
+response = client.models.generate_content(
+    model=GEMINI_MODEL,
+    contents=prompt
+)
         sql = response.text.strip()
 
         sql = sql.replace("```sql", "")