import { pipeline } from 'https://cdn.jsdelivr.net/npm/@huggingface/transformers@3.0.0';

const status = document.getElementById('status');
const output = document.getElementById('output');
const input = document.getElementById('input');
const submitBtn = document.getElementById('submit');

let generator;

// Initialize model directly in browser
async function init() {
    status.textContent = 'Loading model...';
    generator = await pipeline('text-generation', 'onnx-community/Qwen2.5-0.5B-Instruct');
    status.textContent = 'Model Ready!';
    submitBtn.disabled = false;
}

submitBtn.addEventListener('click', async () => {
    const text = input.value;
    if (!text) return;
    
    status.textContent = 'Analyzing...';
    const messages = [{ role: 'user', content: text }];
    const result = await generator(messages, { max_new_tokens: 200 });
    
    output.textContent = result[0].generated_text.at(-1).content;
    status.textContent = 'Done!';
});

init();