const keywords = {

    image: [
        "image",
        "draw",
        "paint",
        "picture",
        "photo",
        "art",
        "illustration",
        "generate image"
    ],

    translation: [
        "translate",
        "translation",
        "convert language"
    ],

    summary: [
        "summarize",
        "summary",
        "shorten"
    ],

    code: [
        "code",
        "python",
        "java",
        "cpp",
        "c++",
        "program"
    ],

    speech: [
        "speech",
        "voice",
        "audio"
    ],

    ocr: [
        "ocr",
        "scan",
        "extract text"
    ],

    caption: [
        "caption",
        "describe image"
    ]

};

function detectModel(prompt){

    prompt = prompt.toLowerCase();

    for(let task in keywords){

        for(let word of keywords[task]){

            if(prompt.includes(word)){

                return task;

            }

        }

    }

    return "text";

}

document.getElementById("generateBtn").onclick=function(){

    const prompt=document.getElementById("prompt").value;

    const task=detectModel(prompt);

    let model="";

    let output="";

    switch(task){

        case "image":

            model="Stable Diffusion XL";

            output="🖼 Image Generation Model Selected";

            break;

        case "translation":

            model="NLLB-200";

            output="🌍 Translation Model Selected";

            break;

        case "summary":

            model="BART";

            output="📄 Summarization Model Selected";

            break;

        case "code":

            model="Code Llama";

            output="💻 Code Generation Model Selected";

            break;

        case "speech":

            model="Whisper";

            output="🎤 Speech Recognition Model Selected";

            break;

        case "ocr":

            model="TrOCR";

            output="📷 OCR Model Selected";

            break;

        case "caption":

            model="BLIP";

            output="🖼 Image Captioning Model Selected";

            break;

        default:

            model="Gemma 3";

            output="💬 Text Generation Model Selected";

    }

    document.getElementById("model").innerHTML=model;

    document.getElementById("output").innerHTML=output;

}