Update app.py

app.py

@@ -21,13 +21,9 @@ def main():
     # Input section in the sidebar
     sequence = st.sidebar.text_input("Enter your amino acid sequence:")
     uploaded_file = st.sidebar.file_uploader("Or upload a CSV file with amino acid sequences", type="csv")
-    
-    # Button to analyze sequence
     analyze_pressed = st.sidebar.button("Analyze Sequence")
-    
-    # Show graphs toggle
     show_graphs = st.sidebar.checkbox("Show Prediction Graphs")
-    
+
     sequences = [sequence] if sequence else []
     if uploaded_file:
         df = pd.read_csv(uploaded_file)
@@ -54,24 +50,26 @@ def main():
             results_df = pd.DataFrame(results)
             st.write("### Results")
             st.dataframe(results_df.style.format(precision=3), width=None, height=None)
-
-    # Display graphs in sidebar if toggle is active
-    if show_graphs and all_data:
-        plot_prediction_graphs(all_data)
+            
+            if show_graphs and all_data:
+                st.write("## Graphs")
+                plot_prediction_graphs(all_data)
 
 def plot_prediction_graphs(data):
-    # Function to plot graphs for predictions in the sidebar
+    # Function to plot graphs for predictions
     for model_name in models.keys():
         plt.figure(figsize=(10, 4))
-        predictions = {seq: values[model_name][0] for seq, values in data.items()}
-        confidences = {seq: values[model_name][1] for seq, values in data.items()}
-        sequences = list(predictions.keys())
-        conf_values = list(confidences.values())
+        predictions = {seq: values[model_name][1] for seq, values in data.items()}  # Using confidence for ordering
+        # Sorting sequences based on confidence, descending
+        sorted_sequences = sorted(predictions.items(), key=lambda x: x[1], reverse=True)
+        sequences = [x[0] for x in sorted_sequences]
+        conf_values = [x[1] for x in sorted_sequences]
         sns.barplot(x=sequences, y=conf_values, palette="viridis")
         plt.title(f'Confidence Scores for {model_name.capitalize()} Model')
         plt.xlabel('Sequences')
         plt.ylabel('Confidence')
-        st.sidebar.pyplot(plt)  # Display each plot in the sidebar
+        plt.xticks(rotation=45)  # Rotate x labels for better visibility
+        st.pyplot(plt)  # Display each plot below the results table
 
 if __name__ == "__main__":
     main()