Add tests, CI/CD, CLI entry point, docker-compose, example config

Author: ORTODOX1

.github/workflows/ci.yml

@@ -0,0 +1,36 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  lint-test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.11"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Lint with ruff
+        run: ruff check argos/ tests/
+
+      - name: Type check with mypy
+        run: mypy argos/ --ignore-missing-imports
+
+      - name: Run tests
+        run: pytest tests/ -v --tb=short

.gitignore

@@ -0,0 +1,56 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+
+# Distribution / packaging
+build/
+dist/
+*.egg-info/
+*.egg
+.eggs/
+
+# Virtual environments
+venv/
+.venv/
+env/
+
+# Environment variables
+.env
+.env.*
+
+# ONNX and PyTorch model weights
+*.onnx
+*.pt
+*.pth
+*.tflite
+*.engine
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# Testing / coverage
+.pytest_cache/
+htmlcov/
+.coverage
+coverage.xml
+
+# mypy
+.mypy_cache/
+
+# ruff
+.ruff_cache/
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Docker
+docker-compose.override.yml
+
+# Logs
+*.log

argos/__main__.py

@@ -0,0 +1,101 @@
+"""
+argos.__main__ -- CLI entry point for the ARGOS inspection system.
+
+Usage:
+    python -m argos --config config.yaml --mode simulation
+"""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import logging
+import sys
+from pathlib import Path
+
+import yaml
+
+from argos.config import ArgosSettings
+from argos.inspector import InspectionEngine
+
+logger = logging.getLogger("argos")
+
+
+def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
+    """Parse command-line arguments."""
+    parser = argparse.ArgumentParser(
+        prog="argos",
+        description="ARGOS -- Autonomous Robot for General Onboard Surveillance",
+    )
+    parser.add_argument(
+        "--config", type=Path, default=Path("config.yaml"),
+        help="Path to YAML configuration file (default: config.yaml)",
+    )
+    parser.add_argument(
+        "--mode", choices=["simulation", "live", "replay"],
+        default="simulation",
+        help="Inspection mode: simulation (fake frames), live (camera), replay (video file)",
+    )
+    parser.add_argument(
+        "--replay-file", type=Path, default=None,
+        help="Path to video file for replay mode",
+    )
+    parser.add_argument(
+        "--interval", type=float, default=5.0,
+        help="Seconds between inspection cycles (default: 5.0)",
+    )
+    parser.add_argument(
+        "--log-level", default="INFO",
+        choices=["DEBUG", "INFO", "WARNING", "ERROR"],
+        help="Logging verbosity (default: INFO)",
+    )
+    return parser.parse_args(argv)
+
+
+def load_config(path: Path) -> ArgosSettings:
+    """Load settings from a YAML file, falling back to env/defaults."""
+    if path.exists():
+        with open(path) as fh:
+            raw = yaml.safe_load(fh) or {}
+        logger.info("Loaded config from %s", path)
+        return ArgosSettings(**raw)
+    logger.warning("Config file %s not found, using defaults", path)
+    return ArgosSettings()
+
+
+async def run(args: argparse.Namespace) -> None:
+    """Initialise and run the inspection engine."""
+    settings = load_config(args.config)
+    engine = InspectionEngine(settings=settings)
+
+    if args.mode == "simulation":
+        logger.info("Starting in SIMULATION mode (interval=%.1fs)", args.interval)
+    elif args.mode == "live":
+        logger.info("Starting in LIVE mode with camera device")
+    elif args.mode == "replay":
+        if not args.replay_file or not args.replay_file.exists():
+            logger.error("Replay mode requires --replay-file with a valid path")
+            sys.exit(1)
+        logger.info("Starting in REPLAY mode from %s", args.replay_file)
+
+    await engine.run_continuous(
+        gps_lat=45.4315, gps_lon=12.3456,
+        interval_s=args.interval,
+    )
+
+
+def main(argv: list[str] | None = None) -> None:
+    """Entry point for ``python -m argos``."""
+    args = parse_args(argv)
+    logging.basicConfig(
+        level=getattr(logging, args.log_level),
+        format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+    )
+    try:
+        asyncio.run(run(args))
+    except KeyboardInterrupt:
+        logger.info("Interrupted by user")
+
+
+if __name__ == "__main__":
+    main()

config.example.yaml

