Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
 FROM ollama/ollama:latest
 
-# Install Python and dependencies
+# Install Python and Gatekeeper dependencies
 RUN apt-get update && apt-get install -y python3 python3-pip && \
     pip3 install flask flask-cors requests
 
@@ -11,24 +11,30 @@ ENV HOME=/home/ollama
 
 RUN mkdir -p /home/ollama/.ollama && chmod 777 /home/ollama/.ollama
 
-# Create the Security Proxy (Cleaner formatting to prevent Exit Code 1)
-RUN printf 'from flask import Flask, request, Response\n\
-import requests\n\
-from flask_cors import CORS\n\
-app = Flask(__name__)\n\
-CORS(app)\n\
-@app.route("/api/generate", methods=["POST"])\n\
-def proxy():\n\
-    key = request.headers.get("x-api-key", "")\n\
-    if not key.startswith("sk-") or len(key) < 10:\n\
-        return {"error": "Invalid API Key"}, 401\n\
-    try:\n\
-        resp = requests.post("http://127.0.0.1:11434/api/generate", json=request.json)\n\
-        return Response(resp.content, resp.status_code, resp.headers.items())\n\
-    except:\n\
-        return {"error": "Ollama not ready"}, 503\n\
-if __name__ == "__main__":\n\
-    app.run(host="0.0.0.0", port=7860)' > /guard.py
-
-# Start Ollama and the Proxy together
+# Create the Security Proxy using a Here-Doc (Fixes Exit Code 1)
+RUN cat <<EOF > /guard.py
+from flask import Flask, request, Response
+import requests
+from flask_cors import CORS
+
+app = Flask(__name__)
+CORS(app)
+
+@app.route("/api/generate", methods=["POST"])
+def proxy():
+    key = request.headers.get("x-api-key", "")
+    # Gatekeeper: Any key starting with sk- and longer than 10 chars is valid
+    if not key.startswith("sk-") or len(key) < 10:
+        return {"error": "Invalid API Key"}, 401
+    try:
+        resp = requests.post("http://127.0.0.1:11434/api/generate", json=request.json)
+        return Response(resp.content, resp.status_code, resp.headers.items())
+    except:
+        return {"error": "Ollama server is still starting..."}, 503
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)
+EOF
+
+# Start Ollama, pull the model, then start the Python Guard
 CMD ["/bin/sh", "-c", "ollama serve & sleep 15 && ollama pull phi3 && python3 /guard.py"]