Move robustness legend beside plot

app.py

@@ -1068,7 +1068,20 @@ def robustness_figure(dataset: str, models: list[str] | None, hover_data: dict |
     fig.update_xaxes(title="Noise fraction")
     metric = metric_text(df["metric"].dropna().iloc[0]) if df["metric"].notna().any() else "score"
     fig.update_yaxes(title=decoding_label(metric))
-    return fig_layout(fig, height=520)
+    fig_layout(fig, height=560)
+    fig.update_layout(
+        margin=dict(l=58, r=170, t=64, b=48),
+        legend=dict(
+            orientation="v",
+            yanchor="top",
+            y=1,
+            xanchor="left",
+            x=1.02,
+            font=dict(size=11),
+            title=None,
+        ),
+    )
+    return fig
 
 
 def robustness_table_frame(dataset: str, models: list[str] | None) -> pd.DataFrame: