Refactor logging in run_training function of app.py to accumulate log messages in a buffer before yielding, improving output consistency and readability during training execution.

app.py

@@ -444,6 +444,7 @@ def run_training(
     save_every: int,
 ) -> Iterable[tuple]:
     # Basic validation
+    log_buf = ""
     if not output_name.strip():
         yield ("[ERROR] OUTPUT NAME is required.", None)
         return
@@ -470,7 +471,8 @@ def run_training(
         yield ("[ERROR] No images uploaded for IMAGE_FOLDER.", None)
         return
     base_filenames = _copy_uploads(base_files, img_dir)
-    yield (f"[QIE] Copied {len(base_filenames)} base images to {img_dir}", None)
+    log_buf += f"[QIE] Copied {len(base_filenames)} base images to {img_dir}\n"
+    yield (log_buf, None)
 
     # Prepare control sets
     control_upload_sets = [
@@ -499,7 +501,8 @@ def run_training(
         # Simply copy; name matching will be handled by create_image_caption_json.py
         _copy_uploads(uploads, cdir)
         control_dirs.append(folder_name)
-        yield (f"[QIE] Copied {len(uploads)} control_{i} images to {cdir}", None)
+        log_buf += f"[QIE] Copied {len(uploads)} control_{i} images to {cdir}\n"
+        yield (log_buf, None)
 
     # Metadata.jsonl will be generated by create_image_caption_json.py in train_QIE.sh
 
@@ -540,8 +543,9 @@ def run_training(
 
 
     shell = _pick_shell()
-    yield (f"[QIE] Using shell: {shell}", None)
-    yield (f"[QIE] Running script: {tmp_script}", None)
+    log_buf += f"[QIE] Using shell: {shell}\n"
+    log_buf += f"[QIE] Running script: {tmp_script}\n"
+    yield (log_buf, None)
 
     # Run and stream output
     proc = subprocess.Popen(
@@ -555,7 +559,8 @@ def run_training(
     try:
         assert proc.stdout is not None
         for line in proc.stdout:
-            yield (line.rstrip("\n"), None)
+            log_buf += line
+            yield (log_buf, None)
     finally:
         code = proc.wait()
         # Try to locate latest LoRA file for download
@@ -574,7 +579,8 @@ def run_training(
                     lora_path = cand[-1][1]
         except Exception:
             pass
-        yield (f"[QIE] Exit code: {code}", lora_path)
+        log_buf += f"[QIE] Exit code: {code}\n"
+        yield (log_buf, lora_path)
 
 
 def build_ui() -> gr.Blocks: