"""
Model client and embedding factories.
Read API keys from environment variables.
"""
import os
from typing import Optional
from ..core import config as core_config

genai_client = None
embeddings = None
qdrant_client = None


def init_genai_client(api_key: Optional[str] = None):
    global genai_client
    try:
        from google import genai
        if api_key is None:
            cfg = core_config.load_config()
            api_key = cfg.get("GOOGLE_API_KEY")
        genai_client = genai.Client(api_key=api_key) if api_key else None
    except Exception:
        genai_client = None
    return genai_client


def init_embeddings(voyage_api_key: Optional[str] = None):
    global embeddings
    try:
        from langchain_voyageai import VoyageAIEmbeddings
        if voyage_api_key is None:
            cfg = core_config.load_config()
            voyage_api_key = cfg.get("VOYAGE_API_KEY")
        if voyage_api_key:
            os.environ.setdefault("VOYAGE_API_KEY", voyage_api_key)
            embeddings = VoyageAIEmbeddings(model="voyage-3-large")
            return embeddings
    except Exception:
        pass
    return None


def init_qdrant_client(url: Optional[str] = None, api_key: Optional[str] = None):
    global qdrant_client
    try:
        from qdrant_client import QdrantClient
        if url is None or api_key is None:
            cfg = core_config.load_config()
            if url is None:
                url = cfg.get("QDRANT_URL")
            if api_key is None:
                api_key = cfg.get("QDRANT_API_KEY")
        if url:
            qdrant_client = QdrantClient(url=url, api_key=api_key, prefer_grpc=False)
            return qdrant_client
    except Exception:
        qdrant_client = None
    return None


class ModelClient:
    """Simple wrapper that exposes current clients as properties.

    The module keeps module-level references (genai_client, embeddings, qdrant_client)
    and this wrapper exposes them dynamically so other modules can import
    `model_client` and access attributes like `model_client.genai_client`.
    """

    @property
    def genai_client(self):
        return genai_client

    @property
    def embeddings(self):
        return embeddings

    @property
    def qdrant_client(self):
        return qdrant_client

    def init_all(self):
        init_genai_client()
        init_embeddings()
        init_qdrant_client()


model_client = ModelClient()