Update Dockerfile

Dockerfile

@@ -1,29 +1,34 @@
 FROM ollama/ollama:latest
 
-# Install Python and Flask for the gateway
-RUN apt-get update && apt-get install -y python3 python3-pip && pip3 install flask flask-cors requests
+# Install Python and dependencies
+RUN apt-get update && apt-get install -y python3 python3-pip && \
+    pip3 install flask flask-cors requests
 
 ENTRYPOINT []
 ENV OLLAMA_HOST=127.0.0.1:11434
 ENV OLLAMA_ORIGINS="*"
 ENV HOME=/home/ollama
+
 RUN mkdir -p /home/ollama/.ollama && chmod 777 /home/ollama/.ollama
 
-# Create a dynamic Security Proxy
-# This version accepts ANY key starting with "sk-"
-RUN echo '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", ""); \
-    if not key.startswith("sk-") or len(key) < 10: \
-        return {"error": "Invalid API Key format"}, 401; \
-    resp = requests.post("http://127.0.0.1:11434/api/generate", json=request.json); \
-    return Response(resp.content, resp.status_code, resp.headers.items()); \
-if __name__ == "__main__": \
+# 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
 
-CMD ["/bin/sh", "-c", "ollama serve & sleep 10 && ollama pull phi3 && python3 /guard.py"]
+# Start Ollama and the Proxy together
+CMD ["/bin/sh", "-c", "ollama serve & sleep 15 && ollama pull phi3 && python3 /guard.py"]