edit app.py

app.py

@@ -182,8 +182,13 @@ if selected_model == 'Cas9':
 
         # Process predictions
         if predict_button and gene_symbol:
+            model_choice = st.radio("mutation or not:", ('normal', 'mutation'))
             with st.spinner('Predicting... Please wait'):
-                predictions, gene_sequence, exons = cas9att.process_gene(gene_symbol, cas9att_path)
+                if model_choice == 'cas9attvcf':
+                    predictions, gene_sequence, exons = cas9attvcf.process_gene(gene_symbol, cas9att_path)
+                else:
+                    predictions, gene_sequence, exons = cas9att.process_gene(gene_symbol, cas9att_path)
+
                 sorted_predictions = sorted(predictions)[:10]
                 st.session_state['on_target_results'] = sorted_predictions
                 st.session_state['gene_sequence'] = gene_sequence  # Save gene sequence in session state