Update app.py

app.py

@@ -13,42 +13,32 @@ CIVITAI_TOKEN = os.environ.get("CIVITAI_TOKEN", None)  # Civitai用トークン
 api = HfApi()
 
 def download_and_upload(url, repo_type):
-    # 入力チェック
     if not url or not repo_type:
         return "URLまたはタイプが未指定です。"
 
-    # Civitai API用ヘッダ
     headers = {}
     if CIVITAI_TOKEN:
         headers["Authorization"] = f"Bearer {CIVITAI_TOKEN}"
 
-    # ファイルダウンロード実行
-    # CivitaiのAPIからモデルを取得
     try:
         response = requests.get(url, headers=headers)
-        response.raise_for_status()  # ステータスコードがエラーなら例外
+        response.raise_for_status()
     except Exception as e:
         return f"ファイルのダウンロードに失敗しました: {e}"
 
-    # Content-Dispositionからfilenameを抽出
-    # 例: Content-Disposition: attachment; filename="model.safetensors"
     content_disp = response.headers.get("content-disposition", "")
     filename = None
     if content_disp:
-        # filename="..." という部分を正規表現で抽出
         match = re.search(r'filename="?([^"]+)"?', content_disp)
         if match:
             filename = match.group(1)
 
     if not filename:
-        # ファイル名がヘッダに無い場合のfallback
         filename = "downloaded_file.bin"
     
-    # ダウンロードファイルをローカル一時ファイルに保存
     with open(filename, "wb") as f:
         f.write(response.content)
 
-    # 選択されたリポジトリを決定
     if repo_type == "model":
         target_repo = MODEL_REPO
     else:
@@ -57,24 +47,50 @@ def download_and_upload(url, repo_type):
     if not target_repo:
         return f"{repo_type}用リポジトリが環境変数で設定されていません。"
 
-    # Hugging Face Hub にアップロード
     try:
         api.upload_file(
             path_or_fileobj=filename,
             path_in_repo=filename,
             repo_id=target_repo,
             token=HF_TOKEN,
-            repo_type="dataset",  # dataset リポジトリとして扱う
+            repo_type="dataset",
             commit_message=f"Add {filename} from {url}"
         )
         return f"ファイル '{filename}' を '{target_repo}' にアップロードしました。"
     except Exception as e:
         return f"アップロード中にエラーが発生しました: {e}"
-    finally:
-        # 必要ならローカルファイル削除などの後処理
-        pass
 
-# Gradio UI構築 (前回とほぼ同様)
+# ファイルリストを取得
+def list_files(repo_type):
+    target_repo = MODEL_REPO if repo_type == "model" else LORA_REPO
+    if not target_repo:
+        return f"{repo_type}用リポジトリが環境変数で設定されていません。"
+
+    try:
+        files = api.list_repo_files(repo_id=target_repo, token=HF_TOKEN, repo_type="dataset")
+        return files
+    except Exception as e:
+        return f"ファイルリストの取得に失敗しました: {e}"
+
+# メタデータ並び替え
+def sort_data(data, column):
+    try:
+        if not data or column not in data[0]:
+            return "並び替えに使用するカラムが見つかりません。"
+
+        sorted_data = sorted(data, key=lambda x: x.get(column, ""))
+        return sorted_data
+    except Exception as e:
+        return f"並び替え中にエラーが発生しました: {e}"
+
+# 初期データ（メモリ内で管理）
+metadata = [
+    {"name": "file1", "size": 123, "type": "model"},
+    {"name": "file2", "size": 456, "type": "lora"},
+    {"name": "file3", "size": 789, "type": "model"}
+]
+
+# Gradio UI構築
 with gr.Blocks() as demo:
     gr.Markdown("## Civitai経由ファイルダウンロード＆アップロードツール")
     gr.Markdown("Civitai APIのURLを指定して、'model'または'lora'リポジトリへアップロードします。")
@@ -86,5 +102,26 @@ with gr.Blocks() as demo:
 
     run_button.click(download_and_upload, inputs=[url_input, repo_choice], outputs=output)
 
+    # ファイルリスト表示セクション
+    gr.Markdown("### リポジトリ内のファイルを表示")
+    load_button = gr.Button("ロード")
+    file_list_output = gr.Textbox(label="リポジトリ内のファイル", interactive=False)
+    load_button.click(list_files, inputs=repo_choice, outputs=file_list_output)
+
+    # メタデータ並び替えセクション
+    gr.Markdown("### メタデータ並び替え")
+    metadata_table = gr.Dataframe(value=metadata, headers=["name", "size", "type"], datatype=["str", "number", "str"], label="並び替え対象のデータ")
+    sort_column = gr.Textbox(label="並び替えの基準列 (例: name, size, type)")
+    sort_button = gr.Button("並び替え実行")
+    sorted_table = gr.Dataframe(headers=["name", "size", "type"], datatype=["str", "number", "str"], label="並び替え後のデータ", interactive=False)
+
+    def update_metadata_table(data, column):
+        sorted_result = sort_data(data, column)
+        if isinstance(sorted_result, str):  # エラーメッセージが返ってきた場合
+            return metadata, sorted_result
+        return sorted_result, "並び替えが完了しました。"
+
+    sort_button.click(update_metadata_table, inputs=[metadata_table, sort_column], outputs=[sorted_table, output])
+
 if __name__ == "__main__":
     demo.launch()