Update docker-compose.yml

docker-compose.yml

@@ -1,26 +1,28 @@
 version: '3.8'
 
 services:
-  persian_ocr:
+  ocr-api:
     build: .
-    container_name: persian_ocr_service
     ports:
-      - "7860:7860"
+      - "8000:8000"
     volumes:
-      - ./:/app
-      - model_cache:/root/.cache  # Cache for downloaded models
+      - ./temp_uploads:/app/temp_uploads
+      - ./output:/app/output
+      - ./config:/app/config
     environment:
       - PYTHONUNBUFFERED=1
-      - GRADIO_SERVER_NAME=0.0.0.0
-      - GRADIO_SERVER_PORT=7860
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: all
-              capabilities: [gpu]
+      - TESSDATA_PREFIX=/usr/share/tesseract-ocr/4.00/tessdata
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
     restart: unless-stopped
 
-volumes:
-  model_cache:  # Persistent volume for model cache
+  # Optional service for monitoring
+  watchtower:
+    image: containrrr/watchtower
+    volumes:
+      - /var/run/docker.sock:/var/run/docker.sock
+    command: --interval 30 --cleanup
+    restart: unless-stopped