Update script.js

script.js

@@ -1,156 +1,40 @@
-const keywords = {
+document.getElementById("generateBtn").onclick = async function(){
 
-    image: [
-        "image",
-        "draw",
-        "paint",
-        "picture",
-        "photo",
-        "art",
-        "illustration",
-        "generate image"
-    ],
+    const prompt = document.getElementById("prompt").value;
 
-    translation: [
-        "translate",
-        "translation",
-        "convert language"
-    ],
+    const task = detectTask(prompt);
 
-    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;
-
-            }
-
-        }
+    let modelName = MODELS[task];
 
+    if(!modelName){
+        modelName = MODELS.text;
     }
 
-    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";
+    document.getElementById("model").innerHTML = modelName;
 
-            output="🖼 Image Generation Model Selected";
+    document.getElementById("output").innerHTML = "Generating...";
 
-            break;
+    const result = await queryHF(modelName,prompt);
 
-        case "translation":
+    if(Array.isArray(result)){
 
-            model="NLLB-200";
+        if(result[0].generated_text){
 
-            output="🌍 Translation Model Selected";
+            document.getElementById("output").innerHTML =
+                result[0].generated_text;
 
-            break;
+        }else{
 
-        case "summary":
+            document.getElementById("output").innerHTML =
+                JSON.stringify(result);
 
-            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";
+    }else{
 
-            output="💬 Text Generation Model Selected";
+        document.getElementById("output").innerHTML =
+            JSON.stringify(result,null,2);
 
     }
 
-    document.getElementById("model").innerHTML=model;
-
-    document.getElementById("output").innerHTML=output;
-
 }