Update app.py

app.py

@@ -332,8 +332,8 @@ def plot_comparative_heatmap(shap_diff, seq1_length, seq2_length, title="SHAP Di
     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 and colorbar
-    fig, ax = plt.subplots(figsize=(12, 3.2))  # Increased height to accommodate everything
+    # Create figure with additional space
+    fig, ax = plt.subplots(figsize=(12, 3.2))
     
     # Plot main heatmap
     cmap = get_zero_centered_cmap()
@@ -370,19 +370,22 @@ 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 with adjusted position
-    cbar = plt.colorbar(cax, orientation='horizontal', pad=0.35, aspect=40, shrink=0.8)
+    # Add colorbar at the bottom with more spacing
+    cbar = plt.colorbar(cax, orientation='horizontal', pad=0.6, aspect=40, shrink=0.8)
     cbar.ax.tick_params(labelsize=8)
     cbar.set_label('SHAP Difference (Seq2 - Seq1)', fontsize=9, labelpad=5)
     
+    # Remove percentage labels from colorbar
+    cbar.ax.set_xticklabels([f'{x:.3f}' for x in cbar.ax.get_xticks()])
+    
     # Adjust labels and layout
     ax.set_yticks([])
-    ax.set_xlabel('Relative Position (%)', fontsize=10)
-    ax2.set_xlabel('Sequence Positions', fontsize=10)
+    ax.set_xlabel('Relative Position (%)', fontsize=10, labelpad=10)
+    ax2.set_xlabel('Sequence Positions', fontsize=10, labelpad=10)
     ax.set_title(title, pad=10)
     
-    # Adjust layout to prevent overlap - increased bottom margin
-    plt.subplots_adjust(bottom=0.45, left=0.05, right=0.95, top=0.85)
+    # Increase bottom margin to prevent overlap
+    plt.subplots_adjust(bottom=0.55, left=0.05, right=0.95, top=0.85)
     
     return fig