minor fixes

Author: ferranAX

src/axiomatic/pic_helpers.py

@@ -121,10 +121,12 @@ def plot_interactive_spectra(
 
     # Defaults
     if spectrum_labels is None and isinstance(spectra, dict):
-        if isinstance(list(spectra.keys())[0], str) and all("," in key for key in spectra.keys()):
+        if all(isinstance(key, str) and "," in key for key in spectra.keys()):
+            spectrum_labels = [f"T {ports.split(',')[0]} -> {ports.split(',')[1]}" for ports in spectra.keys()]
+        elif all(isinstance(key, tuple) and len(key) == 2 for key in spectra.keys()):
             spectrum_labels = [f"T {port_in} -> {port_out}" for port_in, port_out in spectra.keys()]
-    elif isinstance(list(spectra.keys())[0], tuple) and len(list(spectra.keys())[0]) == 2:
-        spectrum_labels = [f"T {port_in} -> {port_out}" for port_in, port_out in spectra.keys()]
+        else:
+            raise ValueError("Invalid keys for spectra dictionary.")
 
     elif spectrum_labels is None:
         spectrum_labels = [f"Spectrum {i}" for i in range(len(spectra))]