Update Dockerfile

Dockerfile

@@ -3,7 +3,7 @@ FROM ubuntu:latest
 
 # Update the package list, install curl, and clean up to reduce image size
 RUN apt-get update && \
-    apt-get install -y curl && \
+    apt-get install -y curl nginx && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
@@ -11,7 +11,8 @@ RUN apt-get update && \
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
 # Set the Ollama host environment variable
-ENV OLLAMA_HOST=0.0.0.0
+# ENV OLLAMA_HOST=0.0.0.0
+ENV OLLAMA_HOST=127.0.0.1
 
 # Create the Ollama directory and set permissions
 RUN mkdir -p /.ollama && chmod 777 /.ollama
@@ -22,9 +23,19 @@ RUN mkdir -p /usr/share/ollama/.ollama/models && chmod -R 777 /usr/share/ollama/
 # Set the Ollama models environment variable
 ENV OLLAMA_MODELS="/usr/share/ollama/.ollama/models"
 
+# Copy the Nginx configuration file
+COPY nginx.conf /etc/nginx/nginx.conf
+
+# Copy Lua script
+COPY validate_api_key.lua /usr/local/openresty/nginx/lua/validate_api_key.lua
+
 # Expose the Ollama server port
-EXPOSE 7860
+EXPOSE 80
 
 # Run the Ollama server and pull the model
-CMD ollama serve & sleep 5 && ollama pull llama3 && ollama pull llava-llama3 && wait
-# CMD ollama serve & sleep 5 && ollama pull llama3:70b && wait
+CMD ollama serve & \
+    sleep 5 && \
+    ollama pull llama3 && \
+    ollama pull llava-llama3 && \
+    wait && \
+    nginx -g 'daemon off;'