Merge pull request #10 from jasonyang-ee/main

Add Action script to build and push docker image to registry

Author: neonwatty

.github/workflows/docker-build.yml

@@ -0,0 +1,57 @@
+name: 'DockerBuild'
+
+on:
+  push:
+    tags:
+      - 'v*'
+  
+jobs:
+  Build_And_Push:
+    runs-on: ubuntu-22.04
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Log in to Github Docker Image Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.TOKEN_GITHUB }}
+
+      # Uncomment to use Docker Hub
+      # - name: Login to Docker Hub
+      #   uses: docker/login-action@v3
+      #   with:
+      #     username: ${{ secrets.USERNAME_DOCKERHUB }}
+      #     password: ${{ secrets.TOKEN_DOCKERHUB }}
+
+      - name: Docker Meta
+        id: meta
+        uses: docker/metadata-action@v4
+        with:
+          images: |
+            ghcr.io/${{ github.actor }}/meme-search
+          # Uncomment to use Docker Hub
+            # ${{ secrets.DOCKERHUB_USERNAME }}/meme-search
+          tags: type=ref,event=tag
+          flavor: latest=true
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build and Upload for AMD64 and ARM64
+        uses: docker/build-push-action@v4
+        with:
+          context: .
+          platforms: linux/amd64,linux/arm64
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

Dockerfile

@@ -12,6 +12,7 @@ ENV PYTHONPATH=.
 
 COPY requirements.txt /home/requirements.txt
 COPY meme_search /home/meme_search
+COPY data/dbs /home/init
 COPY .streamlit /home/.streamlit
 
 RUN pip3 install -r /home/requirements.txt
@@ -20,4 +21,6 @@ EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "/home/meme_search/app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+COPY entrypoint.sh /home/entrypoint.sh
+RUN chmod +x /home/entrypoint.sh
+ENTRYPOINT ["/home/entrypoint.sh"]

docker-compose.yml

@@ -1,13 +1,12 @@
 services:
   meme_search:
-    build:
-      context: .
-    image: meme_search
+    image: ghcr.io/neonwatty/meme_search:latest
     container_name: meme_search
     ports:
       - 8501:8501
     volumes:
-      - ./data:/home/data
+      - ./meme_search/database:/home/data/dbs   # database persistence
+      - ./meme_search/memes:/home/data/input    # meme images
     ## uncomment to enable GPU support for the container
     # deploy:
     #   resources:

entrypoint.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+# Check if database is empty, if so, copy the init files
+if [ -z "$(ls -A /home/data/dbs)" ]; then
+	cp -r /home/init/* /home/data/dbs
+fi
+
+# Start the streamlit app
+streamlit run /home/meme_search/app.py --server.port=8501 --server.address=0.0.0.0