Update app.py

app.py

@@ -43,25 +43,51 @@ import gradio as gr
 import json
 import sys
 import gc
+import os
+
+# Ensure the current directory and parent directory are in Python path (for Hugging Face Spaces)
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)
+for dir_path in [current_dir, parent_dir]:
+    if dir_path not in sys.path:
+        sys.path.insert(0, dir_path)
 
 from depth_anything.dpt import DepthAnything
 from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 import torch.nn.functional as F
 from torchvision.transforms import Compose
 
-from config import (
-    TEMP_DIR,
-    MAX_FILE_AGE_HOURS,
-    MAX_IMAGE_SIZE_MB,
-    ALLOWED_IMAGE_FORMATS,
-    MIN_IMAGE_DIMENSION,
-    MAX_IMAGE_DIMENSION,
-    MAX_PROMPT_LENGTH,
-    MAX_NEGATIVE_PROMPT_LENGTH,
-    DEFAULT_NUM_STEPS,
-    DEFAULT_GUIDANCE_SCALE,
-    DEFAULT_SEED,
-)
+# Try to import config, with fallback values if it fails
+try:
+    from config import (
+        TEMP_DIR,
+        MAX_FILE_AGE_HOURS,
+        MAX_IMAGE_SIZE_MB,
+        ALLOWED_IMAGE_FORMATS,
+        MIN_IMAGE_DIMENSION,
+        MAX_IMAGE_DIMENSION,
+        MAX_PROMPT_LENGTH,
+        MAX_NEGATIVE_PROMPT_LENGTH,
+        DEFAULT_NUM_STEPS,
+        DEFAULT_GUIDANCE_SCALE,
+        DEFAULT_SEED,
+    )
+except ImportError:
+    # Fallback configuration if config.py is not found
+    from pathlib import Path
+    TEMP_DIR = Path(os.getenv("TEMP_DIR", "temp_downloads"))
+    TEMP_DIR.mkdir(exist_ok=True, parents=True)
+    MAX_FILE_AGE_HOURS = int(os.getenv("MAX_FILE_AGE_HOURS", "24"))
+    MAX_IMAGE_SIZE_MB = int(os.getenv("MAX_IMAGE_SIZE_MB", "10"))
+    ALLOWED_IMAGE_FORMATS = {".jpg", ".jpeg", ".png", ".webp"}
+    MIN_IMAGE_DIMENSION = 128
+    MAX_IMAGE_DIMENSION = 4096
+    DEFAULT_NUM_STEPS = int(os.getenv("DEFAULT_NUM_STEPS", "30"))
+    DEFAULT_GUIDANCE_SCALE = float(os.getenv("DEFAULT_GUIDANCE_SCALE", "5.0"))
+    DEFAULT_SEED = int(os.getenv("DEFAULT_SEED", "42"))
+    MAX_PROMPT_LENGTH = int(os.getenv("MAX_PROMPT_LENGTH", "500"))
+    MAX_NEGATIVE_PROMPT_LENGTH = int(os.getenv("MAX_NEGATIVE_PROMPT_LENGTH", "500"))
+    print("⚠ Warning: config.py not found, using default configuration values")
 
 
 # ============ GLOBAL CONFIG ============