Update src/helper_functions.py

src/helper_functions.py

@@ -11,11 +11,10 @@ from datetime import datetime
 import regex
 import os
 
-app_password = os.getenv("APP_PASSWORD")
 
 def check_password():
     def password_entered():
-        if st.session_state["password"] == app_password:
+        if st.session_state["password"] == os.getenv("APP_PASSWORD"):
             st.session_state["password_correct"] = True
             del st.session_state["password"]
         else:
@@ -120,7 +119,7 @@ def watsonx_chat_prompt(
     client=None,
     wx_url=None,
     wx_apikey=None,
-    project_id=None,
+    wx_project_id=None,
     model_id=None,
     params=None,
 ):
@@ -158,7 +157,7 @@ def watsonx_chat_prompt(
     # Use provided client or create new one
     if client is None:
         wx_credentials = Credentials(url=wx_url, api_key=wx_apikey)
-        client = APIClient(credentials=wx_credentials, project_id=project_id)
+        client = APIClient(credentials=wx_credentials, project_id=wx_project_id)
 
     chat_model = ModelInference(api_client=client, model_id=model_id, params=params)