from prometheus_client import Gauge, Counter

APP_UP = Gauge(
    "app_up",
    "Indicates if the application is running (1) or shutting down (0)",
    ["pod_id"],
)

CORE_LABELS = ["pod_id", "model_name", "pipeline_name"]

PREDICTIONS_WRITTEN_COUNT = Counter(
    "laborious_predictions_written_count",
    "Number of predictions written to the database table predictions",
    CORE_LABELS,
)

PREDICTION_CONFIDENCE_MONITOR = Gauge(
    "laborious_prediction_confidence_monitor",
    "Current confidence of each prediction",
    CORE_LABELS,
)

PREDICTION_RESPONSE_TIME_MONITOR = Gauge(
    "laborious_prediction_response_time_monitor",
    "Current response time of each prediction",
    CORE_LABELS,
)