Update Dockerfile

Dockerfile

@@ -1,42 +1,76 @@
-FROM python:3.11.11-slim-bullseye as builder
+FROM python:3.11.11-slim-bullseye
 
+# Install system dependencies
 RUN apt-get update -qq \
   && apt-get -qqq install --no-install-recommends -y pkg-config gcc g++ \
   && apt-get clean \
   && rm -rf /var/lib/apt/lists/*
 
-RUN python -mvenv venv && ./venv/bin/pip install --upgrade pip
+# Create non-root user
+RUN addgroup --system --gid 1032 libretranslate \
+  && adduser --system --uid 1032 --gid 1032 libretranslate
+
+# Set working directory
+WORKDIR /app
+
+# Create virtual environment
+RUN python -m venv venv
 
-COPY . .
+# Upgrade pip
+RUN ./venv/bin/pip install --upgrade pip
 
-# Install package from source code, compile translations
-RUN ./venv/bin/pip install Babel==2.12.1 && ./venv/bin/python scripts/compile_locales.py \
+# Clone LibreTranslate repository
+RUN apt-get update -qq \
+  && apt-get install -y git \
+  && git clone https://github.com/LibreTranslate/LibreTranslate.git . \
+  && apt-get remove -y git \
+  && apt-get autoremove -y \
+  && apt-get clean \
+  && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+RUN ./venv/bin/pip install Babel==2.12.1 \
+  && ./venv/bin/python scripts/compile_locales.py \
   && ./venv/bin/pip install torch==2.2.0 --extra-index-url https://download.pytorch.org/whl/cpu \
   && ./venv/bin/pip install "numpy<2" \
   && ./venv/bin/pip install . \
+  && ./venv/bin/pip install gunicorn \
   && ./venv/bin/pip cache purge
 
-FROM python:3.11.11-slim-bullseye
+# Create necessary directories
+RUN mkdir -p /app/db /app/logs /tmp/prometheus_data \
+  && chown -R libretranslate:libretranslate /app /tmp/prometheus_data
 
-ARG with_models=false
-ARG models=""
+# Copy wsgi.py file (if not in the repo)
+RUN echo 'from app.main import create_app\n\
+\n\
+def app(*args, **kwargs):\n\
+    import argparse\n\
+    parser = argparse.ArgumentParser()\n\
+    import app.main as main\n\
+    main.get_args(parser)\n\
+    args = parser.parse_args([])\n\
+    for k, v in kwargs.items():\n\
+        setattr(args, k, v)\n\
+    return create_app(args)' > /app/wsgi.py
 
-RUN addgroup --system --gid 1032 libretranslate && adduser --system --uid 1032 libretranslate && mkdir -p /home/libretranslate/.local && chown -R libretranslate:libretranslate /home/libretranslate/.local
+# Switch to non-root user
 USER libretranslate
 
-COPY --from=builder --chown=1032:1032 /app /app
-WORKDIR /app
-
-COPY --from=builder --chown=1032:1032 /app/venv/bin/ltmanage /usr/bin/
+# Set environment variables
+ENV PROMETHEUS_MULTIPROC_DIR=/tmp/prometheus_data
+ENV LT_LOAD_ONLY=""
+ENV LT_UPDATE_MODELS=false
 
-RUN if [ "$with_models" = "true" ]; then  \
-  # initialize the language models
-  if [ ! -z "$models" ]; then \
-  ./venv/bin/python scripts/install_models.py --load_only_lang_codes "$models";   \
-  else \
-  ./venv/bin/python scripts/install_models.py;  \
-  fi \
-  fi
+# Expose port
+EXPOSE 7860
 
-EXPOSE 5000
-ENTRYPOINT [ "./venv/bin/libretranslate", "--host", "*" ]
+# Default command with Gunicorn
+CMD ["./venv/bin/gunicorn", \
+     "--workers", "3", \
+     "--bind", "0.0.0.0:7860", \
+     "--timeout", "300", \
+     "--log-level", "info", \
+     "--access-logfile", "-", \
+     "--error-logfile", "-", \
+     "wsgi:app"]