Update app.py

app.py

@@ -97,19 +97,24 @@ mlb      = joblib.load(download_file("data/mlb_597.pkl"))
 GO       = mlb.classes_
 
 # ———————————————————  UI  ——————————————————— #
-# --- aspecto geral da página
 st.set_page_config(page_title="Predição de Funções Moleculares de Proteínas",
                    page_icon="🧬", layout="centered")
 
-# CSS: fundo branco sólido + pequenos ajustes
+# CSS: fundo branco + logo + traço vertical entre colunas
 st.markdown(
     """
     <style>
-        body, .stApp {
-            background-color: #FFFFFF !important;
-        }
+        body, .stApp { background-color: #FFFFFF !important; }
         .block-container { padding-top: 1.5rem; }
         textarea { font-size: 0.9rem !important; }
+
+        /* traço vertical: primeiro column do par recebe border-right */
+        div[data-testid="column"]:first-child {
+            border-right: 1px solid #E0E0E0;
+            padding-right: 1.2rem;
+        }
+        /* empurra logo ligeiramente para baixo */
+        img.logo { margin-top: 0.5rem; }
     </style>
     """,
     unsafe_allow_html=True
@@ -118,7 +123,7 @@ st.markdown(
 # Logo (coloca logo.png na raiz do Space)
 LOGO_PATH = "logo.png"
 if os.path.exists(LOGO_PATH):
-    st.image(LOGO_PATH, width=180)
+    st.image(LOGO_PATH, width=180, output_format="auto", use_column_width=False, caption=None, clamp=False, channels="RGB", format="PNG", class_name="logo")
 
 st.title("Predição de Funções Moleculares de Proteínas (GO:MF)")
 
@@ -136,51 +141,53 @@ def parse_fasta_multiple(fasta_str):
         if not entry.strip():
             continue
         lines = entry.strip().splitlines()
-        if i > 0:          # bloco típico FASTA
+        if i > 0:
             header = lines[0].strip()
             seq = "".join(lines[1:]).replace(" ", "").upper()
-        else:              # sequência sem '>'
+        else:
             header = f"Seq_{i+1}"
             seq = "".join(lines).replace(" ", "").upper()
         if seq:
             parsed.append((header, seq))
     return parsed
 
-# ———————————————————  FUNÇÕES AUXILIARES DE LAYOUT  ——————————————————— #
+# ———————————————————  FUNÇÕES AUXILIARES  ——————————————————— #
 def go_link(go_id, name=""):
-    """Cria link para página do GO term (QuickGO)."""
+    """Markdown link para QuickGO."""
     url = f"https://www.ebi.ac.uk/QuickGO/term/{go_id}"
     label = f"{go_id} — {name}" if name else go_id
     return f"[{label}]({url})"
 
-def prot_link(header):
-    """Tenta gerar link para UniProt usando o primeiro token do header."""
-    pid = header.split()[0]
-    url = f"https://www.uniprot.org/uniprotkb/{pid}"
-    return f"[{header}]({url})"
+def uni_url(header):
+    """URL UniProt (primeiro token do header)."""
+    return f"https://www.uniprot.org/uniprotkb/{header.split()[0]}"
+
+# ———————————————————  MOSTRAR RESULTADOS  ——————————————————— #
+def mostrar(header, y_pred):
+    """Expander com botão UniProt + duas colunas de resultados."""
+    with st.expander(header, expanded=True):
+        # botão que abre UniProt noutra aba
+        st.link_button("🔗 Ver na UniProt", uni_url(header), type="primary")
+        st.markdown("---")
 
-# ———————————————————  FUNÇÃO PRINCIPAL DE RESULTADOS  ——————————————————— #
-def mostrar(tag, y_pred):
-    """Mostra resultados em duas colunas dentro de um expander."""
-    with st.expander(tag, expanded=True):
         col1, col2 = st.columns(2)
 
-        # ——— coluna 1 : termos acima do threshold
+        # --- coluna 1 : termos acima do threshold
         with col1:
             st.markdown(f"**GO terms com prob ≥ {THRESH}**")
             hits = mlb.inverse_transform((y_pred >= THRESH).astype(int))[0]
             if hits:
                 for go_id in hits:
                     name, defin = GO_INFO.get(go_id, ("— sem nome —", ""))
-                    defin = re.sub(r'^\\s*"?(.+?)"?\\s*(\\[[^\\]]*\\])?\\s*$', r'\\1',
+                    defin = re.sub(r'^\\s*\"?(.+?)\"?\\s*(\\[[^\\]]*\\])?\\s*$', r'\\1',
                                    defin or "")
-                    st.markdown(f"- {go_link(go_id, name)}  ")
+                    st.markdown(f"- {go_link(go_id, name)}")
                     if defin:
                         st.caption(defin)
             else:
                 st.code("— nenhum —")
 
-        # ——— coluna 2 : top-N mais prováveis
+        # --- coluna 2 : top-N mais prováveis
         with col2:
             st.markdown(f"**Top {TOP_N} GO terms mais prováveis**")
             for rank, idx in enumerate(np.argsort(-y_pred[0])[:TOP_N], start=1):
@@ -197,23 +204,22 @@ if predict_clicked:
 
     for header, seq in parsed_seqs:
         with st.spinner(f"A processar {header}… (pode demorar alguns minutos)"):
-            # ————————————  EMBEDDINGS  ———————————— #
+            # ---------- EMBEDDINGS ----------
             emb_pb  = embed_seq(FINETUNED_PB,  seq, CHUNK_PB)
             emb_bfd = embed_seq(FINETUNED_BFD, seq, CHUNK_PB)
             emb_esm = embed_seq(BASE_ESM,       seq, CHUNK_ESM)
 
-            # ————————————  PREDIÇÕES MLPs  ———————————— #
+            # ---------- PREDIÇÕES ----------
             y_pb  = mlp_pb.predict(emb_pb)
             y_bfd = mlp_bfd.predict(emb_bfd)
             y_esm = mlp_esm.predict(emb_esm)[:, :597]  # alinhar nº de termos
 
-            # ————————————  STACKING  ———————————— #
             X     = np.concatenate([y_pb, y_bfd, y_esm], axis=1)
             y_ens = stacking.predict(X)
 
-        mostrar(prot_link(header), y_ens)
+        mostrar(header, y_ens)
 
-# ———————————————————  LISTA COMPLETA DE TERMOS SUPORTADOS  ——————————————————— #
+# ———————————————————  LISTA COMPLETA  ——————————————————— #
 with st.expander("Mostrar lista completa dos 597 GO terms possíveis", expanded=False):
     cols = st.columns(3)
     for i, go_id in enumerate(GO):