feat: clean track names in plot

app.py

@@ -648,14 +648,17 @@ def predict(
     if has_bigwigs and bigwig_selected:
         for tid in bigwig_selected:
             idx = bw_names.index(tid)
-            series.append((tid, bw[:, idx][::stride].astype(float)))
+            # Use clean display name instead of track ID
+            display_name = _get_track_display_name(tid)
+            series.append((display_name, bw[:, idx][::stride].astype(float)))
 
     # Add BED elements if available and selected
     if bed_logits is not None and bed_elements:
         probs = _softmax_last(bed_logits)
         for ename in bed_elements:
+            display_name = ename.replace("_", " ")
             eidx = bed_names.index(ename)
-            series.append((ename, probs[:, eidx, 1][::stride].astype(float)))
+            series.append((display_name, probs[:, eidx, 1][::stride].astype(float)))
 
     tprint("figure data processed created")
 

ntv3_tracks_pipeline.py

@@ -86,7 +86,7 @@ ASSEMBLY_TO_API_URL_TEMPLATE = {
 
 # BED element to color mapping (shared between pipeline and app)
 BED_ELEMENT_COLORS = {
-    "protein_coding_gene": "#E74C3C",  # Red
+    "protein coding gene": "#E74C3C",  # Red
     "lncRNA": "#2ECC71",  # Green
     "exon": "#9B59B6",  # Purple
     "intron": "#F39C12",  # Orange
@@ -94,18 +94,18 @@ BED_ELEMENT_COLORS = {
     "splice_acceptor": "#E67E22",  # Dark orange
     "CTCF-bound": "#3498DB",  # Light blue
     "polyA_signal": "#95A5A6",  # Gray
-    "enhancer_Tissue_specific": "#D35400",  # Dark red
-    "enhancer_Tissue_invariant": "#16A085",  # Dark teal
-    "promoter_Tissue_specific": "#C0392B",  # Dark red 2
-    "promoter_Tissue_invariant": "#27AE60",  # Dark green
+    "enhancer Tissue specific": "#D35400",  # Dark red
+    "enhancer Tissue invariant": "#16A085",  # Dark teal
+    "promoter Tissue specific": "#C0392B",  # Dark red 2
+    "promoter Tissue invariant": "#27AE60",  # Dark green
     "5UTR+": "#8E44AD",  # Dark purple
     "5UTR-": "#D68910",  # Dark orange 2
     "3UTR+": "#138D75",  # Dark teal 2
     "3UTR-": "#2874A6",  # Dark blue
-    "skipped_exon": "#7D3C98",  # Purple 2
-    "always_on_exon": "#A93226",  # Red 2
-    "start_codon": "#196F3D",  # Green 2
-    "stop_codon": "#B9770E",  # Brown
+    "skipped exon": "#7D3C98",  # Purple 2
+    "always on exon": "#A93226",  # Red 2
+    "start codon": "#196F3D",  # Green 2
+    "stop codon": "#B9770E",  # Brown
     "ORF": "#1F618D",  # Blue 2
 }