Update app.py

app.py

@@ -4,6 +4,16 @@ from flask import Flask, render_template_string, request, redirect, send_file, j
 from huggingface_hub import HfApi, hf_hub_download, upload_file, delete_file, HfFolder
 from urllib.parse import unquote, quote
 
+# --- FIX FOR PERMISSION ERROR ---
+# In restricted environments (like some containers), the default Hugging Face cache
+# directory (~/.cache/huggingface) might not be writable. We override this by setting
+# the HF_HOME environment variable to a writable directory.
+# This must be done BEFORE any huggingface_hub functions are called.
+HF_CACHE_DIR = os.path.join(tempfile.gettempdir(), "hf_home")
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+os.environ["HF_HOME"] = HF_CACHE_DIR
+# --------------------------------
+
 # --- Environment Setup ---
 # IMPORTANT: Set these environment variables before running the script.
 # You can get a token from https://huggingface.co/settings/tokens
@@ -14,7 +24,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 # REPO_ID = "YourUsername/YourDatasetRepoName"
 # HF_TOKEN = "hf_..."
 
-# Ensure the hf_hub library uses the token
+# Ensure the hf_hub library uses the token. This will now write to the writable HF_HOME dir.
 if HF_TOKEN:
     HfFolder.save_token(HF_TOKEN)
 
@@ -530,7 +540,7 @@ def list_folder(path=""):
                 if dir_path not in seen_dirs:
                     seen_dirs.add(dir_path)
                     items.append({"type": "dir", "name": dir_name, "path": dir_path})
-            elif relative_path and relative_path != '.gitkeep':
+            elif relative_path and relative_path != '.gitkeep' and relative_path != '.keep':
                 items.append({"type": "file", "name": relative_path, "path": f})
         
         items.sort(key=lambda x: (x["type"] != "dir", x["name"].lower()))
@@ -560,7 +570,8 @@ def index():
 def download():
     path = unquote(request.args.get("path", ""))
     try:
-        local_file = hf_hub_download(repo_id=REPO_ID, filename=path, repo_type="dataset", cache_dir=tempfile.gettempdir())
+        # The HF_HOME env var automatically handles the cache location
+        local_file = hf_hub_download(repo_id=REPO_ID, filename=path, repo_type="dataset")
         return send_file(local_file, as_attachment=True, download_name=os.path.basename(path))
     except Exception as e:
         return f"Error downloading file: {e}", 500
@@ -586,24 +597,24 @@ def upload():
 @app.route("/delete", methods=["POST"])
 def delete():
     path_to_delete = request.get_json()["path"].strip("/")
-    # Check if it's a file or folder by seeing if it has a '/' at the end of its relative path
     all_repo_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset")
     
+    # Check if the path is a file or a directory by looking for exact matches (file)
+    # vs. paths that start with the given path + "/" (directory)
     is_file = any(f == path_to_delete for f in all_repo_files)
 
     if is_file:
         delete_file(repo_id=REPO_ID, path_in_repo=path_to_delete, repo_type="dataset", commit_message=f"Delete file {path_to_delete}")
     else: # It's a folder
-        # For folders, we need to delete all files within it.
-        # This is simpler with multiple calls than batching for this app.
         files_in_folder = [f for f in all_repo_files if f.startswith(path_to_delete + "/")]
         for f in files_in_folder:
-             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", commit_message=f"Delete file {f} from folder")
-        # Also delete the .gitkeep if it exists
-        try:
-            delete_file(repo_id=REPO_ID, path_in_repo=f"{path_to_delete}/.gitkeep", repo_type="dataset", commit_message=f"Delete folder placeholder for {path_to_delete}")
-        except Exception:
-            pass # No .gitkeep file, which is fine
+             delete_file(repo_id=REPO_ID, path_in_repo=f, repo_type="dataset", commit_message=f"Delete file {f} from folder {path_to_delete}")
+        # Delete placeholder files if they exist to remove the empty folder
+        for placeholder in ['.gitkeep', '.keep']:
+            try:
+                delete_file(repo_id=REPO_ID, path_in_repo=f"{path_to_delete}/{placeholder}", repo_type="dataset", commit_message=f"Delete folder placeholder for {path_to_delete}", ignore_patterns=None)
+            except Exception:
+                pass # Placeholder did not exist, which is fine
 
     return jsonify(status="ok")
 
@@ -613,7 +624,7 @@ def create_folder():
     keep_file_path = f"{folder_path}/.gitkeep"
     
     with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix=".tmp") as tmp:
-        tmp.write("")
+        tmp.write("") # Create an empty file
         tmp_path = tmp.name
         
     try:
@@ -635,21 +646,24 @@ def rename():
     all_repo_files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset")
     is_file = any(f == old_path for f in all_repo_files)
 
-    files_to_process = []
+    operations = []
     if is_file:
-        files_to_process.append(old_path)
+        operations.append((old_path, new_path))
     else: # It's a folder
-        files_to_process = [f for f in all_repo_files if f.startswith(old_path + "/") or f == f"{old_path}/.gitkeep"]
-
-    for f_path in files_to_process:
-        local_file = hf_hub_download(repo_id=REPO_ID, filename=f_path, repo_type="dataset", cache_dir=tempfile.gettempdir())
-        
-        # Determine the new path in the repository
-        relative_to_old = os.path.relpath(f_path, old_path)
-        new_file_path = os.path.join(new_path, relative_to_old).replace("\\", "/")
-        
-        upload_file(path_or_fileobj=local_file, path_in_repo=new_file_path, repo_id=REPO_ID, repo_type="dataset", commit_message=f"Rename {f_path} to {new_file_path}")
-        delete_file(repo_id=REPO_ID, path_in_repo=f_path, repo_type="dataset")
+        folder_prefix = old_path + "/"
+        for f in all_repo_files:
+            if f.startswith(folder_prefix):
+                new_file_path = new_path + f[len(old_path):]
+                operations.append((f, new_file_path))
+
+    for old_file, new_file in operations:
+        # This move is done server-side on the Hub, no local download needed
+        api.move_file(
+            from_path=old_file,
+            to_path=new_file,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+        )
         
     return jsonify(status="ok")