Update app.py

app.py

@@ -5,14 +5,15 @@ import shutil
 import os
 
 def fold(id, sequence):
-    sequence = sequence.upper().strip()
+    sequence = sequence.upper().replace(" ", "").replace("\n", "").replace("T", "U")
     if len(sequence) > 512:
-        return "", "", "", "", "Max sequence length is 512"
+        return "", "Max sequence length is 512", "", ""
+    if len(sequence) < 3:
+        return "", "Min sequence length is 3", "", ""
     shutil.rmtree("output", ignore_errors=True)
     ct_file = f"output/{id}.ct"
     structure_draw = f"output/{id}.png"
     fasta_file = f"output/{id}.fasta"
-    msg = "Completed."
     
     resolution = 1
     if len(sequence)>100:
@@ -26,12 +27,12 @@ def fold(id, sequence):
     if not dotbracket:
         dotbracket = None
         structure_draw = None
-        msg += " CT to dotbracket conversion failed (check .ct file)"
+        dotbracket = "CT to dotbracket conversion failed (.ct file should be ok)"
 
     with open(fasta_file, "w") as f:
         f.write(f">{id}\n{sequence}\n{dotbracket}")
 
-    return structure_draw, dotbracket, fasta_file, ct_file, msg 
+    return ct_file, dotbracket, structure_draw, fasta_file  
 
 
 def run():
@@ -45,11 +46,10 @@ def run():
             gr.Textbox(label="RNA Sequence",
                         value="GAUAAACCUUUAGCAAUAAACGAAAGUUUAACUAAGCCAUACUAACCCCAGGGUUGGUCAAUUUCGUGCCAGCCACCGCGGUCACACGAUUAACCCAAGCCAAUAGAAAUCGGCGUAAAGAGUGUUUUAGAUCAAUCCCCCAAUAAAGCUAAAAUUCACCUG")], 
         outputs=[
+            gr.File(label="Model output"),
+            gr.Textbox(label="Dot-bracket"),
             gr.Image(label="Structure"),
-            gr.Textbox(label="Dotbracket structure"),
-            gr.File(label="Fasta file"), 
-            gr.File(label="CT file"),
-            gr.Textbox(label="Output"),
+            gr.File(label="Fasta format"), 
             ], 
         examples=[
             ["srp_Bruc.abor._AE017223",	"AACCGGGUCAGGUCCGGAAGGAAGCAGCCCUAA"],