Update backend/app.py

backend/app.py

@@ -226,7 +226,8 @@ app.add_middleware(
     expose_headers=["*"]  # Allow access to response headers
 )
 
-OUTPUT_DIR = "outputs"
+# Use /tmp for outputs in Hugging Face Spaces (writable directory)
+OUTPUT_DIR = os.environ.get("OUTPUT_DIR", "/tmp/outputs")
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 # Create image outputs dir
@@ -1000,6 +1001,12 @@ def health():
 
 # Serve frontend only if it has been built; avoid startup failure when dist/ is missing.
 FRONTEND_DIST = os.path.abspath(os.path.join(os.path.dirname(__file__), "../frontend/dist"))
+
+# Check if frontend/dist exists in /app (Docker), otherwise check relative to script location
+if not os.path.isdir(FRONTEND_DIST):
+    # Fallback for Docker: frontend is copied to ./frontend/dist during build
+    FRONTEND_DIST = os.path.join(os.path.dirname(__file__), "frontend/dist")
+
 ASSETS_DIR = os.path.join(FRONTEND_DIST, "assets")
 
 if os.path.isdir(ASSETS_DIR):