import os
from datetime import datetime
from pathlib import Path
from dotenv import load_dotenv

# Завантаження змінних середовища
load_dotenv()

# Максимальна кількість токенів для різних типів промптів
MAX_TOKENS = {
    "risk_assessment": 500,  # Для оцінки ризику
    "classification": 1000,  # Для класифікації ризику
    "response": 2000        # Для фінального повідомлення
}

# Температура для різних типів промптів
TEMPERATURE = {
    "risk_assessment": 0,   # Чітка оцінка
    "classification": 0,    # Чітка класифікація
    "response": 0          # Чітка відповідь
}

# Базові шляхи
BASE_DIR = Path(__file__).parent.parent
PROMPTS_DIR = BASE_DIR / "prompts"
DATA_DIR = BASE_DIR / "data"
LOGS_DIR = BASE_DIR / "logs"

# Конфігурація API
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
AZURE_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
AZURE_ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")

# Конфігурація логування
LOG_FILE = LOGS_DIR / f"api_calls_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"

# Конфігурація моделей
AVAILABLE_MODELS = {
    "anthropic": "claude-3-5-sonnet-20241022",
    "openai": "gpt-4o-2024-11-20",
    "azure": "gpt-4o"
}

# Налаштування затримок між API викликами
MIN_DELAY = 1
MAX_DELAY = 3

# Максимальна кількість спроб для API викликів
MAX_RETRIES = 3

# Конфігурація фільтрів для даних пацієнта
# Встановіть в None щоб відключити відповідну фільтрацію
EXCLUDE_SECTIONS = ["RECENT VITALS", "ACUTE VITALS & MEASUREMENTS"]
EXCLUDE_START_TIME = None  # "2025-01-01T00:00:00Z"
EXCLUDE_END_TIME = None    # "2025-01-31T23:59:59Z"