Update Dockerfile

Dockerfile

@@ -1,14 +1,31 @@
-FROM python:3.10-slim
+# Use the official NewAPI image as the foundation
+FROM calciumion/new-api:latest
 
-WORKDIR /app
+USER root
+
+# 1. Install Python and Flask for the Shield/UI
+RUN apt-get update && apt-get install -y \
+    python3 \
+    python3-pip \
+    ca-certificates \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install dependencies
-RUN apt-get update && apt-get install -y ca-certificates && rm -rf /var/lib/apt/lists/*
-RUN pip install flask requests mysql-connector-python
+# 2. Install Python dependencies
+RUN pip3 install flask requests mysql-connector-python --break-system-packages
+
+# 3. Set up the working directory
+WORKDIR /app
 
-# Copy ALL files (shield.py AND index.html)
-COPY . .
+# 4. Copy your custom UI and the Shield script
+# Make sure these files are in the same folder as your Dockerfile!
+COPY shield.py /app/shield.py
+COPY index.html /app/index.html
 
+# 5. Hugging Face uses port 7860
+ENV PORT=7860
 EXPOSE 7860
 
-CMD ["python3", "shield.py"]
+# 6. The "Magic" Launch Command
+# We start the real NewAPI on port 3000 in the background (&)
+# Then we start the Python Shield on 7860 in the foreground
+CMD /new-api --port 3000 & python3 /app/shield.py