Hugging Face's logo

Spaces:
tokev
/
traffic-visualizer
Running

App Files Community
traffic-visualizer
File size: 3,150 Bytes
5893134
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8a8749d
 
5893134
 
 
 
 
 
 
 
 
8a8749d
5893134
 
 
94d050a
 
5893134
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
# ── Single-Space: CityFlow + Visualizer Dashboard ──────────────────────────
#
# Two-stage build:
#   1. builder  - compiles the vendored CityFlow Python extension
#   2. runtime  - installs API + visualizer dependencies, serves the PIXI.js
#                 dashboard via server/visualizer_app.py
#
# Runtime env vars:
#   DATA_DIR          generated CityFlow dataset root (default: /app/data/generated)
#   REPLAY_ROOT       on-disk replay cache            (default: /app/results/replays)
#   CHECKPOINT_PATH   DQN checkpoint                  (default: /app/artifacts/dqn_shared/best_validation.pt)
# ---------------------------------------------------------------------------

# ── Stage 1: Build CityFlow ─────────────────────────────────────────────────
FROM python:3.12-slim AS builder

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential \
        cmake \
        libboost-all-dev \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /build
COPY third_party/CityFlow ./CityFlow
RUN rm -rf ./CityFlow/build
RUN pip install --no-cache-dir ./CityFlow


# ── Stage 2: Runtime ────────────────────────────────────────────────────────
FROM python:3.12-slim AS runtime

WORKDIR /app

# CityFlow compiled extension
COPY --from=builder /usr/local/lib/python3.12/site-packages/cityflow* \
                    /usr/local/lib/python3.12/site-packages/

# Python dependencies
# openenv_app/requirements.txt: fastapi, uvicorn, torch, openenv-core, etc.
# server/requirements.txt: joblib, python-multipart (extras not in openenv_app)
COPY openenv_app/requirements.txt ./openenv_app/requirements.txt
COPY server/requirements.txt      ./server/requirements.txt
RUN pip install --no-cache-dir \
      -r openenv_app/requirements.txt \
      -r server/requirements.txt

# Application source
COPY agents/         ./agents/
COPY district_llm/   ./district_llm/
COPY env/            ./env/
COPY openenv_app/    ./openenv_app/
COPY server/         ./server/
COPY training/       ./training/

# PIXI.js frontend
COPY third_party/CityFlow/frontend/ ./third_party/CityFlow/frontend/

# City data (all bundled cities)
COPY data/bundled/ ./data/bundled/

# Artifacts
COPY artifacts/dqn_shared/best_validation.pt \
     ./artifacts/dqn_shared/best_validation.pt
COPY artifacts/district_llm_adapter_v3/main_run/adapter/ \
     ./artifacts/district_llm_adapter_v3/main_run/adapter/

RUN mkdir -p /app/results/replays

ENV DATA_DIR=/app/data/bundled
ENV REPLAY_ROOT=/app/results/replays
ENV CHECKPOINT_PATH=/app/artifacts/dqn_shared/best_validation.pt
ENV DISTRICT_LLM_ADAPTER_PATH=/app/artifacts/district_llm_adapter_v3/main_run/adapter
ENV TRANSFORMERS_OFFLINE=1
ENV HF_HUB_OFFLINE=1

EXPOSE 7860

CMD ["sh", "-c", "uvicorn server.visualizer_app:app --host 0.0.0.0 --port ${PORT:-7860}"]