Really-amin commited on
Commit
eed76a9
·
verified ·
1 Parent(s): 0a63b97

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -12
Dockerfile CHANGED
@@ -41,7 +41,7 @@ RUN apt-get update && apt-get install -y \
41
  libmupdf-dev \
42
  ffmpeg \
43
  libopencv-dev \
44
- libgl1-mesa-glx \
45
  libglib2.0-0 \
46
  libsm6 \
47
  libxext6 \
@@ -56,7 +56,7 @@ RUN apt-get update && apt-get install -y \
56
  libpangocairo-1.0-0 \
57
  libatk1.0-0 \
58
  libcairo-gobject2 \
59
- libgdk-pixbuf2.0-0 \
60
  libffi-dev \
61
  libssl-dev \
62
  curl \
@@ -75,21 +75,16 @@ ENV PATH="/opt/venv/bin:$PATH"
75
  # Set working directory
76
  WORKDIR /app
77
 
78
- # Copy all files (ensuring frontend stays in /app/frontend)
79
  COPY --chown=appuser:appuser . .
80
 
81
- # ✅ Ensure frontend exists
82
- RUN mkdir -p /app/frontend && chmod -R 755 /app/frontend
83
-
84
  # Pre-download TrOCR model to speed up startup
85
  RUN python -c "from transformers import TrOCRProcessor, VisionEncoderDecoderModel; \
86
  TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed'); \
87
- VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-printed')"
88
 
89
  # Environment variables
90
  ENV PYTHONPATH=/app
91
- ENV DATABASE_DIR=/app/data
92
- ENV DATABASE_PATH=/app/data/legal_documents.db
93
  ENV TRANSFORMERS_CACHE=/app/cache
94
  ENV HF_HOME=/app/cache
95
  ENV LOG_LEVEL=INFO
@@ -104,7 +99,7 @@ EXPOSE 7860
104
 
105
  # Healthcheck
106
  HEALTHCHECK --interval=45s --timeout=30s --start-period=180s --retries=10 \
107
- CMD curl -fs http://localhost:7860/health || exit 1
108
 
109
- # Start app
110
- CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1", "--log-level", "info"]
 
41
  libmupdf-dev \
42
  ffmpeg \
43
  libopencv-dev \
44
+ libgl1-mesa-dri \
45
  libglib2.0-0 \
46
  libsm6 \
47
  libxext6 \
 
56
  libpangocairo-1.0-0 \
57
  libatk1.0-0 \
58
  libcairo-gobject2 \
59
+ libgdk-pixbuf-2.0-0 \
60
  libffi-dev \
61
  libssl-dev \
62
  curl \
 
75
  # Set working directory
76
  WORKDIR /app
77
 
78
+ # Copy all files
79
  COPY --chown=appuser:appuser . .
80
 
 
 
 
81
  # Pre-download TrOCR model to speed up startup
82
  RUN python -c "from transformers import TrOCRProcessor, VisionEncoderDecoderModel; \
83
  TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed'); \
84
+ VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-printed')" || echo "Model download failed, will download at runtime"
85
 
86
  # Environment variables
87
  ENV PYTHONPATH=/app
 
 
88
  ENV TRANSFORMERS_CACHE=/app/cache
89
  ENV HF_HOME=/app/cache
90
  ENV LOG_LEVEL=INFO
 
99
 
100
  # Healthcheck
101
  HEALTHCHECK --interval=45s --timeout=30s --start-period=180s --retries=10 \
102
+ CMD python -c "import requests; requests.get('http://localhost:7860')" || exit 1
103
 
104
+ # Start Gradio app (not FastAPI)
105
+ CMD ["python", "app.py"]