Update app.py

app.py

@@ -329,20 +329,11 @@ def compute_shap_difference(shap1_norm, shap2_norm):
     return shap2_norm - shap1_norm
 
 def plot_comparative_heatmap(shap_diff, seq1_length, seq2_length, title="SHAP Difference Heatmap"):
-    """
-    Plot heatmap using both relative positions (0-100%) and actual sequence positions
-    
-    Parameters:
-        shap_diff: numpy array of SHAP differences
-        seq1_length: length of sequence 1
-        seq2_length: length of sequence 2
-        title: plot title
-    """
     heatmap_data = shap_diff.reshape(1, -1)
     extent = max(abs(np.min(shap_diff)), abs(np.max(shap_diff)))
     
-    # Create figure with additional space for the second x-axis
-    fig, ax = plt.subplots(figsize=(12, 2.4))
+    # Create figure with additional space for the second x-axis and colorbar
+    fig, ax = plt.subplots(figsize=(12, 3.2))  # Increased height to accommodate everything
     
     # Plot main heatmap
     cmap = get_zero_centered_cmap()
@@ -379,8 +370,8 @@ def plot_comparative_heatmap(shap_diff, seq1_length, seq2_length, title="SHAP Di
     ax2.set_xticks(tick_positions)
     ax2.set_xticklabels([f"S1: {s1}\nS2: {s2}" for s1, s2 in zip(seq1_labels, seq2_labels)])
     
-    # Add colorbar
-    cbar = plt.colorbar(cax, orientation='horizontal', pad=0.25, aspect=40, shrink=0.8)
+    # Add colorbar with adjusted position
+    cbar = plt.colorbar(cax, orientation='horizontal', pad=0.35, aspect=40, shrink=0.8)
     cbar.ax.tick_params(labelsize=8)
     cbar.set_label('SHAP Difference (Seq2 - Seq1)', fontsize=9, labelpad=5)
     
@@ -390,8 +381,8 @@ def plot_comparative_heatmap(shap_diff, seq1_length, seq2_length, title="SHAP Di
     ax2.set_xlabel('Sequence Positions', fontsize=10)
     ax.set_title(title, pad=10)
     
-    # Adjust layout to prevent label overlap
-    plt.subplots_adjust(bottom=0.35, left=0.05, right=0.95, top=0.85)
+    # Adjust layout to prevent overlap - increased bottom margin
+    plt.subplots_adjust(bottom=0.45, left=0.05, right=0.95, top=0.85)
     
     return fig