Update entrypoint.sh

entrypoint.sh

@@ -11,20 +11,27 @@ if [ -z "$AIBOM_INFERENCE_URL" ]; then
 fi
 
 # Check if command is provided
-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}
+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
 else
-  # Default: run as CLI
-  echo "Running AIBOM Generator CLI..."
-  exec python -m src.aibom_generator.cli "$@"
+  # Default: run as API server (web UI) 
+  echo "Starting AIBOM Generator API server (web UI)..."
+  exec uvicorn src.aibom_generator.api:app --host 0.0.0.0 --port ${PORT:-5000}
 fi