fix error regarding progress

vms/ui/app_ui.py

@@ -399,7 +399,9 @@ class AppUI:
             outputs=[
                 self.project_tabs["train_tab"].components["status_box"],
                 self.project_tabs["train_tab"].components["log_box"],
-                self.project_tabs["train_tab"].components["current_task_box"] if "current_task_box" in self.project_tabs["train_tab"].components else None
+                self.project_tabs["train_tab"].components["current_task_box"] if "current_task_box" in self.project_tabs["train_tab"].components else None,
+                self.project_tabs["manage_tab"].components["download_model_btn"],
+                self.project_tabs["manage_tab"].components["download_checkpoint_btn"]
             ]
         )
         

vms/ui/models/tabs/training_tab.py

@@ -88,9 +88,8 @@ class TrainingTab(BaseTab):
                             gr.Markdown(model.model_display_name or "Unknown")
                         
                         with gr.Column(scale=2, min_width=20):
-                            progress_text = f"Step {model.current_step}/{model.total_steps}"
+                            progress_text = f"Step {model.current_step}/{model.total_steps} ({model.training_progress:.1f}%)"
                             gr.Markdown(progress_text)
-                            gr.Progress(value=model.training_progress/100)
                         
                         with gr.Column(scale=2, min_width=20):
                             with gr.Row():

vms/ui/project/tabs/train_tab.py

@@ -1201,7 +1201,12 @@ Full finetune mode trains all parameters of the model, requiring more VRAM but p
         if hasattr(self.app, 'log_parser') and self.app.log_parser is not None:
             current_task = self.app.log_parser.get_current_task_display()
         
-        return message, logs, current_task
+        # Update download button texts
+        manage_tab = self.app.tabs["manage_tab"]
+        download_btn_text = gr.update(value=manage_tab.get_download_button_text())
+        checkpoint_btn_text = gr.update(value=manage_tab.get_checkpoint_button_text())
+        
+        return message, logs, current_task, download_btn_text, checkpoint_btn_text
 
     def get_button_updates(self):
         """Get button updates (with variant property)"""