feat: changed tmp dir for all

run_fastapi.py

@@ -15,31 +15,42 @@ from models.dino import DinoWrapper
 
 from huggingface_hub import hf_hub_download
 
+# Check write permissions
 print("WRITE to ./weights:", os.access("weights", os.W_OK))
 print("WRITE to /tmp:", os.access("/tmp", os.W_OK))
 
-# --- Init app and database
+# --- Set base directory depending on environment
+BASE_DIR = "/tmp" if os.access("/tmp", os.W_OK) else "."
+
+WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
+UPLOADS_DIR = os.path.join(BASE_DIR, "uploads")
+OUTPUTS_DIR = os.path.join(BASE_DIR, "outputs")
+
+# Create directories if not exist
+os.makedirs(WEIGHTS_DIR, exist_ok=True)
+os.makedirs(UPLOADS_DIR, exist_ok=True)
+os.makedirs(OUTPUTS_DIR, exist_ok=True)
+
+# --- Initialize FastAPI and database
 app = FastAPI()
 init_db()
 
-# --- Static and templates
-app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")
+# --- Mount static files and templates
+app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")  # still serve static from project root
 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
 
-# === Model checkpoint setup ===
+# === Download and load model checkpoint ===
 FILENAME = "sam_vit_b_01ec64.pth"
 REPO_ID = "stkrk/sam-vit-b-checkpoint"
-MODEL_DIR = "weights"
-MODEL_PATH = os.path.join(MODEL_DIR, FILENAME)
+MODEL_PATH = os.path.join(WEIGHTS_DIR, FILENAME)
 
-# Download if not exists locally
 if not os.path.exists(MODEL_PATH):
     print(f"Model not found locally. Downloading from {REPO_ID}...")
     cached_path = hf_hub_download(
         repo_id=REPO_ID,
         filename=FILENAME,
-        cache_dir="./weights/.hf_cache",
+        cache_dir=WEIGHTS_DIR,
         local_dir_use_symlinks=False
     )
     shutil.copy(cached_path, MODEL_PATH)
@@ -47,11 +58,11 @@ if not os.path.exists(MODEL_PATH):
 else:
     print(f"Model already exists at {MODEL_PATH}.")
 
-# --- Model initialization (once)
+# --- Initialize models
 sam = SamWrapper(
-        model_type="vit_b",
-        checkpoint_path=MODEL_PATH
-    )
+    model_type="vit_b",
+    checkpoint_path=MODEL_PATH
+)
 dino = DinoWrapper()