Avinashnalla7 commited on
Commit
96c8a10
·
1 Parent(s): 3c5210f

Fix worker startup: run as module, add package init

Browse files
Files changed (1) hide show
  1. start.sh +14 -3
start.sh CHANGED
@@ -1,11 +1,22 @@
1
  #!/bin/sh
2
  set -eu
3
 
 
 
 
4
  python -u -m backend.worker.worker &
5
  WORKER_PID=$!
6
 
 
 
 
 
 
7
  uvicorn health_api:app --host 0.0.0.0 --port "${PORT:-7860}" &
8
- UVICORN_PID=$!
 
 
 
9
 
10
- wait "$UVICORN_PID"
11
- kill "$WORKER_PID" 2>/dev/null || true
 
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"