fixing bucket issue

Dockerfile

@@ -28,7 +28,8 @@ COPY . .
 
 # Pre-create /data so local Docker runs work; on HF Spaces the volume is mounted
 # here at runtime (owned by root, container also runs as root — no permission issue)
-RUN mkdir -p /data
+#RUN mkdir -p /data
+RUN mkdir -p /data && chmod 777 /data
 
 # Add venv to PATH
 ENV PATH="/home/app/.venv/bin:$PATH" \

README.md

@@ -7,6 +7,11 @@ sdk: docker
 app_port: 7860
 pinned: false
 short_description: Conflict-free batch scheduling engine for GeeksforGeeks
+volumes:
+    type: "bucket"
+    source: "arcshukla/ai-scheduling-platform-storage"
+    mount_path: "/app/data"
+
 ---
 
 # GFG SmartSched

main.py

@@ -6,19 +6,33 @@ setup_logging(settings.log_level)
 logger = get_logger(__name__)
 
 # ── Now safe to import everything else (logs visible from here on) ────────────
+import uuid
 from contextlib import asynccontextmanager
 
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
+from starlette.middleware.base import BaseHTTPMiddleware
 
 from app.constants import APP_NAME, APP_VERSION
 from app.db.database import init_db
+from app.logging_config import set_request_id, clear_request_id
 
 logger.info("Importing routers...")
 from app.routers import router
 logger.info("Routers loaded")
 
 
+class RequestIdMiddleware(BaseHTTPMiddleware):
+    """Injects a short request ID into the logging context for every request."""
+    async def dispatch(self, request: Request, call_next):
+        req_id = request.headers.get("X-Request-ID") or uuid.uuid4().hex[:8]
+        set_request_id(req_id)
+        response = await call_next(request)
+        response.headers["X-Request-ID"] = req_id
+        clear_request_id()
+        return response
+
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     logger.info("Starting %s v%s (log_level=%s)", APP_NAME, APP_VERSION, settings.log_level)
@@ -31,6 +45,7 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(title=APP_NAME, version=APP_VERSION, lifespan=lifespan)
 
+app.add_middleware(RequestIdMiddleware)
 app.mount("/static", StaticFiles(directory="app/static"), name="static")
 
 app.include_router(router)

test_data/hf_logs.py

@@ -0,0 +1,127 @@
+#!/usr/bin/env python
+"""
+hf_logs.py  —  Stream Hugging Face Spaces logs to the terminal.
+
+Usage
+-----
+  uv run python test_data/hf_logs.py              # build + run logs (default)
+  uv run python test_data/hf_logs.py --kind run   # runtime logs only
+  uv run python test_data/hf_logs.py --kind build # build logs only
+  uv run python test_data/hf_logs.py --status     # space status + last error
+  uv run python test_data/hf_logs.py --restart    # restart the space, then stream run logs
+
+Reads HF_TOKEN and HF_SPACE_NAME from .env (falls back to env vars).
+"""
+import argparse
+import json
+import os
+import sys
+import urllib.request
+from dotenv import load_dotenv
+
+load_dotenv()
+
+# ── Helpers ────────────────────────────────────────────────────────────────────
+
+def _token() -> str:
+    t = os.environ.get("HF_TOKEN", "")
+    if not t:
+        sys.exit("HF_TOKEN not set — add it to .env")
+    return t
+
+def _space() -> str:
+    s = os.environ.get("HF_SPACE_NAME", "")
+    if not s:
+        sys.exit("HF_SPACE_NAME not set — add it to .env")
+    return s
+
+def _req(url: str, method: str = "GET") -> urllib.request.Request:
+    return urllib.request.Request(
+        url, method=method,
+        headers={
+            "Authorization": f"Bearer {_token()}",
+            "Content-Length": "0",
+        },
+    )
+
+def _api(path: str) -> str:
+    return f"https://huggingface.co/api/spaces/{_space()}/{path}"
+
+
+# ── Commands ───────────────────────────────────────────────────────────────────
+
+def cmd_status():
+    with urllib.request.urlopen(_req(_api("")), timeout=10) as r:
+        d = json.loads(r.read())
+    rt = d.get("runtime", {})
+    hw = rt.get("hardware", {})
+    print(f"Space  : {_space()}")
+    print(f"SHA    : {d.get('sha','')[:12]}")
+    print(f"Stage  : {rt.get('stage')}")
+    print(f"HW     : {hw.get('requested')} (current: {hw.get('current')})")
+    err = rt.get("errorMessage")
+    if err:
+        print(f"Error  : {err}")
+
+
+def cmd_restart():
+    with urllib.request.urlopen(_req(_api("restart"), method="POST"), timeout=10) as r:
+        d = json.loads(r.read())
+    print(f"Restarted → stage: {d.get('stage')}")
+
+
+def cmd_stream(kind: str):
+    """Stream SSE log lines for the given kind ('build' or 'run')."""
+    url = _api(f"logs/{kind}")
+    print(f"=== {_space()} / {kind} logs ===")
+    try:
+        with urllib.request.urlopen(_req(url), timeout=120) as r:
+            while True:
+                raw = r.readline()
+                if not raw:
+                    break
+                line = raw.decode("utf-8", errors="replace").strip()
+                if not line:
+                    continue
+                if line.startswith("data:"):
+                    payload = line[5:].strip()
+                    try:
+                        msg = json.loads(payload).get("data", "")
+                        if msg:
+                            print(msg)
+                    except json.JSONDecodeError:
+                        if payload:
+                            print(payload)
+    except KeyboardInterrupt:
+        print("\n[interrupted]")
+    except Exception as e:
+        print(f"[stream ended: {e}]")
+
+
+# ── Main ───────────────────────────────────────────────────────────────────────
+
+def main():
+    parser = argparse.ArgumentParser(description="HF Spaces log viewer")
+    parser.add_argument("--kind",    choices=["build", "run", "both"], default="both",
+                        help="Which log stream to show (default: both)")
+    parser.add_argument("--status",  action="store_true", help="Show space status and exit")
+    parser.add_argument("--restart", action="store_true", help="Restart the space then stream run logs")
+    args = parser.parse_args()
+
+    if args.status:
+        cmd_status()
+        return
+
+    if args.restart:
+        cmd_restart()
+        args.kind = "run"       # fall through to stream
+
+    if args.kind == "both":
+        cmd_stream("build")
+        cmd_stream("run")
+    else:
+        cmd_stream(args.kind)
+
+
+if __name__ == "__main__":
+    main()