Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import json
 import re
 import requests
 import gradio as gr
@@ -74,21 +75,33 @@ def list_files_with_no(repo_type):
     except Exception as e:
         return [[f"エラー: ファイルリストの取得に失敗しました: {e}", None]]
 
-# 並び替えをリポジトリに反映
-def apply_reorder_to_repo(data, repo_type):
+# 並び替え結果をリポジトリにメタデータとして保存
+def save_reorder_to_repo(data, repo_type):
     try:
         target_repo = MODEL_REPO if repo_type == "model" else LORA_REPO
         if not target_repo:
             return "エラー: リポジトリが設定されていません。"
         
-        # 並び替えられたファイルリストをリポジトリに反映 (例としてMockで結果を出力)
+        # 並び替えられたファイルリストをJSONとして保存
         reordered_files = sorted(data, key=lambda x: int(x[0]))  # No.列の順序に基づいて並び替え
-        reordered_filenames = [file[1] for file in reordered_files]
+        metadata = {"order": [{"No": file[0], "Filename": file[1]} for file in reordered_files]}
 
-        # Mock反映: 並び替えたファイル名を表示
-        # 実際のリポジトリ反映処理はAPIやスクリプトで実装
-        result_message = f"リポジトリ順序を以下の順に反映しました:\n" + "\n".join(reordered_filenames)
-        return reordered_files, result_message
+        # JSONファイルに保存
+        metadata_filename = "file_order.json"
+        with open(metadata_filename, "w") as f:
+            json.dump(metadata, f, indent=4)
+
+        # JSONファイルをリポジトリにアップロード
+        api.upload_file(
+            path_or_fileobj=metadata_filename,
+            path_in_repo=metadata_filename,
+            repo_id=target_repo,
+            token=HF_TOKEN,
+            repo_type="dataset",
+            commit_message="Update file order metadata"
+        )
+
+        return reordered_files, f"ファイル順序がリポジトリに反映されました ({metadata_filename})"
     except Exception as e:
         return data, f"リポジトリ更新中にエラーが発生しました: {e}"
 
@@ -111,12 +124,12 @@ with gr.Blocks() as demo:
     load_button.click(list_files_with_no, inputs=repo_choice, outputs=file_list_output)
 
     # 並び替えセクション
-    gr.Markdown("### No.列を変更して並び替えを適用し、リポジトリに反映")
+    gr.Markdown("### No.列を変更して並び替えを適用し、リポジトリに保存")
     reorder_button = gr.Button("並び替え適用")
     sorted_table = gr.Dataframe(headers=["No.", "Filename"], datatype=["number", "str"], interactive=False)
 
     def apply_reorder(data, repo_type):
-        reordered_data, message = apply_reorder_to_repo(data, repo_type)
+        reordered_data, message = save_reorder_to_repo(data, repo_type)
         return reordered_data, message
 
     reorder_button.click(apply_reorder, inputs=[file_list_output, repo_choice], outputs=[sorted_table, output])