Avinashnalla7 commited on
Commit
70eb619
·
1 Parent(s): 96c8a10

Fix worker start: run as module + keep health server

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +8 -0
  3. start.sh +4 -18
Dockerfile CHANGED
@@ -10,4 +10,4 @@ COPY . /app
10
  ENV PYTHONUNBUFFERED=1
11
  EXPOSE 7860
12
 
13
- CMD ["sh","-lc","set -e; python -u backend/worker/worker.py & exec uvicorn health_api:app --host 0.0.0.0 --port ${PORT:-7860}"]
 
10
  ENV PYTHONUNBUFFERED=1
11
  EXPOSE 7860
12
 
13
+ CMD ["sh","-lc","./start.sh"]
requirements.txt CHANGED
@@ -13,3 +13,11 @@ PyMuPDF>=1.23.0
13
  fastapi==0.128.0
14
  uvicorn==0.40.0
15
  python-dotenv
 
 
 
 
 
 
 
 
 
13
  fastapi==0.128.0
14
  uvicorn==0.40.0
15
  python-dotenv
16
+ google-auth
17
+ google-auth-oauthlib
18
+ google-api-python-client
19
+ openai>=1.0.0
20
+ PyMuPDF==1.26.7
21
+ google-auth-httplib2
22
+ openai
23
+ Pillow
start.sh CHANGED
@@ -1,22 +1,8 @@
1
- #!/bin/sh
2
  set -eu
3
 
4
- export PYTHONUNBUFFERED=1
5
- export PYTHONPATH=/app
6
-
7
  python -u -m backend.worker.worker &
8
- WORKER_PID=$!
9
-
10
- cleanup() {
11
- kill "$WORKER_PID" 2>/dev/null || true
12
- }
13
- trap cleanup EXIT
14
-
15
- uvicorn health_api:app --host 0.0.0.0 --port "${PORT:-7860}" &
16
- API_PID=$!
17
-
18
- wait -n "$WORKER_PID" "$API_PID"
19
- STATUS=$?
20
 
21
- kill "$WORKER_PID" "$API_PID" 2>/dev/null || true
22
- exit "$STATUS"
 
1
+ #!/usr/bin/env sh
2
  set -eu
3
 
4
+ # Start the worker (must be module so relative imports work)
 
 
5
  python -u -m backend.worker.worker &
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
+ # Health server (HF needs a listening web process)
8
+ exec uvicorn health_api:app --host 0.0.0.0 --port "${PORT:-7860}"