from google import genai
from google.genai import types
from process.sys_prompt import GENERAL_PROMPT, NEWS_PROMPT, PHILO_PROMPT


NARRATIVE_PROMPT = ""
POEM_PROMPT = ""


def get_interpretation(
    genre: str, api_key: str, text: str, learn_language: str, prof_language: str
) -> str:
    if not api_key:
        return "Error: Gemini API Key not found."
    if not text:
        return "Error: text not found."

    try:
        client = genai.Client(api_key=api_key)
    except Exception as e:
        return f"ERROR: {str(e)}"

    lang_map = {
        "AR": "Arabic",
        "DE": "German",
        "ES": "Spanish",
        "EN": "English",
        "FR": "French",
        "IT": "Italian",
        "JA": "Japanese",
        "RU": "Russian",
        "ZH": "Chinese",
    }
    learn_lang = lang_map.get(learn_language.upper(), "English")
    prof_lang = lang_map.get(prof_language.upper(), "English")
    genres = {
        "general": GENERAL_PROMPT,
        "news": NEWS_PROMPT,
        "narrative": NARRATIVE_PROMPT,
        "poem": POEM_PROMPT,
        "philosophy": PHILO_PROMPT,
    }
    if genre.lower() in ["general", "news", "philosophy"]:
        sys_prompt = (
            genres[genre.lower()]
            .replace("[LEARN_LANGUAGE]", learn_lang)
            .replace("[PROF_LANGUAGE]", prof_lang)
        )

    response = client.models.generate_content(
        model="gemini-2.5-pro",
        config=types.GenerateContentConfig(
            system_instruction=sys_prompt,
            temperature=0.3,
        ),
        contents=[text],
    )
    return response.text