Update src/streamlit_app.py

src/streamlit_app.py

@@ -307,45 +307,58 @@ if genai is None or ChatGoogleGenerativeAI is None:
 if 'enable_llm' not in st.session_state:
     st.session_state.enable_llm = False
 
-if st.button("Test Google API Connection"):
-    if not st.session_state.enable_llm or llm is None:
-        st.error("LLM is not enabled or initialized. Check the checkbox and API key.")
-    else:
-        try:
-            with st.spinner("Pinging Google API..."):
-                # Use the simplest possible LLM call
-                llm.invoke("Hello, are you working?")
-                st.success("✅ SUCCESS: API call worked!")
-        except Exception as e:
-            st.error(f"❌ API call FAILED. This is the error: {e}")
 
-enable_llm_checkbox = st.checkbox("Enable LLM features (requires GOOGLE_API_KEY)", value=st.session_state.enable_llm)
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+api_key_exists = bool(GOOGLE_API_KEY)
+
+if not api_key_exists:
+    st.warning("⚠️ GOOGLE_API_KEY not found in environment variables.")
+    st.info("Add GOOGLE_API_KEY to your Hugging Face Space secrets to enable AI features.")
+
+# LLM Enable Checkbox
+enable_llm_checkbox = st.checkbox(
+    "Enable LLM features (requires GOOGLE_API_KEY)", 
+    value=st.session_state.enable_llm,
+    disabled=not api_key_exists,
+    help="AI-powered question generation and evaluation"
+)
 st.session_state.enable_llm = enable_llm_checkbox
 
-if st.session_state.enable_llm:
+# Initialize LLM if enabled
+if st.session_state.enable_llm and api_key_exists:
     try:
-        GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
-        if not GOOGLE_API_KEY:
-            st.error("THIS IS THE REAL ERROR: GOOGLE_API_KEY not set. Check your Hugging Face secrets.")
-            # st.session_state.enable_llm = False
-        else:
-            # Try to configure genai and perform a lightweight sanity check.
-            try:
-                genai.configure(api_key=GOOGLE_API_KEY)
-                # Initialize LLM and model wrappers
-                llm = get_llm(GOOGLE_API_KEY)
-                gen_q_model, intro_model, eval_model = get_models(llm)
-                # optional quick test: attempt to build a simple prompt but don't invoke network here
-            except Exception as e:
-                st.error(f"THIS IS THE REAL ERROR: {e}")
-                # st.session_state.enable_llm = False
-                llm = None
-                gen_q_model = None
-                intro_model = None
-                eval_model = None
+        genai.configure(api_key=GOOGLE_API_KEY)
+        llm = get_llm(GOOGLE_API_KEY)
+        gen_q_model, intro_model, eval_model = get_models(llm)
+        st.success("✅ LLM features enabled successfully")
     except Exception as e:
-        st.warning(f"Unexpected error while enabling LLM: {e}")
+        st.error(f"❌ Could not initialize LLM: {e}")
+        st.info("Check your API key and try again.")
         st.session_state.enable_llm = False
+        llm = None
+        gen_q_model = None
+        intro_model = None
+        eval_model = None
+
+# Test API Button (AFTER initialization)
+if st.button("Test Google API Connection"):
+    if not st.session_state.enable_llm:
+        st.error("❌ LLM features are not enabled. Check the checkbox above first.")
+    elif llm is None:
+        st.error("❌ LLM is not initialized. Check API key configuration.")
+    else:
+        try:
+            with st.spinner("Testing API connection..."):
+                # Simple test call
+                test_response = llm.invoke("Say 'Hello' if you can hear me.")
+                st.success("✅ SUCCESS! API is working correctly.")
+                st.info(f"Response: {test_response.content if hasattr(test_response, 'content') else str(test_response)}")
+        except Exception as e:
+            st.error(f"❌ API call FAILED with error: {e}")
+            st.info("This usually means: invalid API key, quota exceeded, or network issues.")
+
+st.divider()
+
 
 # --- Session State Initialization ---
 # This is crucial for making the app work step-by-step