Update app.py

app.py

@@ -352,26 +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)
-        # 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
+        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(api_key=api_key)
+            st.session_state.groq_client    = groq_client
             st.success("✅ API Connected")
-        st.session_state.api_configured = False
+        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(api_key=api_key)
+                st.success("✅ API Connected")
+            st.session_state.api_configured = False
 
     # Mode selection
     st.markdown('<div class="section-head">Knowledge Source</div>', unsafe_allow_html=True)