fix

app.py

@@ -119,6 +119,16 @@ if selected_model == 'Cas9':
                 df = pd.DataFrame(st.session_state['on_target_results'],
                                   columns=["Gene ID", "Start Pos", "End Pos", "Strand", "Target", "gRNA", "Prediction"])
                 st.dataframe(df)
+                # Define file paths
+                genbank_file_path = f"{gene_symbol}_crispr_targets.gb"
+                bed_file_path = f"{gene_symbol}_crispr_targets.bed"
+                csv_file_path = f"{gene_symbol}_crispr_predictions.csv"
+
+                # Generate files
+                cas9on.generate_genbank_file_from_df(df, gene_sequence, gene_symbol, genbank_file_path)
+                cas9on.create_bed_file_from_df(df, bed_file_path)
+                cas9on.create_csv_from_df(df, csv_file_path)
+
                 # Now create a Plotly plot with the sorted_predictions
                 fig = go.Figure()
 
@@ -174,42 +184,28 @@ if selected_model == 'Cas9':
                     ),
                     showlegend=False  # Hide the legend if it's not necessary
                 )
-
                 # Display the plot
                 st.plotly_chart(fig)
+                # Download buttons
+                col1, col2, col3 = st.columns(3)
+                with col1:
+                    with open(genbank_file_path, "rb") as file:
+                        st.download_button(label="Download GenBank File", data=file, file_name=genbank_file_path,
+                                           mime="text/x-genbank")
+                with col2:
+                    with open(bed_file_path, "rb") as file:
+                        st.download_button(label="Download BED File", data=file, file_name=bed_file_path,
+                                           mime="text/plain")
+                with col3:
+                    with open(csv_file_path, "rb") as file:
+                        st.download_button(label="Download CSV File", data=file, file_name=csv_file_path,
+                                           mime="text/csv")
+
+                # Links for user guidance on using the downloaded files
+                st.markdown(
+                    "GenBank files can be visualized using [PyGenomeViz](https://pygenomeviz.streamlit.app/). "
+                    "BED files can be used with the [UCSC Genome Browser](https://genome.ucsc.edu/cgi-bin/hgCustom).")
 
-                if 'gene_sequence' in st.session_state and st.session_state['gene_sequence']:
-                    gene_sequence = st.session_state['gene_sequence']
-
-                    # Define file paths
-                    genbank_file_path = f"{gene_symbol}_crispr_targets.gb"
-                    bed_file_path = f"{gene_symbol}_crispr_targets.bed"
-                    csv_file_path = f"{gene_symbol}_crispr_predictions.csv"
-
-                    # Generate files
-                    cas9on.generate_genbank_file_from_df(df, gene_sequence, gene_symbol, genbank_file_path)
-                    cas9on.create_bed_file_from_df(df, bed_file_path)
-                    cas9on.create_csv_from_df(df, csv_file_path)
-
-                    # Layout for download buttons
-                    col1, col2, col3 = st.columns(3)
-                    with col1:
-                        with open(genbank_file_path, "rb") as file:
-                            st.download_button(label="Download GenBank File", data=file, file_name=genbank_file_path,
-                                               mime="text/x-genbank")
-                    with col2:
-                        with open(bed_file_path, "rb") as file:
-                            st.download_button(label="Download BED File", data=file, file_name=bed_file_path,
-                                               mime="text/plain")
-                    with col3:
-                        with open(csv_file_path, "rb") as file:
-                            st.download_button(label="Download CSV File", data=file, file_name=csv_file_path,
-                                               mime="text/csv")
-
-                    # Links for user guidance on using the downloaded files
-                    st.markdown(
-                        "GenBank files can be visualized using [PyGenomeViz](https://pygenomeviz.streamlit.app/). "
-                        "BED files can be used with the [UCSC Genome Browser](https://genome.ucsc.edu/cgi-bin/hgCustom).")
 
     elif target_selection == 'off-target':
         ENTRY_METHODS = dict(