Upload app.py

app.py

@@ -37,9 +37,9 @@ def func(name):
 
 
 def func_mult(name):
-    sequence_list = name.split("\n")
-    sequence_list = [s.strip() for s in sequence_list]
-    sequence_list = [x for x in sequence_list if x]  # 列表推导式
+    sequence_list = process_fasta(name)
+    #sequence_list = [s.strip() for s in sequence_list]
+    #sequence_list = [x for x in sequence_list if x]  # 列表推导式
     output=[]
     for i in range(0, len(sequence_list), 1):
         output.append(func(sequence_list[i]))
@@ -47,6 +47,14 @@ def func_mult(name):
     return result
 
 
+def process_fasta(fasta_content):
+    sequences = []
+    fasta_file = io.StringIO(fasta_content)
+    for record in SeqIO.parse(fasta_file, "fasta"):
+        sequences.append(str(record.seq))
+    return sequences
+
+
 def read_fasta_file(file_path):
     sequences = []
     for seq_record in SeqIO.parse(file_path, "fasta"):
@@ -109,7 +117,7 @@ with gr.Blocks(css=css, title="GH29 Prediction", theme=gr.themes.Soft()) as demo
             upload_button = gr.UploadButton("Click to Upload a File", file_types=["fasta"], scale=2, size="sm", file_count="multiple")
             upload_button.upload(upload_file, upload_button, file_upload)
             file_button = gr.Button("Predict", scale=3, size="lg", elem_classes="button")
-            file_button_GenerateFile = gr.Button("Save to file", scale=2, size="sm")
+            file_button_GenerateFile = gr.Button("Save to File", scale=2, size="sm")
 
     single_button.click(func, inputs=single_input, outputs=single_output)
     multiple_button.click(func_mult, inputs=multiple_input, outputs=multiple_output)