refactor(gradio): decompose training interface into focused modules

acestep/ui/gradio/i18n/en.json

@@ -5,8 +5,8 @@
   },
   "common": {
     "language_metadata": {
-    "name": "English",
-    "native_name": "English"
+      "name": "English",
+      "native_name": "English"
     }
   },
   "dataset": {
@@ -424,25 +424,55 @@
     "export_path": "Export Path",
     "export_lora_btn": "📦 Export LoRA",
     "export_status": "Export Status",
-    "stop_no_training": "\u2139 No training in progress",
-    "stop_stopping": "\u23F9\uFE0F Stopping training...",
+    "stop_no_training": "ℹ No training in progress",
+    "stop_stopping": "⏹️ Stopping training...",
     "latest_auto": "Latest (auto)",
-    "export_path_required": "\u274C Please enter an export path",
-    "invalid_lora_output_dir": "\u274C Invalid LoRA output directory",
-    "no_checkpoints_found": "\u274C No checkpoints found",
-    "no_trained_model_found": "\u274C No trained model found in {path}",
-    "invalid_export_path": "\u274C Invalid export path",
-    "lora_exported": "\u2705 LoRA exported to {path}",
-    "export_failed": "\u274C Export failed: {error}",
-    "lokr_output_dir_required": "\u26A0\uFE0F Enter LoKr output directory first",
-    "lokr_no_checkpoints_use_latest": "\u2139 No checkpoints found; export will use latest available weights",
-    "lokr_no_exportable_checkpoints": "\u2139 No exportable epoch checkpoints found",
-    "lokr_found_checkpoints": "\u2705 Found {count} LoKr checkpoints",
-    "lokr_selected_epoch_not_found": "\u274C Selected epoch not found: {chosen}. Available: {available}",
-    "lokr_no_weights_selected_epoch": "\u274C No LoKr weights found for selected epoch: {epoch}",
-    "lokr_no_weights_latest_checkpoint": "\u274C No LoKr weights found in latest checkpoint: {checkpoint}",
-    "lokr_no_trained_weights_found": "\u274C No trained LoKr weights found in {path}",
-    "lokr_exported": "\u2705 LoKr exported to {path}"
+    "export_path_required": "❌ Please enter an export path",
+    "invalid_lora_output_dir": "❌ Invalid LoRA output directory",
+    "no_checkpoints_found": "❌ No checkpoints found",
+    "no_trained_model_found": "❌ No trained model found in {path}",
+    "invalid_export_path": "❌ Invalid export path",
+    "lora_exported": "✅ LoRA exported to {path}",
+    "export_failed": "❌ Export failed: {error}",
+    "lokr_output_dir_required": "⚠️ Enter LoKr output directory first",
+    "lokr_no_checkpoints_use_latest": "ℹ No checkpoints found; export will use latest available weights",
+    "lokr_no_exportable_checkpoints": "ℹ No exportable epoch checkpoints found",
+    "lokr_found_checkpoints": "✅ Found {count} LoKr checkpoints",
+    "lokr_selected_epoch_not_found": "❌ Selected epoch not found: {chosen}. Available: {available}",
+    "lokr_no_weights_selected_epoch": "❌ No LoKr weights found for selected epoch: {epoch}",
+    "lokr_no_weights_latest_checkpoint": "❌ No LoKr weights found in latest checkpoint: {checkpoint}",
+    "lokr_no_trained_weights_found": "❌ No trained LoKr weights found in {path}",
+    "lokr_exported": "✅ LoKr exported to {path}",
+    "tab_train_lokr": "🚀 Train LoKr",
+    "train_section_tensors": "Preprocessed Tensors",
+    "train_section_lora": "LoRA Settings",
+    "train_section_params": "Training Parameters",
+    "lokr_section_tensors": "Preprocessed Tensors",
+    "lokr_section_settings": "LoKr Settings",
+    "lokr_tensor_selection_desc": "Select the directory containing preprocessed tensor files (`.pt` files).\nThese are created using the 'Preprocess' button in the 'Dataset Builder' tab.",
+    "lokr_linear_dim": "LoKr Linear Dim",
+    "lokr_linear_dim_info": "Rank (dimension) for LoKr adaptation matrices.",
+    "lokr_linear_alpha": "LoKr Linear Alpha",
+    "lokr_linear_alpha_info": "Scaling factor for LoKr (usually similar to dim).",
+    "lokr_factor": "LoKr Factor",
+    "lokr_factor_info": "Kronecker factor (-1 for auto).",
+    "lokr_decompose_both": "Decompose Both Sides",
+    "lokr_decompose_both_info": "When enabled, decomposes both left and right matrices.",
+    "lokr_use_tucker": "Use Tucker Decomposition",
+    "lokr_use_tucker_info": "Apply Tucker decomposition when applicable.",
+    "lokr_use_scalar": "Use Scalar Gate",
+    "lokr_use_scalar_info": "Enable scalar gating for LoKr weights.",
+    "lokr_weight_decompose": "Weight Decompose (WD)",
+    "lokr_weight_decompose_info": "Enable weight decomposition for more stable LoKr training.",
+    "lokr_learning_rate_info": "LoKr commonly uses a higher LR than LoRA. Tune per dataset.",
+    "lokr_output_dir_info": "Directory to save trained LoKr weights.",
+    "start_lokr_training_btn": "🚀 Start Training LoKr",
+    "lokr_training_loss_title": "LoKr Training Loss",
+    "lokr_export_header": "Export LoKr",
+    "export_lokr_btn": "📦 Export LoKr",
+    "lokr_checkpoint_epoch": "Checkpoint Epoch",
+    "lokr_checkpoint_epoch_info": "Select a specific epoch checkpoint to export, or keep Latest (auto).",
+    "refresh_epochs_btn": "↻ Refresh Epochs"
   },
   "help": {
     "btn_label": "?",

acestep/ui/gradio/i18n/he.json

@@ -442,7 +442,37 @@
     "lokr_no_weights_selected_epoch": "❌ לא נמצאו משקלי LoKr לאיטרציה שנבחרה: {epoch}",
     "lokr_no_weights_latest_checkpoint": "❌ לא נמצאו משקלי LoKr בנקודת הביקורת האחרונה",
     "lokr_no_trained_weights_found": "❌ לא נמצאו משקלי LoKr מאומנים ב-{path}",
-    "lokr_exported": "✅ LoKr יוצא ל-{path}"
+    "lokr_exported": "✅ LoKr יוצא ל-{path}",
+    "tab_train_lokr": "🚀 אימון LoKr",
+    "train_section_tensors": "Preprocessed Tensors",
+    "train_section_lora": "LoRA Settings",
+    "train_section_params": "Training Parameters",
+    "lokr_section_tensors": "Preprocessed Tensors",
+    "lokr_section_settings": "LoKr Settings",
+    "lokr_tensor_selection_desc": "בחר את הספרייה המכילה את קבצי הטנזורים המעובדים (`.pt`).",
+    "lokr_linear_dim": "LoKr Linear Dim",
+    "lokr_linear_dim_info": "Rank (dimension) for LoKr adaptation matrices.",
+    "lokr_linear_alpha": "LoKr Linear Alpha",
+    "lokr_linear_alpha_info": "Scaling factor for LoKr (usually similar to dim).",
+    "lokr_factor": "LoKr Factor",
+    "lokr_factor_info": "Kronecker factor (-1 for auto).",
+    "lokr_decompose_both": "Decompose Both Sides",
+    "lokr_decompose_both_info": "When enabled, decomposes both left and right matrices.",
+    "lokr_use_tucker": "Use Tucker Decomposition",
+    "lokr_use_tucker_info": "Apply Tucker decomposition when applicable.",
+    "lokr_use_scalar": "Use Scalar Gate",
+    "lokr_use_scalar_info": "Enable scalar gating for LoKr weights.",
+    "lokr_weight_decompose": "Weight Decompose (WD)",
+    "lokr_weight_decompose_info": "Enable weight decomposition for more stable LoKr training.",
+    "lokr_learning_rate_info": "LoKr commonly uses a higher LR than LoRA. Tune per dataset.",
+    "lokr_output_dir_info": "ספרייה לשמירת משקלי LoKr המאומנים.",
+    "start_lokr_training_btn": "🚀 התחל אימון LoKr",
+    "lokr_training_loss_title": "LoKr אובדן אימון (Loss)",
+    "lokr_export_header": "ייצוא LoKr",
+    "export_lokr_btn": "📦 ייצוא LoKr",
+    "lokr_checkpoint_epoch": "Checkpoint Epoch",
+    "lokr_checkpoint_epoch_info": "Select a specific epoch checkpoint to export, or keep Latest (auto).",
+    "refresh_epochs_btn": "↻ רענון Epochs"
   },
   "help": {
     "btn_label": "?",

acestep/ui/gradio/i18n/ja.json

@@ -5,8 +5,8 @@
   },
   "common": {
     "language_metadata": {
-    "name": "Japanese",
-    "native_name": "日本語"
+      "name": "Japanese",
+      "native_name": "日本語"
     }
   },
   "dataset": {
@@ -426,7 +426,34 @@
     "lokr_no_weights_selected_epoch": "❌ 選択したエポック {epoch} にLoKr重みが見つかりません",
     "lokr_no_weights_latest_checkpoint": "❌ 最新チェックポイント {checkpoint} にLoKr重みが見つかりません",
     "lokr_no_trained_weights_found": "❌ {path} にトレーニング済みLoKr重みが見つかりません",
-    "lokr_exported": "✅ LoKr を {path} にエクスポートしました"
+    "lokr_exported": "✅ LoKr を {path} にエクスポートしました",
+    "tab_train_lokr": "🚀 LoKr をトレーニング",
+    "lokr_section_tensors": "前処理済みデータセット選択",
+    "lokr_section_settings": "LoKr 設定",
+    "lokr_tensor_selection_desc": "前処理されたテンソルファイル（`.pt` ファイル）を含むディレクトリを選択します。\nこれらは「データセットビルダー」タブの「前処理」ボタンを使用して作成されます。",
+    "lokr_linear_dim": "LoKr Linear Dim",
+    "lokr_linear_dim_info": "Rank (dimension) for LoKr adaptation matrices.",
+    "lokr_linear_alpha": "LoKr Linear Alpha",
+    "lokr_linear_alpha_info": "Scaling factor for LoKr (usually similar to dim).",
+    "lokr_factor": "LoKr Factor",
+    "lokr_factor_info": "Kronecker factor (-1 for auto).",
+    "lokr_decompose_both": "Decompose Both Sides",
+    "lokr_decompose_both_info": "When enabled, decomposes both left and right matrices.",
+    "lokr_use_tucker": "Use Tucker Decomposition",
+    "lokr_use_tucker_info": "Apply Tucker decomposition when applicable.",
+    "lokr_use_scalar": "Use Scalar Gate",
+    "lokr_use_scalar_info": "Enable scalar gating for LoKr weights.",
+    "lokr_weight_decompose": "Weight Decompose (WD)",
+    "lokr_weight_decompose_info": "Enable weight decomposition for more stable LoKr training.",
+    "lokr_learning_rate_info": "LoKr commonly uses a higher LR than LoRA. Tune per dataset.",
+    "lokr_output_dir_info": "トレーニング済み LoKr 重みの保存先ディレクトリ。",
+    "start_lokr_training_btn": "🚀 トレーニング開始 LoKr",
+    "lokr_training_loss_title": "LoKr トレーニング損失",
+    "lokr_export_header": "LoKr をエクスポート",
+    "export_lokr_btn": "📦 LoKr をエクスポート",
+    "lokr_checkpoint_epoch": "Checkpoint Epoch",
+    "lokr_checkpoint_epoch_info": "Select a specific epoch checkpoint to export, or keep Latest (auto).",
+    "refresh_epochs_btn": "↻ 更新 Epochs"
   },
   "help": {
     "btn_label": "?",

acestep/ui/gradio/i18n/zh.json

@@ -5,8 +5,8 @@
   },
   "common": {
     "language_metadata": {
-    "name": "Chinese",
-    "native_name": "中文"
+      "name": "Chinese",
+      "native_name": "中文"
     }
   },
   "dataset": {
@@ -426,7 +426,34 @@
     "lokr_no_weights_selected_epoch": "❌ 所选轮次 {epoch} 中未找到 LoKr 权重",
     "lokr_no_weights_latest_checkpoint": "❌ 最新检查点 {checkpoint} 中未找到 LoKr 权重",
     "lokr_no_trained_weights_found": "❌ 在 {path} 中未找到训练好的 LoKr 权重",
-    "lokr_exported": "✅ LoKr 已导出到 {path}"
+    "lokr_exported": "✅ LoKr 已导出到 {path}",
+    "tab_train_lokr": "🚀 训练 LoKr",
+    "lokr_section_tensors": "预处理数据集选择",
+    "lokr_section_settings": "LoKr 设置",
+    "lokr_tensor_selection_desc": "选择包含预处理张量文件（`.pt` 文件）的目录。\n这些文件在「数据集构建」标签页中使用「预处理」按钮创建。",
+    "lokr_linear_dim": "LoKr Linear Dim",
+    "lokr_linear_dim_info": "Rank (dimension) for LoKr adaptation matrices.",
+    "lokr_linear_alpha": "LoKr Linear Alpha",
+    "lokr_linear_alpha_info": "Scaling factor for LoKr (usually similar to dim).",
+    "lokr_factor": "LoKr Factor",
+    "lokr_factor_info": "Kronecker factor (-1 for auto).",
+    "lokr_decompose_both": "Decompose Both Sides",
+    "lokr_decompose_both_info": "When enabled, decomposes both left and right matrices.",
+    "lokr_use_tucker": "Use Tucker Decomposition",
+    "lokr_use_tucker_info": "Apply Tucker decomposition when applicable.",
+    "lokr_use_scalar": "Use Scalar Gate",
+    "lokr_use_scalar_info": "Enable scalar gating for LoKr weights.",
+    "lokr_weight_decompose": "Weight Decompose (WD)",
+    "lokr_weight_decompose_info": "Enable weight decomposition for more stable LoKr training.",
+    "lokr_learning_rate_info": "LoKr commonly uses a higher LR than LoRA. Tune per dataset.",
+    "lokr_output_dir_info": "保存训练后 LoKr 权重的目录。",
+    "start_lokr_training_btn": "🚀 开始训练 LoKr",
+    "lokr_training_loss_title": "LoKr 训练损失",
+    "lokr_export_header": "导出 LoKr",
+    "export_lokr_btn": "📦 导出 LoKr",
+    "lokr_checkpoint_epoch": "Checkpoint Epoch",
+    "lokr_checkpoint_epoch_info": "Select a specific epoch checkpoint to export, or keep Latest (auto).",
+    "refresh_epochs_btn": "↻ 刷新 Epochs"
   },
   "help": {
     "btn_label": "?",