Update app.py

app.py

@@ -62,9 +62,7 @@ except Exception as e:
     st.error(f"Gemini Initialization Error: {e}")
 # -----------------------------
 # Gemini Client
-# -----------------------------
-
-client = genai.Client(api_key=api_key)
+# ----------------------------
 
 # -----------------------------
 # Connection helpers
@@ -386,29 +384,30 @@ Question
     # Generate SQL
     # -----------------------------
 
-    try:
+  try:
 
-        with st.spinner("Generating SQL..."):
-           if client is None:
-    st.error("Gemini API key is missing or invalid.")
-    st.stop()
+    with st.spinner("Generating SQL..."):
 
-response = client.models.generate_content(
-    model=GEMINI_MODEL,
-    contents=prompt
-)
-        sql = response.text.strip()
+        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", "")
         sql = sql.replace("```", "")
         sql = sql.strip()
 
-    except Exception as e:
+except Exception as e:
 
-        logger.exception("Gemini generation failed")
-        st.error("Gemini Error")
-        st.exception(e)
-        st.stop()
+    logger.exception("Gemini generation failed")
+    st.error("Gemini Error")
+    st.exception(e)
+    st.stop()
 
     # -----------------------------
     # Safety Check