Update app.py

app.py

@@ -4,13 +4,20 @@ import os
 import json
 import time
 
-# Initialize the Groq client with the API key from Hugging Face secrets
-api_key = os.environ.get("GROQ_API_KEY")
-if not api_key:
-    st.error("No Groq API key found. Please add your GROQ_API_KEY to the Hugging Face Space secrets.")
-    st.stop()
+def get_groq_client():
+    api_key = os.environ.get("GROQ_API_KEY")
+    if not api_key:
+        api_key = st.session_state.get('groq_api_key')
+    
+    if api_key:
+        return groq.Groq(api_key=api_key)
+    return None
+
+def set_api_key():
+    st.session_state['groq_api_key'] = st.session_state.api_key_input
+    st.experimental_rerun()
 
-client = groq.Groq(api_key=api_key)
+client = get_groq_client()
 
 def make_api_call(messages, max_tokens, is_final_answer=False):
     for attempt in range(3):
@@ -44,10 +51,13 @@ def main():
     Open source [repository here](https://github.com/bklieger-groq)
     """)
     
-    if not api_key:
-        st.warning("Please add your GROQ_API_KEY to the Hugging Face Space secrets to use this application.")
+    if not client:
+        st.warning("No Groq API key found. Please enter your API key below or add it to the Hugging Face Space secrets.")
+        api_key_input = st.text_input("Enter your Groq API key:", type="password", key="api_key_input")
+        st.button("Submit API Key", on_click=set_api_key)
+        
         st.markdown("""
-        To add your API key:
+        To add your API key to the Hugging Face Space secrets:
         1. Go to the Settings tab of this Space
         2. Scroll down to the "Repository secrets" section
         3. Click on "New secret"