gallery fix

app.py

@@ -26,6 +26,20 @@ os.environ["HTTPX_VERIFY"] = "0"
 os.environ["UI_PORT"] = "7860"  # Standard Spaces port
 os.environ["HF_SPACES"] = "1"  # Flag to indicate we're running in Spaces
 
+# Configure persistent data directory for Spaces
+# Use /home/user/app/data for persistent storage in Hugging Face Spaces
+persistent_data_dir = "/home/user/app/data"
+os.environ["DATA_DIR"] = persistent_data_dir  # This is critical for gallery to work
+os.environ["IMAGE_OUTPUT_DIR"] = f"{persistent_data_dir}/images"
+os.environ["MODEL_OUTPUT_DIR"] = f"{persistent_data_dir}/models"
+logger.info(f"Set persistent data directory to {persistent_data_dir}")
+
+# Create directories if they don't exist
+for subdir in ["", "images", "models", "downloads"]:
+    dir_path = f"{persistent_data_dir}/{subdir}" if subdir else persistent_data_dir
+    os.makedirs(dir_path, exist_ok=True)
+    logger.info(f"Ensured directory exists: {dir_path}")
+
 # Set model environment variables explicitly for Hugging Face Spaces
 # These will override any variables loaded from .env.spaces
 os.environ["MODEL_ID"] = (

app/core/pipeline.py

@@ -116,17 +116,42 @@ class CreativePipeline:
         logger.info(f"Initializing ImageTo3DGenerator with app_id: {i2m_app_id}")
         self.image_to_3d = ImageTo3DGenerator(stub, app_id=i2m_app_id)
 
-        # Get directory paths based on environment or defaults
-        if spaces_mode:
-            data_root = os.environ.get("DATA_DIR", "/tmp/data")
-            data_dir = Path(data_root)
+        # Get directory paths based on environment variables or defaults
+        # First check for specific environment variables
+        if os.environ.get("DATA_DIR"):
+            # Use persistent data directory from environment
+            data_dir = Path(os.environ.get("DATA_DIR"))
+            logger.info(f"Using DATA_DIR environment variable: {data_dir}")
+        elif spaces_mode:
+            # Use persistent location for Hugging Face Spaces
+            data_dir = Path("/home/user/app/data")
+            logger.info(f"Using persistent storage in Spaces: {data_dir}")
         else:
+            # Fall back to default local path
             data_dir = Path(__file__).parent.parent / "data"
+            logger.info(f"Using default local data path: {data_dir}")
 
         # Ensure app/data directories exist
         data_dir.mkdir(exist_ok=True)
-        (data_dir / "images").mkdir(exist_ok=True)
-        (data_dir / "models").mkdir(exist_ok=True)
+
+        # Use specific output directories if provided in environment variables
+        if os.environ.get("IMAGE_OUTPUT_DIR"):
+            images_dir = Path(os.environ.get("IMAGE_OUTPUT_DIR"))
+            logger.info(f"Using IMAGE_OUTPUT_DIR from environment: {images_dir}")
+        else:
+            images_dir = data_dir / "images"
+            logger.info(f"Using default images directory: {images_dir}")
+
+        if os.environ.get("MODEL_OUTPUT_DIR"):
+            models_dir = Path(os.environ.get("MODEL_OUTPUT_DIR"))
+            logger.info(f"Using MODEL_OUTPUT_DIR from environment: {models_dir}")
+        else:
+            models_dir = data_dir / "models"
+            logger.info(f"Using default models directory: {models_dir}")
+
+        # Create subdirectories
+        images_dir.mkdir(exist_ok=True)
+        models_dir.mkdir(exist_ok=True)
         (data_dir / "downloads").mkdir(exist_ok=True)
 
         logger.info("Creative pipeline initialized successfully")

app/ui/app.py

@@ -60,19 +60,43 @@ IMAGE_TO_3D_APP_ID = os.environ.get(
 def main():
     """AI Creative application interface"""
 
-    # Paths for saving generated content - adjust for HF Spaces environment
-    if RUNNING_IN_SPACES:
-        data_path = Path("/tmp/data")
+    # Paths for saving generated content - prefer environment variable for persistent storage
+    data_path = None
+
+    # Check for DATA_DIR environment variable first (set in the app.py entry point)
+    if os.environ.get("DATA_DIR"):
+        data_path = Path(os.environ.get("DATA_DIR"))
+        logger.info(f"Using DATA_DIR environment variable: {data_path}")
+    # Fall back to default paths otherwise
+    elif RUNNING_IN_SPACES:
+        data_path = Path("/home/user/app/data")  # Persistent storage in Spaces
+        logger.info(f"Using persistent storage in Spaces: {data_path}")
     else:
         data_path = Path(__file__).parent.parent / "data"
+        logger.info(f"Using default local data path: {data_path}")
 
-    images_path = data_path / "images"
-    models_path = data_path / "models"
+    # Use environment variables for image and model output dirs if specified
+    if os.environ.get("IMAGE_OUTPUT_DIR"):
+        images_path = Path(os.environ.get("IMAGE_OUTPUT_DIR"))
+        logger.info(f"Using IMAGE_OUTPUT_DIR from environment: {images_path}")
+    else:
+        images_path = data_path / "images"
+        logger.info(f"Using default images path: {images_path}")
+
+    if os.environ.get("MODEL_OUTPUT_DIR"):
+        models_path = Path(os.environ.get("MODEL_OUTPUT_DIR"))
+        logger.info(f"Using MODEL_OUTPUT_DIR from environment: {models_path}")
+    else:
+        models_path = data_path / "models"
+        logger.info(f"Using default models path: {models_path}")
 
     # Ensure necessary directories exist
     images_path.mkdir(exist_ok=True, parents=True)
     models_path.mkdir(exist_ok=True, parents=True)
 
+    logger.info(f"Image gallery path: {images_path}")
+    logger.info(f"Model gallery path: {models_path}")
+
     # Initialize pipeline only if modules are available
     pipeline = None
     if CORE_MODULES_AVAILABLE: