Update cas12.py (#7)

- Update cas12.py (16050d9dffc8a618e1b1e11c4f1a7ca3be0c7cb0)


Co-authored-by: Qingyang Liu <LfOreVEr@users.noreply.huggingface.co>

cas12.py

@@ -136,14 +136,19 @@ def format_prediction_output(targets, model_path):
 def process_gene(gene_symbol, model_path):
     transcripts = fetch_ensembl_transcripts(gene_symbol)
     results = []
+    all_exons = []
+    all_gene_sequences = []
+    
     if transcripts:
         for i in range(len(transcripts)):
             Exons = transcripts[i]['Exon']
+            all_exons.append(Exons)
             transcript_id = transcripts[i]['id']
             for j in range(len(Exons)):
                 exon_id = Exons[j]['id']
                 gene_sequence = fetch_ensembl_sequence(exon_id)
                 if gene_sequence:
+                    all_gene_sequences.append(gene_sequence)
                     start = Exons[j]['start']
                     end = Exons[j]['end']
                     strand = Exons[j]['strand']
@@ -159,7 +164,7 @@ def process_gene(gene_symbol, model_path):
     else:
         print("Failed to retrieve transcripts.")
 
-    return results, gene_sequence, Exons
+    return results, all_gene_sequences, all_exons
 
 
 # def create_genbank_features(formatted_data):