Update app.py

app.py

@@ -31,7 +31,7 @@ def handle_token_change(token):
         user_info = api.whoami()
         username = user_info.get('name')
         updates = {
-            manage_files_btn: gr.update(interactive=True), delete_repo_btn: gr.update(interactive=True),
+            manage_files_btn: gr.update(interactive=True), archive_repo_btn: gr.update(interactive=True), delete_repo_btn: gr.update(interactive=True),
             commit_btn: gr.update(interactive=True), author_input: gr.update(value=username),
             download_author_input: gr.update(value=username), whoami_output: gr.update(value=user_info, visible=True)
         }
@@ -39,7 +39,7 @@ def handle_token_change(token):
     except HfHubHTTPError as e:
         gr.Warning(f"Invalid Token: {e}. You can only perform read-only actions.")
         updates = {
-            manage_files_btn: gr.update(interactive=False), delete_repo_btn: gr.update(interactive=False),
+            manage_files_btn: gr.update(interactive=False), archive_repo_btn: gr.update(interactive=False), delete_repo_btn: gr.update(interactive=False),
             commit_btn: gr.update(interactive=False), whoami_output: gr.update(value=None, visible=False)
         }
         return (token, *updates.values())
@@ -93,6 +93,20 @@ def delete_repo(token, repo_id, repo_type):
         gr.Error(f"Failed to delete repository: {e}")
         return repo_id, gr.update(visible=True), gr.update(visible=False)
 
+def archive_repo(token, repo_id, archive_repo):
+    """Deletes the selected repository."""
+    if not token:
+        gr.Error("A write-enabled Hugging Face token is required to delete a repository.")
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
+    try:
+        api = get_hf_api(token)
+        archive_space=f'{archive_repo}/{repo_id.split("/")[1]}'
+        api.move_repo(repo_id=repo_id, archive_space=archive_space)
+        gr.Info(f"Successfully moved to '{archive_space}'.")
+        return None, gr.update(visible=False), gr.update(visible=False)
+    except HfHubHTTPError as e:
+        gr.Error(f"Failed to archive repository: {e}")
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
 # --- File Management Functions ---
 
 def show_files_and_load_first(token, repo_id, repo_type):
@@ -206,6 +220,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
                         gr.Markdown("### 2. Choose an Action")
                         manage_files_btn = gr.Button("Manage Files", interactive=False)
                         delete_repo_btn = gr.Button("Delete This Repo", variant="stop", interactive=False)
+                        archive_repo_name = gr.Textbox(label="Archive",placeholder="my_archive, will append /repo_name")
+                        archive_repo_btn = gr.Button("Send to Archive", variant="secondary", interactive=False)
                     with gr.Column(visible=False) as editor_panel:
                         gr.Markdown("### 3. Edit Files")
                         file_selector = gr.Dropdown(label="Select File", interactive=True)
@@ -236,14 +252,18 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
 
     # --- Global and Cross-Tab Event Handlers ---
     hf_token.change(fn=handle_token_change, inputs=hf_token,
-                    outputs=[hf_token_state, manage_files_btn, delete_repo_btn, commit_btn, author_input, download_author_input, whoami_output])
+                    outputs=[hf_token_state, manage_files_btn, archive_repo_btn, delete_repo_btn, commit_btn, author_input, download_author_input, whoami_output])
 
     manage_repo_dropdown.select(fn=on_manage_repo_select, inputs=manage_repo_dropdown, outputs=action_panel)
     
     manage_files_btn.click(fn=show_files_and_load_first,
                            inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
                            outputs=[editor_panel, file_selector, code_editor])
-
+    
+    archive_repo_btn.click(fn=archive_repo, inputs=[hf_token_state, manage_repo_dropdown, archive_repo_name],
+                          outputs=[manage_repo_dropdown, action_panel, editor_panel],
+                          js="() => confirm('Are you sure you want to archive this repo?')")
+    
     delete_repo_btn.click(fn=delete_repo, inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
                           outputs=[manage_repo_dropdown, action_panel, editor_panel],
                           js="() => confirm('Are you sure you want to permanently delete this repository?')")