Merge pull request #20 from Axiomatic-AI/plot-spectrum-hline-fix

fixed hline bug in plot spectrum

Author: lsarra-ax

src/axiomatic/pic_helpers.py

@@ -97,7 +97,7 @@ def plot_single_spectrum(
         )  # Add vertical line
     for y_val in hlines:
         plt.axhline(
-            x=y_val, color="red", linestyle="--", label=f"Transmission (y={y_val})"
+            y=y_val, color="red", linestyle="--", label=f"Transmission (y={y_val})"
         )  # Add vertical line
     return plt.gcf()
 
@@ -141,6 +141,9 @@ def plot_interactive_spectra(
     # Calculate global y-limits across all arrays
     y_min = min(min(min(arr2) for arr2 in arr1) for arr1 in spectra)
     y_max = max(max(max(arr2) for arr2 in arr1) for arr1 in spectra)
+    if hlines:
+        y_min = min(hlines + [y_min])*0.95
+        y_max = max(hlines + [y_max])*1.05
 
     slider_index = slider_index or list(range(len(spectra[0])))
     spectrum_labels = spectrum_labels or [f"Spectrum {i}" for i in range(len(spectra))]
@@ -159,7 +162,7 @@ def plot_array(index=0):
             )  # Add vertical line
         for y_val in hlines:
             plt.axhline(
-                x=y_val, color="red", linestyle="--", label=f"Transmission (y={y_val})"
+                y=y_val, color="red", linestyle="--", label=f"Transmission (y={y_val})"
             )  # Add vertical line
         plt.title(f"Iteration: {index}")
         plt.xlabel("X")