Spaces:

Text-to-Document-Generation
/

Docgenie-API

Paused

Ahadhassan-2003

deploy: update HF Space

dc4e6da 6 months ago

2.2 kB

#!/bin/bash
set -e

echo "🚀 Starting DocGenie API + Worker..."
echo "============================================================"
echo "🔧 Configuration:"
echo "   PORT: ${PORT:-7860}"
echo "   REDIS_URL: ${REDIS_URL:0:30}..."
echo "   ANTHROPIC_API_KEY: ${ANTHROPIC_API_KEY:0:20}..."
echo "============================================================"

# Function to add prefix to logs
prefix_logs() {
    local prefix=$1
    while IFS= read -r line; do
        echo "[$prefix] $line"
    done
}

# Start API server in background with log prefix
echo "🌐 Starting FastAPI server..."
uvicorn api.main:app --host 0.0.0.0 --port ${PORT:-7860} 2>&1 | prefix_logs "API" &
API_PID=$!
echo "   PID: $API_PID"

# Give API a moment to start
sleep 2

# Start RQ worker in background with log prefix
# Use timestamp + random number to ensure unique worker name across deployments
WORKER_ID="docgenie-worker-$(date +%s)-${RANDOM}"
echo "👷 Starting RQ Worker..."
rq worker ${RQ_QUEUE_NAME:-docgenie} --url $REDIS_URL --name "$WORKER_ID" --verbose 2>&1 | prefix_logs "WORKER" &
WORKER_PID=$!
echo "   PID: $WORKER_PID"

echo "============================================================"
echo "✓ Both processes started successfully"
echo "   API: http://0.0.0.0:${PORT:-7860}"
echo "   Worker: Connected to Redis"
echo "============================================================"

# Function to handle shutdown
shutdown() {
    echo ""
    echo "🛑 Shutting down gracefully..."
    kill -TERM $API_PID 2>/dev/null || true
    kill -TERM $WORKER_PID 2>/dev/null || true
    wait $API_PID 2>/dev/null || true
    wait $WORKER_PID 2>/dev/null || true
    echo "✓ Shutdown complete"
    exit 0
}

# Trap SIGTERM and SIGINT
trap shutdown SIGTERM SIGINT

# Wait for both processes - if either exits, shutdown both
while kill -0 $API_PID 2>/dev/null && kill -0 $WORKER_PID 2>/dev/null; do
    sleep 1
done

# If we get here, one process died
echo ""
echo "❌ One of the processes exited unexpectedly"
if ! kill -0 $API_PID 2>/dev/null; then
    echo "   API process died (PID: $API_PID)"
fi
if ! kill -0 $WORKER_PID 2>/dev/null; then
    echo "   Worker process died (PID: $WORKER_PID)"
fi

shutdown

	#!/bin/bash
	set -e

	echo "🚀 Starting DocGenie API + Worker..."
	echo "============================================================"
	echo "🔧 Configuration:"
	echo " PORT: ${PORT:-7860}"
	echo " REDIS_URL: ${REDIS_URL:0:30}..."
	echo " ANTHROPIC_API_KEY: ${ANTHROPIC_API_KEY:0:20}..."
	echo "============================================================"

	# Function to add prefix to logs
	prefix_logs() {
	local prefix=$1
	while IFS= read -r line; do
	echo "[$prefix] $line"
	done
	}

	# Start API server in background with log prefix
	echo "🌐 Starting FastAPI server..."
	uvicorn api.main:app --host 0.0.0.0 --port ${PORT:-7860} 2>&1 \| prefix_logs "API" &
	API_PID=$!
	echo " PID: $API_PID"

	# Give API a moment to start
	sleep 2

	# Start RQ worker in background with log prefix
	# Use timestamp + random number to ensure unique worker name across deployments
	WORKER_ID="docgenie-worker-$(date +%s)-${RANDOM}"
	echo "👷 Starting RQ Worker..."
	rq worker ${RQ_QUEUE_NAME:-docgenie} --url $REDIS_URL --name "$WORKER_ID" --verbose 2>&1 \| prefix_logs "WORKER" &
	WORKER_PID=$!
	echo " PID: $WORKER_PID"

	echo "============================================================"
	echo "✓ Both processes started successfully"
	echo " API: http://0.0.0.0:${PORT:-7860}"
	echo " Worker: Connected to Redis"
	echo "============================================================"

	# Function to handle shutdown
	shutdown() {
	echo ""
	echo "🛑 Shutting down gracefully..."
	kill -TERM $API_PID 2>/dev/null \|\| true
	kill -TERM $WORKER_PID 2>/dev/null \|\| true
	wait $API_PID 2>/dev/null \|\| true
	wait $WORKER_PID 2>/dev/null \|\| true
	echo "✓ Shutdown complete"
	exit 0
	}

	# Trap SIGTERM and SIGINT
	trap shutdown SIGTERM SIGINT

	# Wait for both processes - if either exits, shutdown both
	while kill -0 $API_PID 2>/dev/null && kill -0 $WORKER_PID 2>/dev/null; do
	sleep 1
	done

	# If we get here, one process died
	echo ""
	echo "❌ One of the processes exited unexpectedly"
	if ! kill -0 $API_PID 2>/dev/null; then
	echo " API process died (PID: $API_PID)"
	fi
	if ! kill -0 $WORKER_PID 2>/dev/null; then
	echo " Worker process died (PID: $WORKER_PID)"
	fi

	shutdown