import os
import logging
from app.config import settings
from app.ai.prompts import get_prompt

logger = logging.getLogger(__name__)


class LLMClient:
    def __init__(self):
        self.api_key = settings.GROQ_API_KEY
        self.fallback_key = settings.GOOGLE_API_KEY
        self.model = settings.DEFAULT_MODEL
        self.fallback_model = "gemini/gemini-2.0-flash"
        self.pro_model = settings.PRO_MODEL

    def has_api_key(self) -> bool:
        return bool(self.api_key) or bool(self.fallback_key)

    def get_providers(self) -> list[dict]:
        providers = []
        if self.api_key:
            providers.append({"model": self.model, "api_key": self.api_key, "name": "groq"})
        if self.fallback_key:
            providers.append({"model": self.fallback_model, "api_key": self.fallback_key, "name": "gemini"})
        return providers

    def build_prompt(self, pipeline_type: str, context: dict) -> str:
        template = get_prompt(pipeline_type)
        blast = context.get("blast", {})
        top = blast.get("top_hit", {})
        uniprot = context.get("uniprot", {}) or {}
        af = context.get("alphafold", {}) or {}

        return template.format(
            blast_count=blast.get("count", 0),
            top_hit_accession=top.get("accession", "N/A"),
            top_hit_description=top.get("description", "N/A"),
            top_hit_evalue=top.get("evalue", "N/A"),
            top_hit_identity_pct=top.get("identity_pct", "N/A"),
            top_hit_bit_score=top.get("bit_score", "N/A"),
            uniprot_name=uniprot.get("full_name", "N/A"),
            uniprot_organism=uniprot.get("organism", "N/A"),
            uniprot_genes=", ".join(uniprot.get("gene_names", []) or []) or "N/A",
            uniprot_functions="; ".join(uniprot.get("functions", []) or []) or "N/A",
            uniprot_locations="; ".join(uniprot.get("subcellular_locations", []) or []) or "N/A",
            uniprot_keywords=", ".join(uniprot.get("keywords", []) or []) or "N/A",
            uniprot_go_terms=", ".join(uniprot.get("go_terms", []) or []) or "N/A",
            uniprot_features="; ".join(
                f"{f.get('type', '')}: {f.get('description', '')}" for f in (uniprot.get("features", []) or [])
            ) or "N/A",
            alphafold_available="Yes" if af.get("structure_available") else "No",
            alphafold_confidence=af.get("confidence", "N/A"),
        )


llm_client = LLMClient()