Upload 3 files

Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY src/ /app/src/
+
+# Set environment variables
+ENV PYTHONPATH=/app
+
+# Create entrypoint script
+COPY entrypoint.sh /app/
+RUN chmod +x /app/entrypoint.sh
+
+# Command to run the application
+ENTRYPOINT ["/app/entrypoint.sh"]

entrypoint.sh

@@ -0,0 +1,30 @@
+#!/bin/bash
+set -e
+
+# Default values
+DEFAULT_INFERENCE_URL="http://inference-model:8000/extract"
+
+# 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
+
+# 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}
+else
+  # Default: run as CLI
+  echo "Running AIBOM Generator CLI..."
+  exec python -m src.aibom_generator.cli "$@"
+fi

requirements.txt

@@ -0,0 +1,12 @@
+huggingface_hub>=0.19.0
+transformers>=4.36.0
+torch>=2.0.0
+fastapi>=0.104.0
+uvicorn>=0.24.0
+pydantic>=2.4.0
+requests>=2.31.0
+python-dotenv>=1.0.0
+PyYAML>=6.0.1
+flask>=2.3.0
+gunicorn>=21.2.0
+cyclonedx-python-lib>=4.0.0