| 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 | |