Update Dockerfile

Dockerfile

@@ -1,9 +1,12 @@
 # Use Python 3.10
 FROM python:3.10-slim
 
-# 1. Install Java 21 (Required for LanguageTool) and Build Tools
+# 1. Install System Tools (Git, Java, Wget)
+# Java is required for LanguageTool
 RUN apt-get update && \
     apt-get install -y \
+    git \
+    wget \
     openjdk-21-jre-headless \
     build-essential \
     python3-dev \
@@ -16,19 +19,39 @@ ENV JAVA_HOME=/usr/lib/jvm/java-21-openjdk-amd64
 # 2. Set working directory
 WORKDIR /app
 
-# 3. SET WRITABLE CACHE FOLDERS (Fixes Runtime Error)
+# 3. Setup Cache Folders (Prevents Permission Errors)
 ENV XDG_CACHE_HOME=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
 ENV HF_HOME=/app/cache
 RUN mkdir -p /app/cache && chmod 777 /app/cache
 
-# 4. Install dependencies
+# 4. Install Dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# 5. Copy app
+# 5. INSTALL GECToR LIBRARY
+# We clone the library code so python can "import gector"
+RUN git clone https://github.com/gotutiyan/gector.git /app/gector_lib
+RUN pip install --no-cache-dir /app/gector_lib
+
+# 6. SETUP DATA & MODEL
+RUN mkdir -p /app/data
+
+# A. DOWNLOAD MODEL WEIGHTS (Heavy file - 500MB)
+# We download this here so you don't have to upload it to Git
+RUN wget -O /app/data/gector_model.th https://huggingface.co/gotutiyan/gector-roberta-base-5k/resolve/main/pytorch_model.bin
+
+# B. MOVE YOUR UPLOADED VOCAB FILE
+# COPY . . (in step 7) will bring 'verb-form-vocab.txt' to /app/
+# We will move it to /app/data/ after copying, or just point the code to it.
+
+# 7. Copy Your App Code (Includes verb-form-vocab.txt)
 COPY . .
 
-# 6. Run app
+# 8. Move the vocab file to the expected folder (Optional but clean)
+# Assuming you uploaded it to the root of your Space
+RUN mv verb-form-vocab.txt /app/data/verb-form-vocab.txt || echo "File not found in root, skipping move"
+
+# 9. Run App
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]