Update script.js

script.js

@@ -1,9 +1,156 @@
+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(){
 
-let prompt=document.getElementById("prompt").value;
+    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="Analyzing...";
+    document.getElementById("model").innerHTML=model;
 
-document.getElementById("output").innerHTML=prompt;
+    document.getElementById("output").innerHTML=output;
 
 }