Update app.py

app.py

@@ -76,7 +76,7 @@ class CustomTrainerCallback(TrainerCallback):
                 job.progress = 0.1 + (prog * 0.8)
                 if state.global_step % 1 == 0:
                     loss = state.log_history[-1].get('loss', 'N/A') if state.log_history else '...'
-                    job.add_log(f"Step {state.global_step}/{state.max_steps} | Loss: {loss}")
+                    job.add_log(f"Training Step {state.global_step}/{state.max_steps} | Loss: {loss}")
         return control
 
     def on_save(self, args, state, control, **kwargs):
@@ -281,7 +281,14 @@ def background_train_task(job_id, hf_token, model_name, new_repo_name, lora_r, l
             if content and content.strip():
                 try:
                     data = json.loads(content)
-                    with open(os.path.join(final_path, fname), 'w') as f:
+                    file_path = os.path.join(final_path, fname)
+                    if os.path.exists(file_path):
+                        with open(file_path, 'r', encoding='utf-8') as f:
+                            existing_data = json.load(f)
+                        existing_data.update(data)
+                        data = existing_data
+                    
+                    with open(file_path, 'w', encoding='utf-8') as f:
                         json.dump(data, f, indent=2)
                 except:
                     pass