fix genbank

app.py

@@ -96,50 +96,33 @@ if selected_model == 'Cas9':
     if target_selection == 'on-target':
         # Gene symbol entry
         gene_symbol = st.text_input('Enter a Gene Symbol:', key='gene_symbol')
-
         # Prediction button
         predict_button = st.button('Predict on-target')
-
         # Process predictions
         if predict_button and gene_symbol:
             predictions, gene_sequence = cas9on.process_gene(gene_symbol, cas9on_path)
             sorted_predictions = sorted(predictions, key=lambda x: x[-1], reverse=True)[:10]
             st.session_state['on_target_results'] = sorted_predictions
 
-        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", "gRNA", "Prediction"])
-
-            # Pass the gene_sequence to the function
-            genbank_file_path = f"{gene_symbol}_crispr_targets.gb"
-            cas9on.generate_genbank_file_from_df(df, gene_sequence, gene_symbol, genbank_file_path)
-            st.write('Top on-target predictions:')
-            st.dataframe(df)
-
-            # Initialize GenomeViz
-            gv = GenomeViz()
-            genome_size = max(
-                df["End Pos"])  # Assuming the max end position approximates the genome size for visualization purposes
-            track = gv.add_feature_track("CRISPR Targets", genome_size)
-
-            for _, row in df.iterrows():
-                start, end, strand = row["Start Pos"], row["End Pos"], row["Strand"]
-                label = row["gRNA"]
-                track.add_feature(start, end, strand, label=label)
-
-            # Save and display the visualization
-            fig = gv.plotfig()
-            st.pyplot(fig)
-
-            # After the GenomeViz plot, include the download button
-            with open(genbank_file_path, "rb") as file:
-                btn = st.download_button(
-                    label="Download GenBank file",
-                    data=file,
-                    file_name=genbank_file_path,
-                    mime="application/octet-stream"
-                )
+            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", "gRNA", "Prediction"])
 
+                # Make sure gene_sequence is defined before this point
+                if gene_sequence:  # Add a check to ensure gene_sequence is not empty
+                    genbank_file_path = f"{gene_symbol}_crispr_targets.gb"
+                    cas9on.generate_genbank_file_from_df(df, gene_sequence, gene_symbol, genbank_file_path)
+                    st.write('Top on-target predictions:')
+                    st.dataframe(df)
+
+                    # Add a download button for the GenBank file
+                    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"
+                        )
             os.remove(genbank_file_path)
 
     elif target_selection == 'off-target':