Upload 4 files

app.js

@@ -123,6 +123,7 @@ const midiBpmModeGroup = document.querySelector("#midiBpmModeGroup");
 const midiBpmModeButtons = midiBpmModeGroup ? midiBpmModeGroup.querySelectorAll(".toggle-btn") : [];
 
 let midiSeedActive = false;
+let bachSeedActive = false;
 let midiBpmMode = "lock";
 let midiTokens = [];
 let midiStartBar = 0;
@@ -581,8 +582,6 @@ const pianoSamples = {
   C8: "C8.mp3",
 };
 
-const abcNotePattern = /^(?:\^\^|__|\^|_|=)?[A-Ga-g][,']*$/;
-
 // Status Handler
 function setStatus(message, state = "idle") {
   if (appStatusEl) appStatusEl.textContent = message;
@@ -781,7 +780,8 @@ async function start() {
       midiSeedActive: midiSeedActive,
       midiTokens: midiSeedActive ? midiTokens : [],
       midiStartBar: midiStartBar,
-      midiBpmMode: midiBpmMode
+      midiBpmMode: midiBpmMode,
+      bachSeedActive: bachSeedActive
     });
   });
   
@@ -1487,9 +1487,15 @@ toggleButtons.forEach(btn => {
     if (value === "midi") {
       midiSeedActive = true;
       midiSeedControls.style.display = "block";
+      bachSeedActive = false;
+    } else if (value === "bach") {
+      midiSeedActive = false;
+      midiSeedControls.style.display = "none";
+      bachSeedActive = true;
     } else {
       midiSeedActive = false;
       midiSeedControls.style.display = "none";
+      bachSeedActive = false;
     }
   });
 });

index.html

@@ -121,6 +121,7 @@
                 <div class="toggle-group" id="seedSourceGroup">
                   <button type="button" class="toggle-btn active" data-value="default">Start from Blank</button>
                   <button type="button" class="toggle-btn" data-value="midi">Start from MIDI</button>
+                  <button type="button" class="toggle-btn" data-value="bach">Bach Seed Test</button>
                 </div>
               </div>
             </div>

worker.js

@@ -10,7 +10,6 @@ let session = null;
 let stoi = null;
 let itos = null;
 let vocabSize = 0;
-let activeTokenizer = "abc";
 
 let h = null;
 let c = null;
@@ -29,6 +28,7 @@ let midiBpmMode = "lock";
 
 // Seeding settings
 let midiSeedActive = false;
+let bachSeedActive = false;
 let midiTokens = [];
 let midiStartBar = 0;
 let isWarmingUp = false;
@@ -37,8 +37,7 @@ let isWarmingUp = false;
 let tempQueue = [];
 let parser = null;
 
