Spaces:
Running
Running
File size: 2,273 Bytes
4862c84 b369b7e 4862c84 b369b7e 4862c84 b369b7e 4862c84 b369b7e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
"""
Utilities for persistent storage in Hugging Face Spaces.
"""
import os
from pathlib import Path
from typing import Optional
def get_persistent_data_dir() -> Optional[Path]:
"""Get the persistent data directory if available.
Returns:
Path to persistent storage directory if available, None otherwise.
"""
if os.path.isdir("/data"):
data_dir = Path("/data/app_data")
data_dir.mkdir(exist_ok=True)
return data_dir
return None
def get_cache_dir() -> Path:
"""Get the appropriate cache directory (persistent if available, temp otherwise).
Returns:
Path to cache directory.
"""
if os.path.isdir("/data"):
cache_dir = Path("/data/.cache")
cache_dir.mkdir(exist_ok=True)
return cache_dir
else:
# Fallback to temp directory
import tempfile
return Path(tempfile.gettempdir()) / "app_cache"
def save_uploaded_file(uploaded_file, filename: str) -> Optional[Path]:
"""Save an uploaded file to persistent storage.
Args:
uploaded_file: Gradio uploaded file object
filename: Name to save the file as
Returns:
Path to saved file if successful, None otherwise.
"""
persistent_dir = get_persistent_data_dir()
if persistent_dir and uploaded_file:
save_path = persistent_dir / filename
save_path.parent.mkdir(parents=True, exist_ok=True)
# Copy the uploaded file to persistent storage
import shutil
shutil.copy2(uploaded_file, save_path)
return save_path
return None
def is_persistent_storage_available() -> bool:
"""Check if persistent storage is available.
Returns:
True if persistent storage is available, False otherwise.
"""
return os.path.isdir("/data")
def get_persistent_results_dir() -> Optional[Path]:
"""Get the persistent results directory for storing pipeline results.
Returns:
Path to persistent results directory if available, None otherwise.
"""
persistent_dir = get_persistent_data_dir()
if persistent_dir:
results_dir = persistent_dir / "results"
results_dir.mkdir(exist_ok=True)
return results_dir
return None |