Update index.js

index.js

@@ -1,6 +1,8 @@
+// index.js
+
 import { pipeline } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers';
 
-// Front-end copy of the available models for dropdown
+// Front-end copy of the Python AVAILABLE_MODELS list for dropdown population:
 const AVAILABLE_MODELS = [
   { name: "Moonshot Kimi-K2", id: "moonshotai/Kimi-K2-Instruct" },
   { name: "DeepSeek V3", id: "deepseek-ai/DeepSeek-V3-0324" },
@@ -8,78 +10,68 @@ const AVAILABLE_MODELS = [
   { name: "ERNIE-4.5-VL", id: "baidu/ERNIE-4.5-VL-424B-A47B-Base-PT" },
   { name: "MiniMax M1", id: "MiniMaxAI/MiniMax-M1-80k" },
   { name: "Qwen3-235B-A22B", id: "Qwen/Qwen3-235B-A22B" },
+  { name: "Qwen3-32B", id: "Qwen/Qwen3-32B" },                    // newly added
   { name: "SmolLM3-3B", id: "HuggingFaceTB/SmolLM3-3B" },
   { name: "GLM-4.1V-9B-Thinking", id: "THUDM/GLM-4.1V-9B-Thinking" },
+  { name: "LLaMA-2-70B-Chat", id: "meta-llama/Llama-2-70b-chat" },  // powerful addition
+  { name: "Mistral-7B-Instruct", id: "mistralai/Mistral-7B-Instruct" }, // powerful addition
   { name: "Qwen3-235B-A22B-Instruct-2507", id: "Qwen/Qwen3-235B-A22B-Instruct-2507" },
-  { name: "Qwen3-Coder-480B-A35B", id: "Qwen/Qwen3-Coder-480B-A35B-Instruct" },
-  { name: "OpenAI GPT-4", id: "openai/gpt-4" },
-  { name: "Gemini Pro", id: "gemini/pro" },
-  { name: "Fireworks AI", id: "fireworks-ai/fireworks-v1" }
+  { name: "Qwen3-Coder-480B-A35B", id: "Qwen/Qwen3-Coder-480B-A35B-Instruct" }
 ];
 
-// DOM Elements
 const modelSelect   = document.getElementById('modelSelect');
-const promptInput   = document.getElementById('promptInput');
-const generateBtn   = document.getElementById('generateBtn');
-const codeOutput    = document.getElementById('codeOutput');
+const analyzeForm   = document.getElementById('analyze-form');
+const inputText     = document.getElementById('inputText');
+const analyzeButton = document.getElementById('analyzeButton');
+const resultSection = document.getElementById('result');
 
-let codePipeline = null;
+let sentimentPipeline = null;
 
-// Populate model dropdown
-function populateModels() {
-  AVAILABLE_MODELS.forEach(model => {
-    const option = document.createElement('option');
-    option.value = model.id;
-    option.textContent = model.name;
-    modelSelect.appendChild(option);
-  });
-}
+// Populate the model dropdown
+AVAILABLE_MODELS.forEach(model => {
+  const option = document.createElement('option');
+  option.value = model.id;
+  option.textContent = model.name;
+  modelSelect.appendChild(option);
+});
 
-// Initialize pipeline for code generation
+// Initialize pipeline for the selected model
 async function initPipeline(modelId) {
-  generateBtn.disabled = true;
-  generateBtn.textContent = 'Loading model…';
-  codePipeline = await pipeline('text-generation', modelId, { trustRemoteCode: true });
-  generateBtn.textContent = 'Generate Code';
-  generateBtn.disabled = false;
+  analyzeButton.disabled = true;
+  analyzeButton.textContent = 'Loading model…';
+  sentimentPipeline = await pipeline('sentiment-analysis', modelId);
+  analyzeButton.textContent = 'Analyze Sentiment';
+  analyzeButton.disabled = false;
 }
 
-// Handle generation
-async function handleGenerate() {
-  const prompt = promptInput.value.trim();
-  if (!prompt) return;
+// On page load: initialize with the first model
+initPipeline(modelSelect.value);
+
+// When user changes model: reinitialize pipeline
+modelSelect.addEventListener('change', () => {
+  resultSection.textContent = '';
+  initPipeline(modelSelect.value);
+});
+
+// Handle form submission
+analyzeForm.addEventListener('submit', async (event) => {
+  event.preventDefault();
+  const text = inputText.value.trim();
+  if (!text) return;
 
-  generateBtn.disabled = true;
-  generateBtn.textContent = 'Generating…';
-  codeOutput.textContent = '';
+  analyzeButton.disabled = true;
+  analyzeButton.textContent = 'Analyzing…';
+  resultSection.textContent = '';
 
   try {
-    const outputs = await codePipeline(prompt, { max_new_tokens: 512 });
-    const generated = Array.isArray(outputs)
-      ? outputs.map(o => o.generated_text).join('')
-      : outputs.generated_text;
-    codeOutput.textContent = generated;
+    const [output] = await sentimentPipeline(text);
+    const { label, score } = output;
+    resultSection.textContent = `Sentiment: ${label} (Confidence: ${(score * 100).toFixed(2)}%)`;
   } catch (err) {
     console.error(err);
-    codeOutput.textContent = 'Error generating code.';
+    resultSection.textContent = 'Error analyzing sentiment.';
   }
 
-  generateBtn.textContent = 'Generate Code';
-  generateBtn.disabled = false;
-}
-
-// Event bindings
-window.addEventListener('DOMContentLoaded', async () => {
-  populateModels();
-  await initPipeline(modelSelect.value);
-});
-
-modelSelect.addEventListener('change', () => {
-  codeOutput.textContent = '';
-  initPipeline(modelSelect.value);
+  analyzeButton.textContent = 'Analyze Sentiment';
+  analyzeButton.disabled = false;
 });
-
-generateBtn.addEventListener('click', handleGenerate);
-
-// Expose functions (for debugging)
-window._codeApp = { initPipeline, handleGenerate };