Update entrypoint.sh

entrypoint.sh

@@ -1,37 +1,38 @@
 #!/bin/bash
 set -e
 
-# Default values
-DEFAULT_INFERENCE_URL="http://inference-model:8000/extract"
+# Default inference URL for internal inference model service
+DEFAULT_INFERENCE_URL="http://localhost:8000/extract"
+export AIBOM_INFERENCE_URL=${AIBOM_INFERENCE_URL:-$DEFAULT_INFERENCE_URL}
 
-# Check if AIBOM_INFERENCE_URL is set, otherwise use default
-if [ -z "$AIBOM_INFERENCE_URL" ]; then
-  export AIBOM_INFERENCE_URL=$DEFAULT_INFERENCE_URL
-  echo "AIBOM_INFERENCE_URL not set, using default: $AIBOM_INFERENCE_URL"
-fi
+echo "Using AIBOM_INFERENCE_URL: $AIBOM_INFERENCE_URL"
 
-# Check if command is provided
+# Check if command-line arguments are provided
 if [ -n "$1" ]; then
-  # If arguments are provided, use them
-  if [ "$1" = "server" ]; then
-    # Start the API server
-    echo "Starting AIBOM Generator API server..."
-    exec uvicorn src.aibom_generator.api:app --host 0.0.0.0 --port ${PORT:-5000}
-  elif [ "$1" = "worker" ]; then
-    # Start the background worker
-    echo "Starting AIBOM Generator background worker..."
-    exec python -m src.aibom_generator.worker
-  elif [ "$1" = "inference" ]; then
-    # Start the inference model server
-    echo "Starting AIBOM Generator inference model server..."
-    exec python -m src.aibom_generator.inference_model --host 0.0.0.0 --port ${PORT:-8000}
-  else
-    # Run as CLI with provided arguments
-    echo "Running AIBOM Generator CLI..."
-    exec python -m src.aibom_generator.cli "$@"
-  fi
+  case "$1" in
+    server)
+      # Start the API server explicitly (recommended for Hugging Face Spaces)
+      echo "Starting AIBOM Generator API server..."
+      exec uvicorn src.aibom_generator.api:app --host 0.0.0.0 --port ${PORT:-7860}
+      ;;
+    worker)
+      # Start the background worker
+      echo "Starting AIBOM Generator background worker..."
+      exec python -m src.aibom_generator.worker
+      ;;
+    inference)
+      # Start the inference model server
+      echo "Starting AIBOM Generator inference model server..."
+      exec python -m src.aibom_generator.inference_model --host 0.0.0.0 --port ${PORT:-8000}
+      ;;
+    *)
+      # Run as CLI with provided arguments
+      echo "Running AIBOM Generator CLI..."
+      exec python -m src.aibom_generator.cli "$@"
+      ;;
+  esac
 else
-  # Default: run as API server (web UI) 
+  # Default behavior (if no arguments): start API server (web UI mode)
   echo "Starting AIBOM Generator API server (web UI)..."
-  exec uvicorn src.aibom_generator.api:app --host 0.0.0.0 --port ${PORT:-5000}
+  exec uvicorn src.aibom_generator.api:app --host 0.0.0.0 --port ${PORT:-7860}
 fi