Update app.py

app.py

@@ -9,13 +9,14 @@ def main():
     st.set_page_config(layout="wide")  # Keep the wide layout for overall flexibility
     st.title("AA Property Inference Demo", anchor=None)
 
-    # Styling for the app to use monospace font
+    # Instructional text below title
     st.markdown("""
         <style>
         .reportview-container {
             font-family: 'Courier New', monospace;
         }
         </style>
+        <p style='font-size:16px;'><span style='font-size:24px;'>&larr;</span> Don't know where to start? Open tab to input a sequence.</p>
         """, unsafe_allow_html=True)
 
     # Input section in the sidebar
@@ -56,7 +57,11 @@ def main():
                 plot_prediction_graphs(all_data)
 
 def plot_prediction_graphs(data):
-    # Function to plot graphs for predictions, divided by prediction value
+    # Create a color palette that is consistent across graphs
+    unique_sequences = sorted(set(seq for seq in data))
+    palette = sns.color_palette("hsv", len(unique_sequences))
+    color_dict = {seq: color for seq, color in zip(unique_sequences, palette)}
+
     for model_name in models.keys():
         fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(16, 6), sharey=True)
         for prediction_val in [0, 1]:
@@ -66,7 +71,8 @@ def plot_prediction_graphs(data):
             sorted_sequences = sorted(filtered_data.items(), key=lambda x: x[1][1], reverse=True)
             sequences = [x[0] for x in sorted_sequences]
             conf_values = [x[1][1] for x in sorted_sequences]
-            sns.barplot(x=sequences, y=conf_values, palette="coolwarm" if prediction_val == 0 else "cubehelix", ax=ax)
+            colors = [color_dict[seq] for seq in sequences]
+            sns.barplot(x=sequences, y=conf_values, palette=colors, ax=ax)
             ax.set_title(f'Confidence Scores for {model_name.capitalize()} (Prediction {prediction_val})')
             ax.set_xlabel('Sequences')
             ax.set_ylabel('Confidence')