original window bounds fixed

app.py

@@ -210,6 +210,10 @@ def get_metric_reference_labels(mode_label: str) -> tuple[str, str]:
     return label_map[left_key], label_map[right_key]
 
 
+def get_metric_reference_keys(mode_label: str) -> tuple[str, str]:
+    return METRIC_COMPARISONS[mode_label]
+
+
 def metric_explainer(title: str, latex: str, text: str) -> None:
     with st.expander(f"About {title}"):
         st.latex(latex)
@@ -568,6 +572,14 @@ def main() -> None:
         "quantized_windowed": quantized_raw,
     }
     histogram_left, histogram_right = get_metric_reference_pair(metric_mode, histogram_images)
+    histogram_left_key, histogram_right_key = get_metric_reference_keys(metric_mode)
+
+    histogram_bounds = {
+        "original": (float(np.min(raw_image)), float(np.max(raw_image))),
+        "quantized": (float(np.min(quantized_raw)), float(np.max(quantized_raw))),
+        "windowed": (low, high),
+        "quantized_windowed": (low, high),
+    }
 
     mse_value = mean_squared_error(metric_left, metric_right)
     psnr_value = peak_signal_to_noise_ratio(metric_left, metric_right, data_range=value_range.span)
@@ -636,8 +648,8 @@ def main() -> None:
                     histogram_right,
                     top_label=left_label,
                     bottom_label=right_label,
-                    low=low,
-                    high=high,
+                    top_bounds=histogram_bounds[histogram_left_key],
+                    bottom_bounds=histogram_bounds[histogram_right_key],
                 ),
                 clear_figure=True,
             )

utils.py

@@ -418,14 +418,20 @@ def create_histogram_figure(
     image_bottom: np.ndarray,
     top_label: str,
     bottom_label: str,
-    low: float,
-    high: float,
+    top_bounds: tuple[float, float],
+    bottom_bounds: tuple[float, float],
 ) -> plt.Figure:
     stacked = np.concatenate([np.asarray(image_top).ravel(), np.asarray(image_bottom).ravel()])
     bins = np.linspace(float(stacked.min()), float(stacked.max()), 129)
 
     fig, axes = plt.subplots(2, 1, figsize=(6, 5.0), sharex=True)
-    for ax, image, label in zip(axes, [image_top, image_bottom], [top_label, bottom_label]):
+    for ax, image, label, bounds in zip(
+        axes,
+        [image_top, image_bottom],
+        [top_label, bottom_label],
+        [top_bounds, bottom_bounds],
+    ):
+        low, high = bounds
         ax.hist(np.asarray(image).ravel(), bins=bins, color="#4C6A92", alpha=0.9)
         ax.axvline(low, color="#B85450", linestyle="--", linewidth=1.4, label="Lower bound")
         ax.axvline(high, color="#B85450", linestyle="--", linewidth=1.4, label="Upper bound")