main.py

import signal
import threading
import time

from api.alert_api import create_app
from api.alert_manager import AlertManager
from audio.audio_model import AudioClassifier
from audio.audio_stream import AudioStream
from config import settings
from fusion.fusion_engine import FusionEngine
from pipeline.detection_pipeline import DetectionPipeline
from utils.helpers import ensure_directories
from utils.logger import setup_logger
from vision.camera_stream import CameraStream
from vision.vision_model import VisionDetector


def run_api(app) -> None:
    app.run(host=settings.api_host, port=settings.api_port, debug=False, use_reloader=False)


def main() -> None:
    logger = setup_logger(settings.log_level)
    ensure_directories([settings.models_dir, settings.audio_dataset_dir])

    alert_manager = AlertManager()
    audio_stream = AudioStream()
    camera_stream = CameraStream()
    audio_classifier = AudioClassifier()
    vision_detector = VisionDetector()
    fusion_engine = FusionEngine()
    detection_pipeline = DetectionPipeline(
        camera_stream=camera_stream,
        audio_stream=audio_stream,
        vision_detector=vision_detector,
        audio_classifier=audio_classifier,
        fusion_engine=fusion_engine,
        alert_manager=alert_manager,
    )

    app = create_app(alert_manager, camera_stream)
    api_thread = threading.Thread(target=run_api, args=(app,), name="APIServer", daemon=True)
    stop_event = threading.Event()

    def shutdown(*_args) -> None:
        logger.info("Shutdown requested. Stopping services.")
        stop_event.set()
        detection_pipeline.stop()
        audio_stream.stop()
        camera_stream.stop()

    signal.signal(signal.SIGINT, shutdown)
    signal.signal(signal.SIGTERM, shutdown)

    audio_stream.start()
    camera_stream.start()
    detection_pipeline.start()
    api_thread.start()

    logger.info("ForestGuardAI is running. Dashboard available at http://%s:%s", settings.api_host, settings.api_port)
    while not stop_event.is_set():
        time.sleep(0.5)


if __name__ == "__main__":
    main()