Update app.py

app.py

@@ -127,13 +127,19 @@ def generate_caption(protein, prompt):
     x = prediction[0]
     x = [eval(i) for i in x.split('; ')]
     pred_terms = []
+    temp = []
     for i in x:
         txt = i[0]
         prob = i[1]
         sim_list = difflib.get_close_matches(txt.lower(), choices, n=1, cutoff=0.9)
         if len(sim_list) > 0:
-            pred_terms.append((sim_list[0], prob))
-    return str(pred_terms)
+            t_standard = sim_list[0]
+            if t_standard not in temp:
+                pred_terms.append(t_standard+f'(prob)')
+                temp.append(t_standard)
+
+    res_str = f"Based on the given amino acid sequence, the proteinappears to have a primary function of {', '.join(pred_terms)}"
+    return res_str
     # return "test"