Update app.py

app.py

@@ -4,7 +4,8 @@ import uuid
 from huggingface_hub import InferenceClient
 
 # Initialize the HuggingFace inference client
-client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=st.secrets["HUGGINGFACE_API_KEY"])
+def initialize_client(api_key):
+    return InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=api_key)
 
 INIT_PROMPT = """
 \n\nHuman: You are DarijaBot, a helpful assistant that can converse in both Latin and Arabic alphabet Darija. You will help the users learn about Wardley Mapping.
@@ -41,7 +42,7 @@ st.sidebar.markdown("Using HuggingFaceH4/zephyr-7b-beta API")
 st.sidebar.markdown(st.session_state.session_id)
 st.sidebar.divider()
 
-# Check if the user has provided an API key, otherwise default to the secret
+# Prompt the user for the API key
 user_huggingface_api_key = st.sidebar.text_input("Enter your HuggingFace API Key:", placeholder="hf_...", type="password")
 
 if "messages" not in st.session_state:
@@ -52,7 +53,7 @@ if "all_prompts" not in st.session_state:
     st.session_state["all_prompts"] = INIT_PROMPT + TRAINING_PROMPT
 
 if user_huggingface_api_key:
-    client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=user_huggingface_api_key)
+    client = initialize_client(user_huggingface_api_key)
 else:
     st.warning("Please enter your HuggingFace API key", icon="⚠️")