Update app.py

names now required in csv

app.py

@@ -29,11 +29,14 @@ def main():
     if uploaded_file:
         df = pd.read_csv(uploaded_file)
         sequences.extend(df['sequence'].tolist())
+        names = df['name'].tolist()  # Store names from the CSV file
+    else:
+        names = [f"Seq {i+1}" for i in range(len(sequences))]  # Default names if no file
 
     results = []
     all_data = {}
     if analyze_pressed:
-        for seq in sequences:
+        for name, seq in zip(names, sequences):
             if validate_sequence(seq):
                 model_results = {}
                 graph_data = {}
@@ -42,10 +45,10 @@ def main():
                     model_results[f"{model_name}_prediction"] = prediction
                     model_results[f"{model_name}_confidence"] = round(confidence, 3)
                     graph_data[model_name] = (prediction, confidence)
-                results.append({"Sequence": seq, **model_results})
-                all_data[seq] = graph_data
+                results.append({"Name": name, "Sequence": seq, **model_results})
+                all_data[name] = graph_data  # Use name as key
             else:
-                st.sidebar.error(f"Invalid sequence: {seq}")
+                st.sidebar.error(f"Invalid sequence for {name}: {seq}")
 
         if results:
             results_df = pd.DataFrame(results)
@@ -54,7 +57,7 @@ def main():
             
             if show_graphs and all_data:
                 st.write("## Graphs")
-                plot_prediction_graphs(all_data)
+                plot_prediction_graphs(all_data,model.keys())
 
 
 if __name__ == "__main__":