@@ -0,0 +1,47 @@
+# ARGOS Configuration
+# Copy to config.yaml and adjust for your environment.
+
+# --- Camera settings ---
+camera:
+  device_id: 0                # V4L2 device index
+  width: 1920
+  height: 1080
+  fps: 30
+  exposure_auto: true
+  white_balance_k: 5500       # Colour temperature in Kelvin
+
+# --- Edge processor (ONNX Runtime) ---
+edge:
+  detector_model_path: "models/defect_detector_v3.onnx"
+  classifier_model_path: "models/severity_classifier_v2.onnx"
+  execution_provider: "CUDAExecutionProvider"  # or CPUExecutionProvider
+  confidence_threshold: 0.45
+  nms_iou_threshold: 0.50
+  input_size: [640, 640]
+
+# --- SYNIZ TRIZ engine ---
+syniz:
+  ws_endpoint: "wss://syniz.fincantieri.internal/v1/ws"
+  api_key: ""                 # Set via ARGOS_SYNIZ__API_KEY env var
+  timeout_s: 30.0
+  max_reconnect_attempts: 5
+
+# --- POSEIDON-DIAG CAN bus ---
+poseidon:
+  can_interface: "can0"
+  can_bustype: "socketcan"
+  can_bitrate: 250000
+  j1939_source_address: 0xFE
+
+# --- ROS 2 topics ---
+ros:
+  image_topic: "/argos/camera/image_raw"
+  defect_topic: "/argos/defects"
+  odometry_topic: "/argos/odom"
+  health_topic: "/argos/health"
+
+# --- External services ---
+neo4j_uri: "bolt://localhost:7687"
+neo4j_user: "neo4j"
+neo4j_password: ""            # Set via ARGOS_NEO4J_PASSWORD env var
+aegis_dashboard_url: "https://aegis.fincantieri.internal/api/v2"

docker-compose.yml

@@ -0,0 +1,67 @@
+version: "3.9"
+
+services:
+  argos:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: argos-core
+    environment:
+      ARGOS_NEO4J_URI: bolt://neo4j:7687
+      ARGOS_NEO4J_USER: neo4j
+      ARGOS_NEO4J_PASSWORD: ${NEO4J_PASSWORD:-changeme}
+      ARGOS_SYNIZ__WS_ENDPOINT: wss://syniz-backend:8443/v1/ws
+      ARGOS_EDGE__EXECUTION_PROVIDER: CPUExecutionProvider
+    volumes:
+      - ./models:/app/models:ro
+      - ./config.yaml:/app/config.yaml:ro
+    ports:
+      - "8000:8000"
+    depends_on:
+      neo4j:
+        condition: service_healthy
+    networks:
+      - argos-net
+    restart: unless-stopped
+
+  neo4j:
+    image: neo4j:5-community
+    container_name: argos-neo4j
+    environment:
+      NEO4J_AUTH: neo4j/${NEO4J_PASSWORD:-changeme}
+      NEO4J_PLUGINS: '["apoc"]'
+    ports:
+      - "7474:7474"
+      - "7687:7687"
+    volumes:
+      - neo4j-data:/data
+    healthcheck:
+      test: ["CMD-SHELL", "cypher-shell -u neo4j -p ${NEO4J_PASSWORD:-changeme} 'RETURN 1'"]
+      interval: 10s
+      timeout: 5s
+      retries: 5
+    networks:
+      - argos-net
+    restart: unless-stopped
+
+  syniz-backend:
+    image: ghcr.io/ortodox1/syniz:latest
+    container_name: argos-syniz
+    environment:
+      SYNIZ_PORT: "8443"
+      SYNIZ_NEO4J_URI: bolt://neo4j:7687
+    ports:
+      - "8443:8443"
+    depends_on:
+      neo4j:
+        condition: service_healthy
+    networks:
+      - argos-net
+    restart: unless-stopped
+
+volumes:
+  neo4j-data:
+
+networks:
+  argos-net:
+    driver: bridge

pyproject.toml

@@ -31,6 +31,7 @@ dependencies = [
     "numpy>=1.26.0",
     "httpx>=0.27.0",
     "uvicorn>=0.29.0",
+    "pyyaml>=6.0",
 ]
 
 [project.optional-dependencies]

requirements.txt

@@ -8,3 +8,4 @@ pydantic>=2.6.0
 numpy>=1.26.0
 httpx>=0.27.0
 uvicorn>=0.29.0
+pyyaml>=6.0