Update app.py

app.py

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