opencode-interrupt-plugin 0.4.34 → 0.4.35

package/dist/clean-text.d.ts

@@ -0,0 +1,6 @@
+export declare function cleanText(raw: string): string;
+export declare function processTranscription(raw: string): Promise<{
+    text: string;
+    cleaned: boolean;
+    polished: boolean;
+}>;

package/dist/clean-text.js

@@ -0,0 +1,112 @@
+/* ------------------------------------------------------------------ */
+/*  Layer 1: Regex text cleaning — always-on, no API needed           */
+/* ------------------------------------------------------------------ */
+const FILLER_PATTERNS = [
+    /\bum+\b/gi,
+    /\buh+\b/gi,
+    /\blike\b/gi,
+    /\byou know\b/gi,
+    /\bi mean\b/gi,
+    /\bsort of\b/gi,
+    /\bkind of\b/gi,
+    /\byeah\b/gi,
+    /\bso basically\b/gi,
+    /\bright\b/gi,
+    /\bokay\b/gi,
+    /\balright\b/gi,
+    /\banyways?\b/gi,
+    /\bactually\b(?=\s+(?:the|a|an|it|i|we|you|they|he|she)\b)/gi,
+];
+const STUTTER_PATTERN = /\b(\w+)(?: \1\b)+/gi;
+const LEADING_FILLER = /^(?:and |so |but |or |then |well |oh )+/i;
+const CONSECUTIVE_SPACES = /\s{2,}/g;
+export function cleanText(raw) {
+    let t = raw.trim();
+    if (!t)
+        return t;
+    // Remove filler words
+    for (const pat of FILLER_PATTERNS) {
+        t = t.replace(pat, '');
+    }
+    // Remove stutters / repeated words
+    t = t.replace(STUTTER_PATTERN, '$1');
+    // Remove leading fillers (false starts at beginning)
+    t = t.replace(LEADING_FILLER, '');
+    // Collapse whitespace
+    t = t.replace(CONSECUTIVE_SPACES, ' ');
+    // Capitalize first letter
+    if (t.length > 0) {
+        t = t[0].toUpperCase() + t.slice(1);
+    }
+    // Ensure ending punctuation
+    if (t.length > 0 && !/[.!?]/.test(t[t.length - 1])) {
+        t += '.';
+    }
+    return t.trim();
+}
+/* ------------------------------------------------------------------ */
+/*  Layers 2+3: LLM polish — uses OPENAI_API_KEY when set              */
+/* ------------------------------------------------------------------ */
+const POLISH_SYSTEM_PROMPT = `You are a voice transcription cleaner. Your job is to take raw voice-to-text output and produce clean, concise text.
+
+Rules:
+1. Remove all filler words (um, uh, like, you know, etc.)
+2. If the speaker corrected themselves mid-sentence, keep ONLY the final version
+3. Remove false starts and abandoned sentences
+4. Fix capitalization and punctuation
+5. Remove repeated words
+6. If the text is a command or request, make it direct and clear
+7. Output ONLY the cleaned text — no explanations, no quotes, no prefixes`;
+async function polishViaOpenAI(raw) {
+    const apiKey = process.env.OPENAI_API_KEY;
+    if (!apiKey)
+        return null;
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), 10000);
+    try {
+        const resp = await fetch('https://api.openai.com/v1/chat/completions', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${apiKey}`,
+            },
+            body: JSON.stringify({
+                model: 'gpt-4o-mini',
+                messages: [
+                    { role: 'system', content: POLISH_SYSTEM_PROMPT },
+                    { role: 'user', content: raw },
+                ],
+                max_tokens: 500,
+                temperature: 0.1,
+            }),
+            signal: controller.signal,
+        });
+        clearTimeout(timeout);
+        if (!resp.ok)
+            return null;
+        const data = await resp.json();
+        const cleaned = data.choices?.[0]?.message?.content?.trim();
+        return cleaned || null;
+    }
+    catch {
+        clearTimeout(timeout);
+        return null;
+    }
+}
+/* ------------------------------------------------------------------ */
+/*  Public pipeline                                                    */
+/* ------------------------------------------------------------------ */
+export async function processTranscription(raw) {
+    if (!raw)
+        return { text: raw, cleaned: false, polished: false };
+    // Layer 1: always on
+    const layer1 = cleanText(raw);
+    let polished = false;
+    // Layers 2+3: LLM polish when API key is set
+    const llmResult = await polishViaOpenAI(layer1);
+    if (llmResult && llmResult !== layer1) {
+        polished = true;
+        return { text: llmResult, cleaned: true, polished: true };
+    }
+    return { text: layer1, cleaned: true, polished: false };
+}

package/dist/index.js

@@ -1,6 +1,7 @@
 import { resolveConfig } from './config.js';
 import { checkLicense } from './license/guard.js';
 import { debug } from './log.js';
+import { processTranscription } from './clean-text.js';
 import { getSessionState, updateSessionState, clearSessionState, } from './store.js';
 import { prepareInjection } from './injector.js';
 import { onTTSStart, onTTSEnd, isTTSTool } from './audio/tts-tracker.js';
@@ -109,6 +110,15 @@ async function transcribeAndSend(sessionID, directory, api, modelPath) {
         catch { /* ignore */ }
         return;
     }
+    const { text: clean, polished } = await processTranscription(text);
+    if (!clean) {
+        api.ui.toast({ variant: "warning", title: "PTT", message: "⚠️ No meaningful text in recording" });
+        try {
+            unlinkSync(RECORDING_FILE);
+        }
+        catch { /* ignore */ }
+        return;
+    }
     try {
         unlinkSync(RECORDING_FILE);
     }
@@ -117,9 +127,9 @@ async function transcribeAndSend(sessionID, directory, api, modelPath) {
         api.ui.toast({ variant: "warning", title: "PTT", message: "⚠️ Open a session first, then type /ptt" });
         return;
     }
-    api.ui.toast({ variant: "info", title: "PTT", message: "✉️ Sending transcript..." });
-    await api.client.session.prompt({ sessionID, directory, parts: [{ type: "text", text }] });
-    const preview = text.length > 80 ? text.slice(0, 77) + "..." : text;
+    api.ui.toast({ variant: "info", title: "PTT", message: polished ? "✨ Sending polished transcript..." : "✉️ Sending transcript..." });
+    await api.client.session.prompt({ sessionID, directory, parts: [{ type: "text", text: clean }] });
+    const preview = clean.length > 80 ? clean.slice(0, 77) + "..." : clean;
     api.ui.toast({ variant: "success", title: "PTT", message: `✅ Sent: "${preview}"` });
 }
 async function transcribeAndSendV1(sessionID, client, modelPath) {
@@ -149,6 +159,15 @@ async function transcribeAndSendV1(sessionID, client, modelPath) {
         catch { /* ignore */ }
         return;
     }
+    const { text: clean, polished } = await processTranscription(text);
+    if (!clean) {
+        await client.tui.showToast({ body: { title: "PTT", message: "⚠️ No meaningful text in recording", variant: "warning", duration: 4000 } });
+        try {
+            unlinkSync(RECORDING_FILE);
+        }
+        catch { /* ignore */ }
+        return;
+    }
     try {
         unlinkSync(RECORDING_FILE);
     }
@@ -157,9 +176,9 @@ async function transcribeAndSendV1(sessionID, client, modelPath) {
         await client.tui.showToast({ body: { title: "PTT", message: "⚠️ Open a session first, then type /ptt", variant: "warning", duration: 5000 } });
         return;
     }
-    await client.tui.showToast({ body: { title: "PTT", message: "✉️ Sending transcript...", variant: "info" } });
-    await client.session.prompt({ path: { id: sessionID }, body: { parts: [{ type: "text", text }] } });
-    const preview = text.length > 80 ? text.slice(0, 77) + "..." : text;
+    await client.tui.showToast({ body: { title: "PTT", message: polished ? "✨ Sending polished transcript..." : "✉️ Sending transcript...", variant: "info" } });
+    await client.session.prompt({ path: { id: sessionID }, body: { parts: [{ type: "text", text: clean }] } });
+    const preview = clean.length > 80 ? clean.slice(0, 77) + "..." : clean;
     await client.tui.showToast({ body: { title: "PTT", message: `✅ Sent: "${preview}"`, variant: "success", duration: 5000 } });
 }
 const TTS_COMMANDS = [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-interrupt-plugin",
-  "version": "0.4.34",
+  "version": "0.4.35",
   "description": "Streaming TTS + voice interruption for OpenCode. Speaks responses as they arrive and detects when you talk over it.",
   "type": "module",
   "main": "./dist/index.js",