Update app.py

app.py

@@ -352,14 +352,26 @@ with st.sidebar:
     api_key = st.text_input("Groq API Key", value=hf_api_key, type="password", placeholder="gsk_...", help="Get free key at console.groq.com")
 
     if api_key:
-        try:
-            groq_client = Groq(api_key=api_key)
+    try:
+        groq_client = Groq(api_key=api_key)
+        # Test the connection with a simple call
+        test = groq_client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[{"role": "user", "content": "hi"}],
+            max_tokens=5
+        )
+        st.session_state.api_configured = True
+        st.session_state.groq_client    = groq_client
+        st.success("✅ API Connected")
+    except Exception as e:
+        if "invalid_api_key" in str(e).lower() or "401" in str(e):
+            st.error("❌ Invalid API key")
+        else:
+            # Key is valid but other error — still connect
             st.session_state.api_configured = True
-            st.session_state.groq_client = groq_client
+            st.session_state.groq_client    = Groq(api_key=api_key)
             st.success("✅ API Connected")
-        except:
-            st.error("❌ Invalid API key")
-            st.session_state.api_configured = False
+        st.session_state.api_configured = False
 
     # Mode selection
     st.markdown('<div class="section-head">Knowledge Source</div>', unsafe_allow_html=True)