Upload app.py

app.py

@@ -481,7 +481,7 @@ def generate_text_sse():
                 temperature=temperature, cfg_scale=cfg_scale, remasking=remasking,
                 capture_interval=capture_interval
             ):
-                yield f"data: {json.dumps(state)}"
+                yield f"data: {json.dumps(state)}\n\n"
         else:
             streamer = TextIteratorStreamer(
                 tokenizer, skip_prompt=True, skip_special_tokens=True
@@ -504,11 +504,14 @@ def generate_text_sse():
 
             accumulated = []
             for text in streamer:
+                if not text:  # skip empty chunks
+                    continue
                 accumulated.append(text)
-                yield f"data: {json.dumps({'type': 'token', 'text': text})}"
+                current = "".join(accumulated)
+                yield f"data: {json.dumps({'type': 'intermediate', 'text': current})}\n\n"
 
             full_text = "".join(accumulated)
-            yield f"data: {json.dumps({'type': 'final', 'text': full_text, 'total_steps': 1})}"
+            yield f"data: {json.dumps({'type': 'final', 'text': full_text, 'total_steps': 1})}\n\n"
 
     return Response(
         stream(), mimetype='text/event-stream',