Update private_gpt/server/utils/authentication.py

private_gpt/server/utils/authentication.py

@@ -6,13 +6,13 @@ from passlib.context import CryptContext
 from fastapi.security import OAuth2PasswordRequestForm, OAuth2PasswordBearer
 from jose import jwt, JWTError
 from pydantic import BaseModel
-
+import os 
 router = APIRouter(
     prefix='/v1/auth',
     tags=['auth']
 )
 
-SECRET_KEY = PASSWORD
+SECRET_KEY = os.environ.get("PASSWORD")
 ALGORITHM = 'HS256'
 
 bcrypt_context = CryptContext(schemes=['bcrypt'], deprecated='auto')
@@ -21,13 +21,13 @@ oauth2_bearer = OAuth2PasswordBearer(tokenUrl='/v1/auth/token')
 hardcoded_users = [
     {   "id": 1, 
         "username": "test",
-        "password_hash": bcrypt_context.hash(USER_HASH),
+        "password_hash": bcrypt_context.hash(os.environ.get("USER_HASH")),
         "role": "user"
     },
 
     {   "id": 2, 
         "username": "admin",
-        "password_hash": bcrypt_context.hash(ADMIN_HASH),
+        "password_hash": bcrypt_context.hash(os.environ.get("ADMIN_HASH")),
         "role": "admin"
     },