from config import LING_1T, LING_FLASH_2_0, RING_1T, RING_FLASH_2_0, LING_MINI_2_0, RING_MINI_2_0, get_model_display_name
from i18n import get_text

def get_recommended_inputs(lang: str):
    """
    Generates the list of recommended inputs based on the selected language.
    """

    result = [
        {
            "task": get_text("rec_creative_writing_task", lang),
            "model": get_model_display_name(LING_1T),
            "system_prompt": get_text("rec_creative_writing_system_prompt", lang),
            "user_message": get_text("rec_creative_writing_user_message", lang),
            "temperature": 0.8,
        },
        {
            "task": get_text("rec_code_generation_task", lang),
            "model": get_model_display_name(RING_1T),
            "system_prompt": get_text("rec_code_generation_system_prompt", lang),
            "user_message": get_text("rec_code_generation_user_message", lang),
            "temperature": 0.2,
        },
        {
            "task": get_text("rec_email_drafting_task", lang),
            "model": get_model_display_name(LING_FLASH_2_0),
            "system_prompt": get_text("rec_email_drafting_system_prompt", lang),
            "user_message": get_text("rec_email_drafting_user_message", lang),
            "temperature": 0.7,
        },
        {
            "task": get_text("rec_study_plan_task", lang),
            "model": get_model_display_name(LING_MINI_2_0),
            "system_prompt": get_text("rec_study_plan_system_prompt", lang),
            "user_message": get_text("rec_study_plan_user_message", lang),
            "temperature": 0.6,
        },
        {
            "task": get_text("rec_role_play_task", lang),
            "model": get_model_display_name(RING_FLASH_2_0),
            "system_prompt": get_text("rec_role_play_system_prompt", lang),
            "user_message": get_text("rec_role_play_user_message", lang),
            "temperature": 0.9,
        },
        {
            "task": get_text("rec_tech_qa_task", lang),
            "model": get_model_display_name(RING_MINI_2_0),
            "system_prompt": get_text("rec_tech_qa_system_prompt", lang),
            "user_message": get_text("rec_tech_qa_user_message", lang),
            "temperature": 0.4,
        }
    ]
    return result;