Upload folder using huggingface_hub

main.py

@@ -1,7 +1,10 @@
 from smolagents import CodeAgent, InferenceClientModel, GradioUI, tool
 from huggingface_hub import HfApi
+from huggingface_hub import snapshot_download
 import requests
 from typing import List, Dict
+from typing import Optional, Union
+from pathlib import Path
 
 @tool
 def leaderboard_search(query: str) -> str:
@@ -101,6 +104,51 @@ def get_space_content(space_id: str) -> str:
     except Exception as e:
         return f"Error accessing space {space_id}: {str(e)}"
     
+@tool
+def download_space_files(
+    space_id: str,
+    revision: Optional[str] = None,
+    local_dir: Optional[Union[str, Path]] = None,
+    allow_patterns: Optional[Union[List[str], str]] = None,
+    ignore_patterns: Optional[Union[List[str], str]] = None,
+    force_download: bool = False,
+    max_workers: int = 8
+) -> str:
+    """
+    Download all files from a Hugging Face Space (snapshot).
+
+    Args:
+        space_id: e.g. "owner/my-awesome-space"
+        revision: branch, tag or commit hash (defaults to default branch)
+        local_dir: where to put the files (defaults to HF cache)
+        allow_patterns: only download files matching these patterns
+        ignore_patterns: skip files matching these patterns
+        force_download: re‑download even if cached
+        max_workers: parallel downloads
+
+    Returns:
+        The local folder path where the space’s files now live.
+    """
+    try:
+        folder = snapshot_download(
+            repo_id=space_id,
+            repo_type="space",
+            revision=revision,
+            local_dir=local_dir,
+            allow_patterns=allow_patterns,
+            ignore_patterns=ignore_patterns,
+            force_download=force_download,
+            max_workers=max_workers
+        )
+        # List how many files were downloaded
+        files = list(Path(folder).rglob("*"))
+        return (
+            f"✔️ Downloaded {len(files)} files from space `{space_id}`\n"
+            f"📂 Local path: {folder}"
+        )
+    except Exception as e:
+        return f"❌ Failed to download space `{space_id}`: {e}"
+    
 
 @tool
 def get_file_from_space(space_id: str, file_path: str) -> str:
@@ -129,7 +177,7 @@ def get_file_from_space(space_id: str, file_path: str) -> str:
 # Initialize the agent with the leaderboard search and space content tools
 model = InferenceClientModel()
 agent = CodeAgent(
-    tools=[leaderboard_search, get_space_content, get_file_from_space],
+    tools=[leaderboard_search, get_space_content, get_file_from_space, download_space_files],
     additional_authorized_imports=["json", "requests", "pandas"],
     model=model,
     add_base_tools=False,

requirements.txt

@@ -1,3 +1,3 @@
 smolagents
 huggingface_hub
-gradio>=5.29.0
+gradio<=3.12.0