-// MIDI to Pitch Name helpers
-const abcNotePattern = /^(?:\^\^|__|\^|_|=)?[A-Ga-g][,']*$/;
+
 
 function tokenId(token) {
   return stoi?.[token] ?? null;
@@ -119,21 +118,7 @@ function sampleFromLogits(logits) {
 }
 
 // Duration converters
-function durationToSecondsFromAbc(duration) {
-  let eighths = 1;
-  if (duration) {
-    if (duration === "/") eighths = 0.5;
-    else if (duration.startsWith("/")) eighths = 1 / Number(duration.slice(1));
-    else if (duration.includes("/")) {
-      const [a, b] = duration.split("/").map(Number);
-      eighths = a / b;
-    } else {
-      eighths = Number(duration);
-    }
-  }
-  const quarterSeconds = 60 / currentBpm;
-  return Math.max(0.08, eighths * quarterSeconds * 0.5);
-}
+
 
 function durationToSecondsFromEventSteps(steps, grid) {
   const quarterSeconds = 60 / currentBpm;
@@ -145,17 +130,7 @@ function midiToTonePitch(midi) {
   return `${names[((midi % 12) + 12) % 12]}${Math.floor(midi / 12) - 1}`;
 }
 
-function abcToTonePitch(token) {
-  const match = token.match(/^(\^\^|__|\^|_|=)?([A-Ga-g])([,']*)$/);
-  if (!match) return null;
-  const accidental = match[1] || "";
-  const step = match[2];
-  const marks = match[3] || "";
-  let octave = step === step.toLowerCase() ? 5 : 4;
-  for (const mark of marks) octave += mark === "'" ? 1 : -1;
-  const accidentalText = accidental === "^^" ? "##" : accidental === "__" ? "bb" : accidental === "^" ? "#" : accidental === "_" ? "b" : "";
-  return `${step.toUpperCase()}${accidentalText}${octave}`;
-}
+
 
 function pushEvent(event) {
   if (isWarmingUp) return; // Discard prompt history events
@@ -178,68 +153,7 @@ function getTokensUpToBar(tokens, targetBar) {
   return sliced;
 }
 
-// ABC Parser
-function makeAbcParser() {
-  return {
-    pending: null,
-    chord: null,
-    reset() {
-      this.pending = null;
-      this.chord = null;
-    },
-    feed(token) {
-      if (token === "Q:") {
-        this.pending = { type: "tempoHeader" };
-        return;
-      }
-      if (this.pending?.type === "tempoHeader") {
-        const bpm = Number(String(token).split("=").pop());
-        if (Number.isFinite(bpm) && bpm >= 40 && bpm <= 220) {
-          if (!midiSeedActive || isWarmingUp || midiBpmMode === "model") {
-            currentBpm = bpm;
-            self.postMessage({ action: "tempo", bpm: bpm });
-          }
-        }
-        this.pending = null;
-        return;
-      }
-      if (token === "[") {
-        this.chord = [];
-        this.pending = null;
-        return;
-      }
-      if (this.chord) {
-        if (token === "]") {
-          this.pending = { type: "chord", notes: this.chord.map(abcToTonePitch).filter(Boolean) };
-          this.chord = null;
-          return;
-        }
-        if (abcNotePattern.test(token)) this.chord.push(token);
-        return;
-      }
-      if (token === "z") {
-        this.pending = { type: "rest" };
-        return;
-      }
-      if (abcNotePattern.test(token)) {
-        this.pending = { type: "note", note: abcToTonePitch(token) };
-        return;
-      }
-      if (token.startsWith("DUR:") && this.pending) {
-        this.emitPending(token.slice(4));
-        return;
-      }
-      if (this.pending && (token === "|" || token === "<NL>" || token === "<EOP>")) this.emitPending(null);
-    },
-    emitPending(durationToken) {
-      const duration = durationToSecondsFromAbc(durationToken);
-      if (this.pending.type === "rest") pushEvent({ type: "rest", duration });
-      if (this.pending.type === "note" && this.pending.note) pushEvent({ type: "note", notes: [this.pending.note], duration, advance: duration });
-      if (this.pending.type === "chord" && this.pending.notes.length) pushEvent({ type: "note", notes: this.pending.notes.slice(0, 8), duration, advance: duration });
-      this.pending = null;
-    },
-  };
-}
+
 
 // Event Parser
 function makeEventParser() {
@@ -346,18 +260,32 @@ function makeEventParser() {
 }
 
 function makeParser() {
-  return activeTokenizer === "giantmidi_event" ? makeEventParser() : makeAbcParser();
+  return makeEventParser();
 }
 
 async function warmPrompt() {
   h = zeroState();
   c = zeroState();
   
-  const allTokensStr = midiSeedActive && midiTokens.length > 0 
-    ? getTokensUpToBar(midiTokens, midiStartBar) 
-    : (activeTokenizer === "giantmidi_event" 
-        ? ["BOS", "BPM_120", "GRID_64", "BAR", "POS_0"] 
-        : ["X:", "1", "<NL>", "T:", "piece", "<NL>", "M:", "4/4", "<NL>", "L:", "1/8", "<NL>", "Q:", "1/4=120", "<NL>", "K:", "C", "<NL>"]);
+  let allTokensStr;
+  if (midiSeedActive && midiTokens.length > 0) {
+    allTokensStr = getTokensUpToBar(midiTokens, midiStartBar);
+  } else if (bachSeedActive) {
+    allTokensStr = [
+      "BOS", "BPM_100", "GRID_64",
+      "BAR",
+      "POS_0", "NOTE_81", "DUR_4", "VEL_6",
+      "POS_4", "NOTE_79", "DUR_4", "VEL_6",
+      "POS_8", "NOTE_77", "DUR_4", "VEL_6",
+      "POS_12", "NOTE_76", "DUR_4", "VEL_6",
+      "POS_16", "NOTE_74", "DUR_4", "VEL_6",
+      "POS_20", "NOTE_73", "DUR_4", "VEL_6",
+      "POS_24", "NOTE_74", "DUR_16", "VEL_6",
+      "BAR"
+    ];
+  } else {
+    allTokensStr = ["BOS", "BPM_120", "GRID_64", "BAR", "POS_0"];
+  }
   
   isWarmingUp = true;
   
@@ -406,7 +334,6 @@ self.onmessage = async function (e) {
         stoi = data.vocab.stoi;
         itos = Object.fromEntries(Object.entries(data.vocab.itos).map(([key, value]) => [Number(key), value]));
         vocabSize = data.vocab.vocab_size;
-        activeTokenizer = data.vocab.tokenizer || "abc";
         
         session = await ort.InferenceSession.create(data.modelBuffer, {
           executionProviders: ["wasm"],
@@ -430,6 +357,7 @@ self.onmessage = async function (e) {
         midiTokens = data.midiTokens;
         midiStartBar = data.midiStartBar;
         midiBpmMode = data.midiBpmMode || "lock";
+        bachSeedActive = data.bachSeedActive || false;
         
         parser = makeParser();
         parser.reset();