fix visualizer

app.py

@@ -173,12 +173,8 @@ def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, gen_event
     with open(f"output.mid", 'wb') as f:
         f.write(MIDI.score2midi(mid))
     audio = synthesis(MIDI.score2opus(mid), soundfont_path)
-    # resend all msgs
-    msgs = [create_msg("visualizer_end", None), create_msg("visualizer_clear", True)]
-    for tokens in mid_seq:
-        msgs.append(create_msg("visualizer_append", tokenizer.tokens2event(tokens)))
-    msgs.append(create_msg("visualizer_end", None))
-    yield mid_seq, "output.mid", (44100, audio), send_msgs(msgs)
+    events = [tokenizer.tokens2event(tokens) for tokens in mid_seq]
+    yield mid_seq, "output.mid", (44100, audio), send_msgs([create_msg("visualizer_end", events)])
 
 
 def cancel_run(mid_seq):
@@ -188,11 +184,8 @@ def cancel_run(mid_seq):
     with open(f"output.mid", 'wb') as f:
         f.write(MIDI.score2midi(mid))
     audio = synthesis(MIDI.score2opus(mid), soundfont_path)
-    msgs = [create_msg("visualizer_end", None), create_msg("visualizer_clear", True)]
-    for tokens in mid_seq:
-        msgs.append(create_msg("visualizer_append", tokenizer.tokens2event(tokens)))
-    msgs.append(create_msg("visualizer_end", None))
-    return "output.mid", (44100, audio), send_msgs(msgs)
+    events = [tokenizer.tokens2event(tokens) for tokens in mid_seq]
+    return "output.mid", (44100, audio), send_msgs([create_msg("visualizer_end", events)])
 
 
 def load_javascript(dir="javascript"):

javascript/app.js

@@ -427,7 +427,7 @@ customElements.define('midi-visualizer', MidiVisualizer);
         handled_msgs.push(msg.uuid);
         switch (msg.name) {
             case "visualizer_clear":
-                midi_visualizer.clearMidiEvents(msg.data);
+                midi_visualizer.clearMidiEvents(false);
                 createProgressBar(midi_visualizer_container_inited)
                 break;
             case "visualizer_append":
@@ -439,6 +439,10 @@ customElements.define('midi-visualizer', MidiVisualizer);
                 setProgressBar(midi_visualizer_container_inited, progress, total)
                 break;
             case "visualizer_end":
+                midi_visualizer.clearMidiEvents(true);
+                msg.data.forEach( value => {
+                    midi_visualizer.appendMidiEvent(value);
+                })
                 midi_visualizer.finishAppendMidiEvent()
                 midi_visualizer.setPlayTime(0);
                 removeProgressBar(midi_visualizer_container_inited);