fix orientation

alignment_plot.py

@@ -289,11 +289,16 @@ def plot_alignment_image(mirna_seq, target_seq, shap_2d, arrows=True):
             fontsize=12, fontweight='bold')
 
     if arrows:
-        ax.add_patch(patches.Arrow(x=0, y=0.2, dx=2, dy=0, width=0.6, color=COLOR_PALETTE[10]))
+        ax.add_patch(patches.Arrow(x=0, y=4.4, dx=2, dy=0, width=0.6, color=COLOR_PALETTE[10]))
         ax.add_patch(patches.Arrow(
-            x=max(alignment_width - 0.2, 0), y=4.4, dx=-2, dy=0, width=0.6, color=COLOR_PALETTE[10]
+            x=max(alignment_width - 0.2, 0), y=0.2, dx=-2, dy=0, width=0.6, color=COLOR_PALETTE[10]
         ))
 
+    ax.text(-0.1, 4.45, "5'", ha='right', va='center', fontsize=10, fontweight='bold')
+    ax.text(alignment_width + 0.15, 4.45, "3'", ha='left', va='center', fontsize=10, fontweight='bold')
+    ax.text(-0.1, 0.15, "3'", ha='right', va='center', fontsize=10, fontweight='bold')
+    ax.text(alignment_width + 0.15, 0.15, "5'", ha='left', va='center', fontsize=10, fontweight='bold')
+
     for i, (target_base, mirna_base, color_index) in enumerate(
         zip(aligned_target, aligned_mirna, color_indices)
     ):