fix button

app.py

@@ -495,31 +495,36 @@ elif selected_model == 'Cas12':
         ('regular', 'mutation'),
         key='cas12target_selection'
     )
+    if 'current_gene_symbol' not in st.session_state:
+        st.session_state['current_gene_symbol'] = ""
+
+        # Define a function to clean up old files
+
+    def clean_up_old_files(gene_symbol):
+        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"
+        for path in [genbank_file_path, bed_file_path, csv_file_path]:
+            if os.path.exists(path):
+                os.remove(path)
+
     # Gene symbol entry with autocomplete-like feature
     gene_symbol = st.selectbox('Enter a Gene Symbol:', [''] + gene_symbol_list, key='gene_symbol',
                                format_func=lambda x: x if x else "")
 
-    # Initialize the current_gene_symbol in the session state if it doesn't exist
-    if 'current_gene_symbol' not in st.session_state:
-        st.session_state['current_gene_symbol'] = ""
+    # Handle gene symbol change and file cleanup
+    if gene_symbol != st.session_state['current_gene_symbol'] and gene_symbol:
+        if st.session_state['current_gene_symbol']:
+            # Clean up files only if a different gene symbol is entered and a previous symbol exists
+            clean_up_old_files(st.session_state['current_gene_symbol'])
+        # Update the session state with the new gene symbol
+        st.session_state['current_gene_symbol'] = gene_symbol
 
         if cas12target_selection == 'regular':
-            # Prediction button
-            predict_button = st.button('Predict')
-
-            # Function to clean up old files
-            def clean_up_old_files(gene_symbol):
-                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"
-                for path in [genbank_file_path, bed_file_path, csv_file_path]:
-                    if os.path.exists(path):
-                        os.remove(path)
-
+            predict_button = st.button('Predict cas12')
 
-            # Clean up files if a new gene symbol is entered
-            if st.session_state['current_gene_symbol'] and gene_symbol != st.session_state['current_gene_symbol']:
-                clean_up_old_files(st.session_state['current_gene_symbol'])
+            if 'exons' not in st.session_state:
+                st.session_state['exons'] = []
 
             # Process predictions
             if predict_button and gene_symbol:
@@ -546,12 +551,12 @@ elif selected_model == 'Cas12':
                     with col3:
                         st.markdown("**Nuclease**")
                         st.markdown("SpCas9")
-                    # Include "Target" in the DataFrame's columns
+                        # Include "Target" in the DataFrame's columns
                     try:
                         df = pd.DataFrame(st.session_state['on_target_results'],
-                                          columns=["Chr", "Start Pos", "End Pos", "Strand", "Transcript", "Exon",
-                                                   "Target",
-                                                   "gRNA", "Prediction"])
+                                            columns=["Chr", "Start Pos", "End Pos", "Strand", "Transcript", "Exon",
+                                                       "Target",
+                                                       "gRNA", "Prediction"])
                         st.dataframe(df)
                     except ValueError as e:
                         st.error(f"DataFrame creation error: {e}")
@@ -564,7 +569,7 @@ elif selected_model == 'Cas12':
                     EXON_BASE = 0  # Base position for exons and CDS on the Y axis
                     EXON_HEIGHT = 0.02  # How 'tall' the exon markers should appear
 
-                    # Plot Exons as small markers on the X-axis
+                        # Plot Exons as small markers on the X-axis
                     for exon in st.session_state['exons']:
                         exon_start, exon_end = exon['start'], exon['end']
                         fig.add_trace(go.Bar(
@@ -596,7 +601,7 @@ elif selected_model == 'Cas12':
                             y=[y_value],
                             mode='markers+text',
                             marker=dict(symbol='triangle-up' if strand == '1' or strand == '+' else 'triangle-down',
-                                        size=12),
+                                            size=12),
                             text=f"Rank: {i}",  # Text label
                             hoverinfo='text',
                             hovertext=f"Rank: {i}<br>Chromosome: {chrom}<br>Target Sequence: {target}<br>gRNA: {gRNA}<br>Start: {start}<br>End: {end}<br>Strand: {'+' if strand == '1' or strand == '+' else '-'}<br>Transcript: {transcript}<br>Prediction: {prediction_score:.4f}",
@@ -661,7 +666,7 @@ elif selected_model == 'Cas12':
                         )
         elif cas12target_selection == 'mutation':
             # Prediction button
-            predict_button = st.button('Predict')
+            predict_button = st.button('Predict cas12')
             vcf_reader = cyvcf2.VCF('SRR25934512.filter.snps.indels.vcf.gz')
 
             if 'exons' not in st.session_state: