Duplicate from argilla/template-space-docker

Co-authored-by: Daniel Vila <dvilasuero@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.history

Dockerfile

@@ -0,0 +1,38 @@
+FROM docker.elastic.co/elasticsearch/elasticsearch-oss:7.10.2
+
+# Install Python
+RUN cd /etc/yum.repos.d/
+RUN sed -i 's/mirrorlist/#mirrorlist/g' /etc/yum.repos.d/CentOS-*
+RUN sed -i 's|#baseurl=http://mirror.centos.org|baseurl=http://vault.centos.org|g' /etc/yum.repos.d/CentOS-*
+RUN yum install -y python39 python39-pip wget
+
+# Copy setup files
+COPY ./requirements.txt requirements.txt
+COPY ./start.sh start.sh
+RUN chown elasticsearch:elasticsearch requirements.txt
+RUN chown elasticsearch:elasticsearch start.sh
+
+# Change the user to the default user for the elasticsearch image
+USER elasticsearch
+
+# API KEY and password
+RUN --mount=type=secret,id=API_KEY,mode=0444,required=false
+ARG API_KEY=$(cat /run/secrets/API_KEY)
+
+RUN --mount=type=secret,id=PASSWORD,mode=0444,required=false
+ARG PASSWORD=$(cat /run/secrets/PASSWORD)
+
+# Install Python deps
+RUN pip3 install -U -r requirements.txt
+
+RUN wget https://raw.githubusercontent.com/vishnubob/wait-for-it/master/wait-for-it.sh
+
+# Make scripts executable
+RUN chmod +x wait-for-it.sh
+RUN chmod +x start.sh
+
+# Expose the application's port
+EXPOSE 6900
+
+# Set the command to run the shell script
+CMD ["./start.sh"]

README.md

@@ -0,0 +1,18 @@
+---
+title: Template Argilla Space Docker
+emoji: 📊
+colorFrom: gray
+colorTo: yellow
+sdk: docker
+pinned: false
+app_port: 6900
+fullWidth: true
+duplicated_from: argilla/template-space-docker
+---
+
+Use this Space to launch your Argilla app. This space can be configured with two secret env variables (you can find how to this under Settings in your Space page):
+
+- `API_KEY`: You can set up your own API key for uploading and reading data with Argilla Python client. This API_KEY can be any string you want.
+- `PASSWORD`: You can set up the password for login into the UI, in case you want to disable access to other users. The default username is `argilla`. If you don't set up this secret the password will be `1234`
+
+

requirements.txt

@@ -0,0 +1,2 @@
+argilla[server]
+fastapi

start.sh

@@ -0,0 +1,16 @@
+
+ES_JAVA_OPTS=-'Xms512m -Xmx512m' elasticsearch -E node.name=elasticsearch -E cluster.name=es-local -E discovery.type=single-node -E cluster.routing.allocation.disk.threshold_enabled=false &
+./wait-for-it.sh -t 30 localhost:9200
+if [ -z "$API_KEY" ] && [ -z "$PASSWORD" ]; then
+  python3 -m argilla
+elif [ -z "$API_KEY" ] && [ -n "$PASSWORD" ]; then
+  export ARGILLA_LOCAL_AUTH_DEFAULT_PASSWORD=$PASSWORD
+  python3 -m argilla
+elif [ -n "$API_KEY" ] && [ -z "$PASSWORD" ]; then
+  export ARGILLA_LOCAL_AUTH_DEFAULT_APIKEY=$API_KEY
+  python3 -m argilla
+else
+  export ARGILLA_LOCAL_AUTH_DEFAULT_APIKEY=$API_KEY
+  export ARGILLA_LOCAL_AUTH_DEFAULT_PASSWORD=$PASSWORD
+  python3 -m argilla
+fi