Cache parsing and folding

Mainly for folding, to be faster if you re-submit the same
sequence.

app.py

@@ -19,38 +19,48 @@ MGLTTSGGARGFCSLAVLQELVPRPELLFVIDRAFHSGKHAVDMQVVDQEGLGDGVATLLYAHQGLYTCLLQAEARLLGR
 MGLTTSGGARGFCSLAVLQELVPRPELLFVIDRAFHSGKHAVDMQVVDQEGLGDGVATLLYAHQGLYTCLLQAEARLLGREWAAVPALEPNF
 MGAAGYTGSLILAALKQNPDIAVYALNRNDEKLKDVCGQYSNLKGQVCDLSNESQVEALLSGPRKTVVNLVGPYSFYGSRVLNACIEANCHY
 """
-input_sequence = st.text_area("Sequences separated by a newline (max 400 resis each)", default_sequences)
-sequences = []
-
-# Parse and clean input sequences
-for seq in input_sequence.split("\n"):
-    seq = seq.strip()
-    if len(seq) > 400:
-        seq = seq[:400]
-    seq = re.sub("[^ACDEFGHIKLMNPQRSTVWY]", "", seq)
-    if len(seq) > 0:
-        sequences.append(seq)
+input_sequences = st.text_area("Sequences separated by a newline (max 400 resis each)", default_sequences)
+
+@st.cache_data
+def get_sequences(sequences_string):
+    sequences = []
+    # Parse and clean input sequences
+    for seq in sequences_string.split("\n"):
+        seq = seq.strip()
+        if len(seq) > 400:
+            seq = seq[:400]
+        seq = re.sub("[^ACDEFGHIKLMNPQRSTVWY]", "", seq)
+        if len(seq) > 0:
+            sequences.append(seq)
+    return sequences
+
+sequences = get_sequences(input_sequences)
 st.write(f"Found {len(sequences)} valid sequences")
 
 
 pdb_strings = []
 url = "https://api.esmatlas.com/foldSequence/v1/pdb/"
 
-# Fold sequences with ESMfold
-for seq in sequences:
+@st.cache_data
+def get_pdb(sequence):
     retries = 0
     pdb_str = None
     while retries < 3 and pdb_str is None:
-        response = requests.post(url, data=seq)
+        response = requests.post(url, data=sequence)
         pdb_str = response.text
         if pdb_str == "INTERNAL SERVER ERROR":
             retries += 1
             time.sleep(0.1)
             pdb_str = None
-    if pdb_str is not None:
-        pdb_strings.append(pdb_str)
+    return pdb_str
+
+
+# Fold sequences with ESMfold
+for seq in sequences:
+    if pdb := get_pdb(seq):
+        pdb_strings.append(pdb)
     else:
-        st.write("Failed to retrieve PDB structure after 3 retries")
+        st.write(f"Failed to retrieve PDB structure from ESMFold for {seq}")