fix data permission

Dockerfile

@@ -1,11 +1,11 @@
-# Use official Python image for HuggingFace Space
+# HuggingFace Space Docker SDK
+# Use slim Python image - HuggingFace GPU Spaces provide CUDA runtime
 FROM python:3.11-slim
 
 # Set environment variables
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
-ENV GRADIO_SERVER_NAME=0.0.0.0
-ENV GRADIO_SERVER_PORT=7860
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    DEBIAN_FRONTEND=noninteractive
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -14,27 +14,35 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Create non-root user for HuggingFace Space
+# Set up a new user named "user" with user ID 1000 (HuggingFace Space requirement)
 RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user
-ENV PATH=/home/user/.local/bin:$PATH
 
-# Set working directory
+# Create /data directory with proper permissions for persistent storage
+RUN mkdir -p /data && chown user:user /data && chmod 755 /data
+
+# Set environment variables for user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_SERVER_PORT=7860
+
+# Set the working directory
 WORKDIR $HOME/app
 
-# Copy requirements first for better caching
+# Copy requirements first for better Docker layer caching
 COPY --chown=user:user requirements.txt .
 
 # Copy the local nano-vllm package
-COPY --chown=user:user acestep/third_parts/nano-vllm $HOME/app/acestep/third_parts/nano-vllm
+COPY --chown=user:user acestep/third_parts/nano-vllm ./acestep/third_parts/nano-vllm
+
+# Switch to user before installing packages
+USER user
 
-# Install all dependencies from requirements.txt first
-# This includes torch and pre-built flash-attn wheel
+# Install dependencies from requirements.txt (includes PyTorch with CUDA from --extra-index-url)
 RUN pip install --no-cache-dir --user -r requirements.txt
 
 # Install nano-vllm with --no-deps since all dependencies are already installed
-RUN pip install --no-cache-dir --user --no-deps $HOME/app/acestep/third_parts/nano-vllm
+RUN pip install --no-cache-dir --user --no-deps ./acestep/third_parts/nano-vllm
 
 # Copy the rest of the application
 COPY --chown=user:user . .

app.py

@@ -45,11 +45,43 @@ def get_gpu_memory_gb():
         return 0
 
 
+def get_persistent_storage_path():
+    """
+    Detect and return a writable persistent storage path.
+    
+    HuggingFace Space persistent storage requirements:
+    1. Must be enabled in Space settings
+    2. Path is typically /data for Docker SDK
+    3. Falls back to app directory if /data is not writable
+    """
+    # Try HuggingFace Space persistent storage first
+    hf_data_path = "/data"
+    
+    # Check if /data exists and is writable
+    if os.path.exists(hf_data_path):
+        try:
+            test_file = os.path.join(hf_data_path, ".write_test")
+            with open(test_file, 'w') as f:
+                f.write("test")
+            os.remove(test_file)
+            print(f"Using HuggingFace persistent storage: {hf_data_path}")
+            return hf_data_path
+        except (PermissionError, OSError) as e:
+            print(f"Warning: /data exists but is not writable: {e}")
+    
+    # Fall back to app directory (non-persistent but works without special config)
+    fallback_path = os.path.join(current_dir, "data")
+    os.makedirs(fallback_path, exist_ok=True)
+    print(f"Using local storage (non-persistent): {fallback_path}")
+    print("Note: To enable persistent storage, configure it in HuggingFace Space settings")
+    return fallback_path
+
+
 def main():
     """Main entry point for HuggingFace Space"""
     
-    # HuggingFace Space persistent storage path
-    persistent_storage_path = "/data"
+    # Get persistent storage path (auto-detect)
+    persistent_storage_path = get_persistent_storage_path()
     
     # Detect GPU memory for auto-configuration
     gpu_memory_gb = get_gpu_memory_gb()