from base64 import b64decode
from cryptography.fernet import Fernet
from dotenv import load_dotenv
import os
import prompts_enc

load_dotenv()

def get_prompt(name: str) -> str:
    """Возвращает раскодированный промпт по имени."""
    key = os.getenv("PROMPT_KEY")
    if not key:
        raise RuntimeError("PROMPT_KEY отсутствует в переменных окружения")

    if not hasattr(prompts_enc, name):
        raise KeyError(f"Промпт '{name}' не найден")

    fernet = Fernet(key.encode())
    encoded = getattr(prompts_enc, name)
    decoded = fernet.decrypt(b64decode(encoded)).decode()
    return decoded


if __name__ == "__main__":
    print(get_prompt("PROMPT_STYLE"))