Add functionality to refresh scripts and configuration files in app.py

Implement _find_latest_dataset_dir to locate the most recent dataset directory and _collect_scripts_and_config to gather relevant files. Enhance the UI with a button to refresh scripts from the latest dataset, improving user experience and accessibility to training resources.

app.py

@@ -270,6 +270,46 @@ def _list_checkpoints(out_dir: str, limit: int = 20) -> List[str]:
         return []
 
 
+def _find_latest_dataset_dir(root: str) -> Optional[str]:
+    try:
+        if not os.path.isdir(root):
+            return None
+        cand: List[Tuple[float, str]] = []
+        for name in os.listdir(root):
+            if not name.startswith("dataset_"):
+                continue
+            full = os.path.join(root, name)
+            if os.path.isdir(full):
+                try:
+                    cand.append((os.path.getmtime(full), full))
+                except Exception:
+                    pass
+        if not cand:
+            return None
+        cand.sort(reverse=True)
+        return cand[0][1]
+    except Exception:
+        return None
+
+
+def _collect_scripts_and_config(ds_dir: Optional[str]) -> List[str]:
+    files: List[str] = []
+    try:
+        ds_conf = str(Path(AUTO_DIR_RUNTIME) / "dataset_QIE.toml")
+        if os.path.isfile(ds_conf):
+            files.append(ds_conf)
+        if ds_dir and os.path.isdir(ds_dir):
+            used_script = os.path.join(ds_dir, "train_QIE_used.sh")
+            if os.path.isfile(used_script):
+                files.append(used_script)
+            meta = os.path.join(ds_dir, "metadata.jsonl")
+            if os.path.isfile(meta):
+                files.append(meta)
+    except Exception:
+        pass
+    return files
+
+
 def _files_to_gallery(files: Any) -> List[str]:
     items: List[str] = []
     if not files:
@@ -959,6 +999,8 @@ def build_ui() -> gr.Blocks:
                 logs = gr.Textbox(label="Logs", lines=20)
                 ckpt_files = gr.Files(label="Checkpoints (live)", interactive=False)
                 scripts_files = gr.Files(label="Scripts & Config (live)", interactive=False)
+                with gr.Row():
+                    refresh_scripts_btn = gr.Button("ファイルを再取得", variant="secondary")
 
                 # moved max_epochs/save_every above next to OUTPUT NAME
 
@@ -992,6 +1034,20 @@ def build_ui() -> gr.Blocks:
                     outputs=[logs, ckpt_files, scripts_files],
                 )
 
+                # 回収ボタン: 直近の dataset_ ディレクトリからファイルを再取得
+                def _refresh_scripts() -> List[str]:
+                    try:
+                        ds_dir = _find_latest_dataset_dir(DATA_ROOT_RUNTIME)
+                        return _collect_scripts_and_config(ds_dir)
+                    except Exception:
+                        return _collect_scripts_and_config(None)
+
+                refresh_scripts_btn.click(
+                    fn=_refresh_scripts,
+                    inputs=[],
+                    outputs=[scripts_files],
+                )
+
             with gr.TabItem("Prompt Generator"):
                 gr.Markdown("""
                 # 🎨 A→B 変換プロンプト自動生成