Update health_check.sh

health_check.sh

@@ -1,37 +1,81 @@
-# Health check script for Docker container
+#!/bin/bash
+
+# Health check script for Docker container with dynamic path detection
 set -e
 
+# Environment and path detection
+if [ "$HF_SPACES_BUILD" = "1" ] || [ -n "$SPACE_ID" ]; then
+    BASE_DIR="/app/persistent"
+    ENVIRONMENT="huggingface_spaces"
+else
+    BASE_DIR="/tmp"
+    ENVIRONMENT="local"
+fi
+
+MODEL_DIR="$BASE_DIR/model"
+DATA_DIR="$BASE_DIR/data"
+
+echo "Health check running in $ENVIRONMENT environment"
+echo "Base directory: $BASE_DIR"
+
 # Check if FastAPI is responding
-if ! curl -s -f "http://127.0.0.1:8000/docs" > /dev/null; then
+echo "Checking FastAPI health..."
+if ! curl -s -f "http://127.0.0.1:8000/health" > /dev/null; then
     echo "FastAPI health check failed"
     exit 1
 fi
+echo "FastAPI health check passed"
 
-# Check if Streamlit is responding
-if ! curl -s -f "http://127.0.0.1:7860/_stcore/health" > /dev/null; then
-    echo "Streamlit health check failed"
-    exit 1
+# Check if Streamlit is responding (optional)
+echo "Checking Streamlit health..."
+if ! curl -s -f "http://127.0.0.1:7860/_stcore/health" > /dev/null 2>&1; then
+    echo "Streamlit health check failed (this might be normal during startup)"
+    # Don't exit on Streamlit failure as it might not be started yet
+else
+    echo "Streamlit health check passed"
 fi
 
-# Check if required files exist
+# Check required files with dynamic paths
+echo "Checking required files..."
+
 required_files=(
-    "/tmp/model.pkl"
-    "/tmp/vectorizer.pkl"
-    "/tmp/data/combined_dataset.csv"
+    "$MODEL_DIR/pipeline.pkl"
+    "$BASE_DIR/metadata.json"
+    "$DATA_DIR/combined_dataset.csv"
 )
 
+missing_files=0
 for file in "${required_files[@]}"; do
     if [ ! -f "$file" ]; then
         echo "Required file missing: $file"
-        exit 1
+        missing_files=$((missing_files + 1))
+    else
+        file_size=$(stat -c%s "$file" 2>/dev/null || echo "unknown")
+        echo "Required file found: $file (${file_size} bytes)"
     fi
 done
 
-# Check if processes are running
-if ! pgrep -f "schedule_tasks.py" > /dev/null; then
-    echo "Scheduler process not running"
+if [ $missing_files -gt 0 ]; then
+    echo "Health check failed: $missing_files required files missing"
     exit 1
 fi
 
-echo "All health checks passed"
+# Check background processes (optional)
+echo "Checking background processes..."
+
+if pgrep -f "schedule_tasks.py" > /dev/null; then
+    echo "Scheduler process is running"
+else
+    echo "Scheduler process not running (this might be normal)"
+fi
+
+if pgrep -f "monitor_drift.py" > /dev/null; then
+    echo "Monitor process is running"
+else
+    echo "Monitor process not running (this might be normal)"
+fi
+
+echo "All critical health checks passed"
+echo "Environment: $ENVIRONMENT"
+echo "Base directory: $BASE_DIR"
 exit 0