# config.py

import os

# Kafka配置
KAFKA_BROKER = "222.186.10.253:9092"
KAFKA_GROUP_ID_PREFIX = "group"

# Redis配置
REDIS_HOST = "150.158.144.159"
REDIS_PORT = 13003
REDIS_PASSWORD = "Obscura@2024"
MAIN_REDIS_DB = 0
REDIS_API_DB = 2
REDIS_API_USAGE_DB = 3
# 目录配置
UPLOAD_DIR = "/obscura/task/upload"
RESULT_DIR = "/obscura/task/result"

# 确保目录存在
os.makedirs(UPLOAD_DIR, exist_ok=True)
os.makedirs(RESULT_DIR, exist_ok=True)

# 模型配置
YOLO_MODEL_PATH = "/obscura/models/yolov8x.pt"
POSE_MODEL_PATH = "/obscura/models/yolov8x-pose.pt"
QWEN_MODEL_PATH = "/obscura/models/qwen/Qwen2-VL-2B-Instruct"
FALL_MODEL_PATH = "/obscura/models/yolov8n-fall.pt"
FACE_MODEL_PATH = "/obscura/models/yolov8n-face.pt"
MEDIAPIPE_MODEL_PATH = "/obscura/models/face_landmarker.task"    
# COMPARE_MODEL_PATH = "/obscura/models/insightface/insw_r100_glint360k.onnx"
# Ollama配置
OLLAMA_URL = "http://127.0.0.1:11434/api/generate"

# 各个worker的配置
WORKER_CONFIGS = {
    "yolo": {
        "kafka_topic": "yolo",
        "redis_db": 4,
    },
    "pose": {
        "kafka_topic": "pose",
        "redis_db": 5,
    },
    "qwenvl": {
        "kafka_topic": "qwenvl",
        "redis_db": 9,
    },
    "qwenvl_analyze": {
        "kafka_topic": "qwenvl_analyze",
        "redis_db": 32,
    },
    "cpm": {
        "kafka_topic": "cpm",
        "redis_db": 8,
    },
    "cpm_analyze": {
        "kafka_topic": "cpm_analyze",
        "redis_db": 31,
    },
    "fall": {
        "kafka_topic": "fall",
        "redis_db": 6,
    },
    "face": {
        "kafka_topic": "face",
        "redis_db": 7,
    },
    "mediapipe": {
        "kafka_topic": "mediapipe",
        "redis_db": 10,
    },
    "compare": {
        "kafka_topic": "compare",
        "redis_db": 30,
    }      
}

# GPU设置
CUDA_DEVICE_0 = "cuda:0"
CUDA_DEVICE_1 = "cuda:1"