from langchain import PromptTemplate

TALKING_HEAD_WIDTH = 192

# Console to variable
# Pertains to question answering functionality

TOOLS_LIST = ['serpapi', 'wolfram-alpha', 'pal-math',
              'pal-colored-objects','google-search','news-api','tmdb-api','open-meteo-api']
TOOLS_DEFAULT_LIST = ['serpapi']
BUG_FOUND_MSG = "Congratulations, you've found a bug in this application!"
# AUTH_ERR_MSG = "Please paste your OpenAI key from openai.com to use this application. It is not necessary to hit a button or key after pasting it."
AUTHORIZATION_ERROR_MESSAGE = "Please paste your OpenAI key from openai.com to use this application. "
MAX_TOKENS = 512

LOOPING_TALKING_HEAD_VIDEO_PATH = "assets/videos/Masahiro.mp4"

MAX_TALKING_HEAD_TEXT_LENGTH = 1550

# Pertains to Express-inator functionality
NUM_WORDS_DEFAULT = 0
MAX_WORDS = 4000
FORMALITY_DEFAULT = "N/A"
TEMPERATURE_DEFAULT = 0
EMOTION_DEFAULT = "N/A"
LANG_LEVEL_DEFAULT = "N/A"
TRANSLATE_TO_DEFAULT = "N/A"
LITERARY_STYLE_DEFAULT = "N/A"
PROMPT_TEMPLATE = PromptTemplate(
    input_variables=["original_words", "num_words", "formality", "emotions", "lang_level", "translate_to",
                     "literary_style"],
    template="Restate {num_words}{formality}{emotions}{lang_level}{translate_to}{literary_style}the following: \n{original_words}\n",
)