Update app.py

app.py

@@ -38,6 +38,10 @@ try:
 except Exception:
     USE_GEMINI = False
 
+def get_gemini_api_key() -> str:
+    # Return Gemini API key from SECRET_KEY or GOOGLE_API_KEY (if present)
+    return os.getenv('SECRET_KEY') or os.getenv('GOOGLE_API_KEY') or ''
+
 # ===== Transformers (open-source backend) =====
 import torch
 from transformers import AutoProcessor, AutoModelForVision2Seq
@@ -51,14 +55,20 @@ st.markdown(
 
 with st.sidebar:
     st.header("⚙️ Settings")
-    backend = st.selectbox(
-        "Backend",
-        [
-            "Local Qwen2-VL (Open-Source)",
-            "Gemini API" if USE_GEMINI else "Gemini API (unavailable)",
-        ],
-        index=0,
-    )
+
+    # Gemini availability message
+    gem_key = get_gemini_api_key() if USE_GEMINI else ''
+    gem_ready = bool(gem_key)
+
+    backend_opts = ["Local Qwen2-VL (Open-Source)"]
+    if USE_GEMINI and gem_ready:
+        backend_opts.append("Gemini API")
+    elif USE_GEMINI and not gem_ready:
+        backend_opts.append("Gemini API (key missing)")
+    else:
+        backend_opts.append("Gemini API (unavailable)")
+
+    backend = st.selectbox("Backend", backend_opts, index=0)
 
     mode = st.selectbox("Detail level", ["soft", "artistic", "raw"], index=2)
 
@@ -71,11 +81,16 @@ with st.sidebar:
     max_tokens = st.slider("Max new tokens", 64, 512, 220, 8)
     temperature = st.slider("Temperature", 0.0, 1.2, 0.6, 0.05)
 
+    # Gemini status badge
+    if USE_GEMINI:
+        if gem_ready:
+            st.success("Gemini key detected (SECRET_KEY / GOOGLE_API_KEY)")
+        else:
+            st.warning("Gemini key not found. Add SECRET_KEY or GOOGLE_API_KEY in Space Secrets.")
+
     st.divider()
     st.subheader("🔐 Policy")
-    st.caption(
-        "This app only describes consenting adults. It refuses illegal/underage/forced content."
-    )
+    st.caption("This app only describes consenting adults. It refuses illegal/underage/forced content.")
 
 # ---------------- Helpers ----------------
 @st.cache_resource(show_spinner=False)
@@ -190,10 +205,9 @@ def extract_tags(text: str):
 def get_gemini_response(image: Image.Image):
     if not USE_GEMINI:
         raise RuntimeError("Gemini library not available in this Space.")
-    secret_key = os.getenv("SECRET_KEY", "")
+    secret_key = get_gemini_api_key()
     if not secret_key:
-        st.warning("Gemini SECRET_KEY not found; please set it in your Space secrets.")
-        return ""
+        raise RuntimeError("Gemini API key not found. Set SECRET_KEY or GOOGLE_API_KEY in Space secrets.")
     genai.configure(api_key=secret_key)
     model = genai.GenerativeModel("gemini-2.0-flash")
     sys = (