More ideas...

Dockerfile

@@ -1,13 +1,32 @@
-FROM python:3.12
+FROM ubuntu:22.04
+
+USER root
+
+RUN apt-get update
+RUN apt-get install -y python3-pip netcat curl
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Install ollama
+RUN curl -fsSL https://ollama.com/install.sh | sh
+
+RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
 EXPOSE 7860
 
+
+# Copy the ollama_setup.sh script
+COPY ollama_setup.sh /usr/local/bin/ollama_setup.sh
+
+# Set permissions for the ollama_setup.sh script
+RUN chmod +x /usr/local/bin/ollama_setup.sh
+
+# Use the ollama_setup.sh
+RUN /usr/local/bin/ollama_setup.sh
+
 CMD ["shiny", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]

build.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+# Set the platform flag if we're on ARM
+arch=$(uname -m)
+if [[ "$arch" == "arm64" || "$arch" == "aarch64" ]]; then
+    platform_flag="--platform linux/amd64"
+else
+    platform_flag=""
+fi
+
+# Build the Docker image
+
+docker build --progress=plain $platform_flag "$@" -t "openworm_ai" .
+

ollama_setup.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+
+# Start Ollama server in the background
+/usr/local/bin/ollama serve &
+
+# Wait for the server to be ready
+while ! nc -z localhost 11434; do
+    echo "Waiting for Ollama server to start..."
+    sleep 2
+done
+
+# Pull the model
+echo "Pulling the model..."
+#ollama pull nomic-embed-text
+ollama pull llama3.2:1b
+
+# Keep the container running
+#wait

run.sh

@@ -0,0 +1,2 @@
+
+docker run --name=openworm_ai_0 -p 8888:7860 openworm_ai 

stop.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+docker stop openworm_ai_0
+docker rm openworm_ai_0