Update Dockerfile

Dockerfile

@@ -18,7 +18,9 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     libffi-dev \
     cmake \
-    libcurl4-openssl-dev && \
+    libcurl4-openssl-dev \
+    tini \
+    systemd && \
     apt-get clean
 
 # Upgrade pip and install dependencies
@@ -29,10 +31,6 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
 
-RUN systemctl start ollama
-
-# Download the required model
-RUN ollama pull llama3
 
 # Copy the entire application
 COPY . .
@@ -40,5 +38,32 @@ COPY . .
 # Set proper permissions for the translations directory
 RUN chmod -R 777 translations
 
+# Create the Ollama user
+RUN useradd -r -s /bin/false -m -d /usr/share/ollama ollama
+
+# Create the service file
+RUN echo "[Unit]\n\
+Description=Ollama Service\n\
+After=network-online.target\n\n\
+[Service]\n\
+ExecStart=/usr/bin/ollama serve\n\
+User=ollama\n\
+Group=ollama\n\
+Restart=always\n\
+RestartSec=3\n\n\
+[Install]\n\
+WantedBy=default.target" > /etc/systemd/system/ollama.service
+
+# Enable the service
+RUN systemctl enable ollama
+
+# Copy the entrypoint script
+COPY entrypoint.sh /usr/local/bin/entrypoint.sh
+RUN chmod +x /usr/local/bin/entrypoint.sh
+
+# Define the entrypoint to use tini and the custom script
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]
+# Download the required model
+RUN ollama pull llama3
 # Define the command to run the application
 CMD ["python", "./run.py"]