Deploy Space: update src

src/utils.py

@@ -463,8 +463,17 @@ def make_forecast_plot(snapshot: dict | None) -> object | None:
     n_ctx = len(ctx)
     n_fut = len(actuals)
 
-    ctx_x = list(range(n_ctx))
-    fut_x = list(range(n_ctx, n_ctx + n_fut))
+    timestamps = snapshot.get("timestamps", [])
+    if timestamps and len(timestamps) == n_ctx + n_fut:
+        ctx_x = timestamps[:n_ctx]
+        fut_x = timestamps[n_ctx:]
+        x_title = "Time"
+        vline_x = fut_x[0]
+    else:
+        ctx_x = list(range(n_ctx))
+        fut_x = list(range(n_ctx, n_ctx + n_fut))
+        x_title = "Time step"
+        vline_x = n_ctx - 0.5
 
     fig = go.Figure()
 
@@ -513,24 +522,24 @@ def make_forecast_plot(snapshot: dict | None) -> object | None:
             text=f"{dataset_label} — {model_label} ({evaluated_at})",
             font=dict(size=14),
         ),
-        xaxis_title="Time step",
+        xaxis_title=x_title,
         yaxis_title="Value",
         legend=dict(orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1),
         margin=dict(l=40, r=20, t=60, b=40),
         height=350,
         paper_bgcolor="rgba(0,0,0,0)",
         plot_bgcolor="rgba(0,0,0,0)",
-        font=dict(color="#e2e8f0"),
-        xaxis=dict(gridcolor="rgba(255,255,255,0.08)"),
-        yaxis=dict(gridcolor="rgba(255,255,255,0.08)"),
+        font=dict(color="black"),
+        xaxis=dict(gridcolor="rgba(0,0,0,0.08)"),
+        yaxis=dict(gridcolor="rgba(0,0,0,0.08)"),
     )
     # Vertical line separating context from forecast
     fig.add_vline(
-        x=n_ctx - 0.5,
+        x=vline_x,
         line_dash="dot",
-        line_color="rgba(255,255,255,0.3)",
+        line_color="rgba(0,0,0,0.3)",
         annotation_text="forecast start",
-        annotation_font_color="#94a3b8",
+        annotation_font_color="black",
     )
     return fig
 
@@ -578,8 +587,8 @@ def make_rank_trend_plot(history_df: pd.DataFrame, metrics: list[str] | None = N
         height=320,
         paper_bgcolor="rgba(0,0,0,0)",
         plot_bgcolor="rgba(0,0,0,0)",
-        font=dict(color="#e2e8f0"),
-        xaxis=dict(gridcolor="rgba(255,255,255,0.08)"),
-        yaxis=dict(gridcolor="rgba(255,255,255,0.08)"),
+        font=dict(color="black"),
+        xaxis=dict(gridcolor="rgba(0,0,0,0.08)"),
+        yaxis=dict(gridcolor="rgba(0,0,0,0.08)"),
     )
     return fig