Update app.py

app.py

@@ -59,7 +59,7 @@ mlb      = joblib.load(download_file("data/mlb_597.pkl"))
 GO       = mlb.classes_
 
 # ———————————————————  UI  ——————————————————— #
-st.title("🔬 Predição de Funções Moleculares de Proteínas")
+st.title("Predição de Funções Moleculares de Proteínas")
 
 st.markdown(
     """
@@ -68,59 +68,64 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-fasta_input = st.text_area("Insere a sequência FASTA:", height=200)
+fasta_input = st.text_area("Insere uma ou mais sequências FASTA:", height=300)
 predict_clicked = st.button("Prever GO terms")
 
-if predict_clicked:
+# ———————————————————  PARSE DE MÚLTIPLAS SEQUÊNCIAS  ——————————————————— #
+def parse_fasta_multiple(fasta_str):
+    entries = fasta_str.strip().split(">")
+    parsed = []
+    for entry in entries:
+        if not entry.strip():
+            continue
+        lines = entry.strip().splitlines()
+        header = lines[0].strip()
+        seq = "".join(l.strip() for l in lines[1:]).replace(" ", "").upper()
+        if seq:
+            parsed.append((header, seq))
+    return parsed
 
-    # ——— 1) PRÉ-PROCESSAMENTO FASTA ———
-    lines = fasta_input.splitlines()
-    header = next((l for l in lines if l.startswith(">")), None)
-    seq = "".join(l.strip() for l in lines if not l.startswith(">")).replace(" ", "").upper()
+if predict_clicked:
+    parsed_seqs = parse_fasta_multiple(fasta_input)
 
-    if not seq:
-        st.warning("Por favor, insere primeiro uma sequência FASTA válida.")
+    if not parsed_seqs:
+        st.warning("Não foi possível encontrar nenhuma sequência válida.")
         st.stop()
 
-    if header:
-        st.markdown(f"**🧬 ID da proteína:** `{header[1:].strip()}`")
-
-    # ——— 2) EMBEDDINGS ———
-    with st.spinner("⏳ A gerar embeddings…"):
-        emb_pb  = embed_seq("Rostlab/prot_bert",            seq, CHUNK_PB)
-        emb_bfd = embed_seq("Rostlab/prot_bert_bfd",        seq, CHUNK_PB)
-        emb_esm = embed_seq("facebook/esm2_t33_650M_UR50D", seq, CHUNK_ESM)
-
-    # ——— 3) PREDIÇÕES ———
-    with st.spinner("🧠 A fazer predições…"):
-        y_pb  = mlp_pb.predict(emb_pb)
-        y_bfd = mlp_bfd.predict(emb_bfd)
-        y_esm = mlp_esm.predict(emb_esm)[:, :597]
-        X     = np.concatenate([y_pb, y_bfd, y_esm], axis=1)
-        y_ens = stacking.predict(X)
-
-    # ——— 4) RESULTADOS ———
-    def mostrar(tag, y_pred):
-        with st.expander(tag, expanded=(tag == "Ensemble (Stacking)")):
-            hits = mlb.inverse_transform((y_pred >= THRESH).astype(int))[0]
-            st.markdown(f"**GO terms com prob ≥ {THRESH}**")
-            if hits:
-                for go_id in hits:
-                    name, defin = GO_INFO.get(go_id, ("— sem nome —", "— sem definição —"))
-                    st.write(f"**{go_id} — {name}**")
-                    st.caption(defin)
-            else:
-                st.code("— nenhum —")
-
-            st.markdown(f"**Top {TOP_N} GO terms mais prováveis**")
-            for idx in np.argsort(-y_pred[0])[:TOP_N]:
-                go_id = GO[idx]
-                name, _ = GO_INFO.get(go_id, ("", ""))
-                st.write(f"{go_id} — {name} : {y_pred[0][idx]:.4f}")
-
-    # ——— MOSTRAR APENAS O ENSEMBLE POR DEFEITO ———
-    
-    # mostrar("ProtBERT (MLP)",      y_pb)
-    # mostrar("ProtBERT-BFD (MLP)",  y_bfd)
-    # mostrar("ESM-2 (MLP)",         y_esm)
-    mostrar("Ensemble (Stacking)", y_ens)
+    for header, seq in parsed_seqs:
+        with st.spinner(f"A processar {header}…"):
+            emb_pb  = embed_seq("Rostlab/prot_bert",            seq, CHUNK_PB)
+            emb_bfd = embed_seq("Rostlab/prot_bert_bfd",        seq, CHUNK_PB)
+            emb_esm = embed_seq("facebook/esm2_t33_650M_UR50D", seq, CHUNK_ESM)
+
+            y_pb  = mlp_pb.predict(emb_pb)
+            y_bfd = mlp_bfd.predict(emb_bfd)
+            y_esm = mlp_esm.predict(emb_esm)[:, :597]
+            X     = np.concatenate([y_pb, y_bfd, y_esm], axis=1)
+            y_ens = stacking.predict(X)
+
+        # ———————————————————  RESULTADOS ——————————————————— #
+        def mostrar(tag, y_pred):
+            with st.expander(tag, expanded=True):
+                hits = mlb.inverse_transform((y_pred >= THRESH).astype(int))[0]
+                st.markdown(f"**GO terms com prob ≥ {THRESH}**")
+                if hits:
+                    for go_id in hits:
+                        name, defin = GO_INFO.get(go_id, ("— sem nome —", "— sem definição —"))
+                        st.write(f"**{go_id} — {name}**")
+                        st.caption(defin)
+                else:
+                    st.code("— nenhum —")
+
+                st.markdown(f"**Top {TOP_N} GO terms mais prováveis**")
+                for idx in np.argsort(-y_pred[0])[:TOP_N]:
+                    go_id = GO[idx]
+                    name, _ = GO_INFO.get(go_id, ("", ""))
+                    st.write(f"{go_id} — {name} : {y_pred[0][idx]:.4f}")
+
+        # ———————————————————  ESCOLHE QUAIS MOSTRAR ——————————————————— #
+
+        # mostrar(f"{header} — ProtBERT (MLP)",      y_pb)
+        # mostrar(f"{header} — ProtBERT-BFD (MLP)",  y_bfd)
+        # mostrar(f"{header} — ESM-2 (MLP)",         y_esm)
+        mostrar(f"{header}", y_ens)