from config import LING_1T, LING_FLASH_2_0, get_model_display_name
from i18n import get_text

def get_code_examples(lang: str):
    """
    Returns the recommended initial inputs for the code generation tab,
    translated to the specified language.
    """
    return [
        {
            "task": get_text("code_example_knowledge_task", lang),
            "model": get_model_display_name(LING_1T),
            "user_prompt": get_text("code_example_knowledge_prompt", lang)
        },
        {
            "task": get_text("code_example_industry_task", lang),
            "model": get_model_display_name(LING_1T),
            "user_prompt": get_text("code_example_industry_prompt", lang)
        },
        {
            "task": get_text("code_example_hello_world_task", lang),
            "model": get_model_display_name(LING_FLASH_2_0),
            "user_prompt": get_text("code_example_hello_world_prompt", lang)
        },
        {
            "task": get_text("code_example_fireworks_task", lang),
            "model": get_model_display_name(LING_1T),
            "user_prompt": get_text("code_example_fireworks_prompt", lang)
        },
        {
            "task": get_text("code_example_streaming_task", lang),
            "model": get_model_display_name(LING_1T),
            "user_prompt": get_text("code_example_streaming_prompt", lang)
        },
        {
            "task": get_text("code_example_error_test_task", lang),
            "model": get_model_display_name(LING_FLASH_2_0),
            "user_prompt": get_text("code_example_error_test_prompt", lang)
        },
        {
            "task": get_text("code_example_react_error_task", lang),
            "model": get_model_display_name(LING_FLASH_2_0),
            "user_prompt": get_text("code_example_react_error_prompt", lang)
        }
    ]

# For backward compatibility if needed, but we should prefer the function
CODE_EXAMPLES = get_code_examples("zh")