Add local JobAlign resume AI

ai.js

@@ -0,0 +1,681 @@
+import {
+    pipeline,
+    env
+} from "https://cdn.jsdelivr.net/npm/@huggingface/transformers@3.8.1";
+
+
+/* =========================================================
+   JOBALIGN LOCAL AI
+   Runs entirely in the user's browser.
+========================================================= */
+
+env.allowLocalModels = false;
+
+const MODEL_ID =
+    "onnx-community/Qwen2.5-0.5B-Instruct";
+
+
+let generator = null;
+
+let loadingPromise = null;
+
+
+/* =========================================================
+   STATUS HELPER
+========================================================= */
+
+function setAIStatus(message, type = "normal") {
+
+    const status =
+        document.getElementById(
+            "generationStatus"
+        );
+
+
+    if (!status) {
+        return;
+    }
+
+
+    let className = "";
+
+
+    if (type === "success") {
+
+        className =
+            "success-text";
+
+    } else if (type === "error") {
+
+        className =
+            "danger-text";
+
+    } else if (type === "warning") {
+
+        className =
+            "warning-text";
+    }
+
+
+    status.innerHTML = `
+
+        <div class="status-dot"></div>
+
+        <span class="${className}">
+            ${escapeHTML(message)}
+        </span>
+    `;
+}
+
+
+/* =========================================================
+   LOAD MODEL
+========================================================= */
+
+async function loadModel() {
+
+    if (generator) {
+
+        return generator;
+    }
+
+
+    if (loadingPromise) {
+
+        return loadingPromise;
+    }
+
+
+    loadingPromise =
+        (async function () {
+
+            const hasWebGPU =
+                typeof navigator !==
+                    "undefined" &&
+                "gpu" in navigator;
+
+
+            setAIStatus(
+                hasWebGPU
+                    ? "Loading local AI model with WebGPU..."
+                    : "Loading local AI model in CPU mode..."
+            );
+
+
+            const options = {
+
+                dtype: "q4",
+
+                progress_callback:
+                    function (progress) {
+
+                        updateProgress(
+                            progress
+                        );
+                    }
+            };
+
+
+            if (hasWebGPU) {
+
+                options.device =
+                    "webgpu";
+            }
+
+
+            try {
+
+                generator =
+                    await pipeline(
+                        "text-generation",
+                        MODEL_ID,
+                        options
+                    );
+
+
+                setAIStatus(
+                    "Local AI model ready",
+                    "success"
+                );
+
+
+                return generator;
+
+
+            } catch (error) {
+
+                console.error(
+                    "WebGPU/model load failed:",
+                    error
+                );
+
+
+                /*
+                 * If WebGPU failed,
+                 * retry using browser CPU/WASM.
+                 */
+
+                if (hasWebGPU) {
+
+                    setAIStatus(
+                        "WebGPU unavailable. Retrying with CPU..."
+                    );
+
+
+                    generator =
+                        await pipeline(
+                            "text-generation",
+                            MODEL_ID,
+                            {
+                                dtype: "q4",
+
+                                progress_callback:
+                                    function (progress) {
+
+                                        updateProgress(
+                                            progress
+                                        );
+                                    }
+                            }
+                        );
+
+
+                    setAIStatus(
+                        "Local AI model ready in CPU mode",
+                        "success"
+                    );
+
+
+                    return generator;
+                }
+
+
+                throw error;
+            }
+
+        })();
+
+
+    try {
+
+        return await loadingPromise;
+
+    } catch (error) {
+
+        loadingPromise =
+            null;
+
+
+        setAIStatus(
+            "AI model could not load: " +
+            error.message,
+            "error"
+        );
+
+
+        throw error;
+    }
+}
+
+
+/* =========================================================
+   DOWNLOAD PROGRESS
+========================================================= */
+
+function updateProgress(progress) {
+
+    if (!progress) {
+        return;
+    }
+
+
+    if (
+        progress.status ===
+            "progress" &&
+        typeof progress.progress ===
+            "number"
+    ) {
+
+        const percent =
+            Math.round(
+                progress.progress
+            );
+
+
+        setAIStatus(
+            "Downloading local AI model... " +
+            percent +
+            "%"
+        );
+
+
+        return;
+    }
+
+
+    if (
+        progress.status ===
+        "ready"
+    ) {
+
+        setAIStatus(
+            "AI model downloaded. Preparing..."
+        );
+    }
+}
+
+
+/* =========================================================
+   BUILD THE RESUME PROMPT
+========================================================= */
+
+function buildResumeMessages({
+    resume,
+    jobDescription,
+    jobTitle,
+    company,
+    location
+}) {
+
+    /*
+     * Keep inputs within a practical browser-model size.
+     */
+
+    const cleanResume =
+        truncateText(
+            resume,
+            18000
+        );
+
+
+    const cleanJD =
+        truncateText(
+            jobDescription,
+            14000
+        );
+
+
+    const systemPrompt = `
+You are JobAlign, a technical resume alignment assistant.
+
+Your job is to rewrite a candidate's existing resume for one specific target job.
+
+CORE RULES:
+
+1. Preserve the candidate's employers, employment dates, education and factual career history.
+
+2. Strongly emphasize experience that matches the target job.
+
+3. Reorder and rewrite bullets so the target role becomes the natural interpretation of the candidate's background.
+
+4. Use important exact technical nouns from the job description naturally for ATS alignment.
+
+5. Do NOT copy sentences from the job description.
+
+6. Do NOT invent employers, projects, technologies, certifications, degrees, job titles, production systems or accomplishments that are unsupported by the master resume.
+
+7. Do NOT invent exact metrics. Preserve useful metrics already present in the resume.
+
+8. Use technically credible practitioner language. Avoid generic AI-style phrases such as:
+results-driven,
+proven track record,
+leveraged,
+dynamic professional,
+passionate professional.
+
+9. Prefer concrete verbs such as:
+Owned,
+Built,
+Designed,
+Implemented,
+Automated,
+Migrated,
+Tuned,
+Optimized,
+Diagnosed,
+Integrated,
+Delivered,
+Reduced,
+Resolved.
+
+10. Keep the resume ATS-safe:
+plain text,
+standard headings,
+no tables,
+no columns,
+no icons,
+no decorative symbols.
+
+11. Keep the summary concise.
+
+12. Put the most important job-specific technologies in context inside EXPERIENCE, not only SKILLS.
+
+13. Recent and highly relevant experience should receive the strongest detail.
+
+14. Keep projects only when they materially strengthen the target application.
+
+15. The final resume should sound like an experienced human practitioner, not like the job description was pasted into the resume.
+
+OUTPUT ONLY THE COMPLETE REWRITTEN RESUME.
+
+Use these headings where applicable:
+
+SUMMARY
+SKILLS
+EXPERIENCE
+PROJECTS
+EDUCATION
+CERTIFICATIONS
+
+Do not include commentary before or after the resume.
+`.trim();
+
+
+    const userPrompt = `
+TARGET JOB
+
+Title: ${jobTitle || "Not provided"}
+Company: ${company || "Not provided"}
+Location: ${location || "Not provided"}
+
+TARGET JOB DESCRIPTION
+
+${cleanJD}
+
+
+MASTER RESUME
+
+${cleanResume}
+
+
+Rewrite the complete master resume for this specific position.
+
+Prioritize recruiter relevance, ATS keyword coverage, technical credibility and natural human wording.
+
+Return only the finished copy-paste-ready resume.
+`.trim();
+
+
+    return [
+
+        {
+            role: "system",
+            content:
+                systemPrompt
+        },
+
+        {
+            role: "user",
+            content:
+                userPrompt
+        }
+
+    ];
+}
+
+
+/* =========================================================
+   GENERATE RESUME
+========================================================= */
+
+async function generateResume({
+    resume,
+    jobDescription,
+    jobTitle,
+    company,
+    location
+}) {
+
+    if (
+        !resume ||
+        resume.trim().length < 100
+    ) {
+
+        throw new Error(
+            "Master resume is missing."
+        );
+    }
+
+
+    if (
+        !jobDescription ||
+        jobDescription
+            .trim()
+            .length < 100
+    ) {
+
+        throw new Error(
+            "Job description is missing."
+        );
+    }
+
+
+    const model =
+        await loadModel();
+
+
+    setAIStatus(
+        "Rewriting resume locally..."
+    );
+
+
+    const messages =
+        buildResumeMessages({
+            resume,
+            jobDescription,
+            jobTitle,
+            company,
+            location
+        });
+
+
+    const output =
+        await model(
+            messages,
+            {
+                max_new_tokens:
+                    1800,
+
+                do_sample:
+                    false,
+
+                repetition_penalty:
+                    1.08
+            }
+        );
+
+
+    let generatedText = "";
+
+
+    try {
+
+        const generated =
+            output[0]
+                .generated_text;
+
+
+        if (
+            Array.isArray(
+                generated
+            )
+        ) {
+
+            const lastMessage =
+                generated[
+                    generated.length -
+                    1
+                ];
+
+
+            generatedText =
+                lastMessage.content ||
+                "";
+
+        } else {
+
+            generatedText =
+                String(
+                    generated || ""
+                );
+        }
+
+
+    } catch (error) {
+
+        console.error(
+            "Could not parse model output:",
+            error
+        );
+
+
+        throw new Error(
+            "The AI returned an unreadable response."
+        );
+    }
+
+
+    generatedText =
+        cleanGeneratedResume(
+            generatedText
+        );
+
+
+    if (
+        generatedText.length <
+        200
+    ) {
+
+        throw new Error(
+            "The AI response was too short. Please try again."
+        );
+    }
+
+
+    setAIStatus(
+        "Targeted resume generated",
+        "success"
+    );
+
+
+    return generatedText;
+}
+
+
+/* =========================================================
+   CLEAN OUTPUT
+========================================================= */
+
+function cleanGeneratedResume(text) {
+
+    let result =
+        String(
+            text || ""
+        )
+            .trim();
+
+
+    result =
+        result.replace(
+            /^```(?:text|markdown)?/i,
+            ""
+        );
+
+
+    result =
+        result.replace(
+            /```$/i,
+            ""
+        );
+
+
+    result =
+        result.replace(
+            /^Here(?:'s| is).*?resume[:\s]*/i,
+            ""
+        );
+
+
+    return result.trim();
+}
+
+
+/* =========================================================
+   TRUNCATE SAFELY
+========================================================= */
+
+function truncateText(
+    text,
+    maxCharacters
+) {
+
+    const value =
+        String(
+            text || ""
+        )
+            .trim();
+
+
+    if (
+        value.length <=
+        maxCharacters
+    ) {
+
+        return value;
+    }
+
+
+    return (
+        value.slice(
+            0,
+            maxCharacters
+        ) +
+        "\n\n[Input shortened for local browser processing.]"
+    );
+}
+
+
+/* =========================================================
+   HTML SAFETY
+========================================================= */
+
+function escapeHTML(value) {
+
+    const element =
+        document.createElement(
+            "div"
+        );
+
+
+    element.textContent =
+        String(
+            value || ""
+        );
+
+
+    return element.innerHTML;
+}
+
+
+/* =========================================================
+   EXPOSE AI TO APP.JS
+========================================================= */
+
+window.JobAlignAI = {
+
+    generateResume:
+        generateResume,
+
+    loadModel:
+        loadModel,
+
+    model:
+        MODEL_ID
+};
+
+
+window.dispatchEvent(
+    new CustomEvent(
+        "jobalign-ai-ready"
+    )
+);