update dockerfile

Dockerfile

@@ -13,6 +13,7 @@ COPY *.whl /packages/
 # Copy users db file along with execution script
 COPY start.sh /
 COPY load_data.py /
+COPY users.py /packages
 
 # Install packages
 RUN apt update

start.sh

@@ -9,22 +9,10 @@ echo "ubuntu" | sudo -S su user
 admin_password=$(htpasswd -nbB "" "$ADMIN_PASSWORD" | cut -d ":" -f 2 | tr -d "\n")
 argilla_password=$(htpasswd -nbB "" "$ARGILLA_PASSWORD" | cut -d ":" -f 2 | tr -d "\n")
 
-# Create users.yml file
-sudo cat >/packages/users.yml <<EOF
-- username: "admin"
-  api_key: $ADMIN_API_KEY
-  full_name: Hugging Face
-  email: [email protected]
-  hashed_password: $admin_password
-  workspaces: []
-
-- username: "argilla"
-  api_key: $ARGILLA_API_KEY
-  full_name: Hugging Face
-  email: [email protected]
-  hashed_password: $argilla_password
-  workspaces: ["admin"]
-EOF
+sed -i "s/ADMIN_PASSWORD/$ADMIN_PASSWORD/g" /packages/users.yml
+sed -i "s/ARGILLA_PASSWORD/$ARGILLA_PASSWORD/g" /packages/users.yml
+sed -i "s/ADMIN_API_KEY/$ADMIN_API_KEY/g" /packages/users.yml
+sed -i "s/ARGILLA_API_KEY/$ARGILLA_API_KEY/g" /packages/users.yml
 
 # Disable security in elasticsearch configuration
 sudo sed -i "s/xpack.security.enabled: true/xpack.security.enabled: false/g" /etc/elasticsearch/elasticsearch.yml

users.yaml

@@ -0,0 +1,13 @@
+- username: "admin"
+  api_key: ADMIN_API_KEY
+  full_name: Hugging Face
+  email: [email protected]
+  hashed_password: ADMIN_PASSWORD
+  workspaces: [ ]
+
+- username: "argilla"
+  api_key: ARGILLA_API_KEY
+  full_name: Hugging Face
+  email: [email protected]
+  hashed_password: ARGILLA_PASSWORD
+  workspaces: [ "admin" ]