Update app.py

app.py

@@ -439,12 +439,14 @@ elif selected_model == 'Cas12':
             predictions, gene_sequence, exons = cas12.process_gene(gene_symbol,cas12_path)
             sorted_predictions = sorted(predictions, key=lambda x: x[-1], reverse=True)[:10]
             st.session_state['on_target_results'] = sorted_predictions
+            st.session_state['gene_sequence'] = gene_sequence  # Save gene sequence in session state
+            st.session_state['exons'] = exons  # Store exon data
         st.success('Prediction completed!')
 
         # Visualization and file generation
         if 'on_target_results' in st.session_state and st.session_state['on_target_results']:
             df = pd.DataFrame(st.session_state['on_target_results'],
-                              columns=["Gene ID", "Start Pos", "End Pos", "Strand", "Target", "gRNA", "Prediction"])
+                              columns=["Chr", "Start Pos", "End Pos", "Strand", "Transcript", "Exon", "Target", "gRNA", "Prediction"])
             st.dataframe(df)
             # Now create a Plotly plot with the sorted_predictions
             fig = go.Figure()