diff --git a/app.py b/app.py index 518544d464fa395ba7a6db85ccbc65bb5943d5c5..728e70417190138470251f05c67ad9ef8342a6b0 100644 --- a/app.py +++ b/app.py @@ -9,6 +9,22 @@ from langchain.prompts import ( SystemMessagePromptTemplate, HumanMessagePromptTemplate, ) +import os + +def manage_sensitive(name): + secret_fpath = f'/run/secrets/{name}' + existence = os.path.exists(secret_fpath) + + if existence: + v2 = open(secret_fpath).read().rstrip('\n') + return v2 + + if not existence: + return KeyError(f'{name}') + +OPENAI_API_KEY = manage_sensitive("openai_api_key") +os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY + st.set_page_config( page_title="MPOG Helper ", page_icon="🤖", diff --git a/docker-compose.yml b/docker-compose.yml index ae09531d1660b959c4522410151634c25c1b8f85..d98a77070b519ee06fc2ac60df3bad40ff2521cb 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -10,19 +10,7 @@ services: environment: - PYTHONUNBUFFERED=1 secrets: - - db_server - - db_name - - db_user - - db_password - openai_api_key secrets: - db_server: - file: secrets/db_server.txt - db_name: - file: secrets/db_name.txt - db_user: - file: secrets/db_user.txt - db_password: - file: secrets/db_password.txt openai_api_key: file: secrets/openai_api_key.txt