Update utils_layout.py

utils_layout.py

@@ -1,79 +1,147 @@
-# utils_layout.py
+# utils.py
+# -*- coding: utf-8 -*-
 import os
+from typing import Optional
+
 import streamlit as st
-from PIL import Image
+
+try:
+    from PIL import Image
+    _HAS_PIL = True
+except Exception:
+    _HAS_PIL = False
+
+
+def _debug(msg: str):
+    """Mostra diagnóstico leve quando LOGO_DEBUG=1 (Settings → Secrets)."""
+    if os.getenv("LOGO_DEBUG", "0") == "1":
+        st.caption(f"🧪 LOGO_DEBUG: {msg}")
 
 
-def _resolve_logo_path():
+def _resolve_logo_path() -> Optional[str]:
     """
     Resolve automaticamente o caminho da logo:
-    1. Variável de ambiente LOGO_PATH (opcional, via Secrets)
-    2. logo.png na raiz
-    3. assets/logo.png
-    4. images/logo.png
-    5. static/logo.png
 
-    Retorna o caminho absoluto se existir, caso contrário None.
+    1) LOGO_PATH (Secrets/ENV) — aceita relativo ou absoluto
+    2) logo.(png|svg|jpg|jpeg) na raiz
+    3) assets/logo.(png|svg|jpg|jpeg)
+    4) images/logo.(png|svg|jpg|jpeg)
+    5) static/logo.(png|svg|jpg|jpeg)
+
+    Retorna caminho **absoluto** se existir, senão None.
     """
+    base = os.path.abspath(os.getcwd())
     candidates = []
 
-    # 1) Caminho definido via Secrets → LOGO_PATH
+    # 1) via Secrets/ENV
     env_path = os.getenv("LOGO_PATH")
     if env_path:
         candidates.append(env_path)
 
-    # 2) Candidatos fixos
+    # 2–5) Padrões comuns (PNG → SVG → JPG/JPEG)
     candidates += [
         "logo.png",
         os.path.join("assets", "logo.png"),
         os.path.join("images", "logo.png"),
         os.path.join("static", "logo.png"),
+
+        "logo.svg",
+        os.path.join("assets", "logo.svg"),
+        os.path.join("images", "logo.svg"),
+        os.path.join("static", "logo.svg"),
+
+        "logo.jpg", "logo.jpeg",
+        os.path.join("assets", "logo.jpg"),
+        os.path.join("assets", "logo.jpeg"),
+        os.path.join("images", "logo.jpg"),
+        os.path.join("images", "logo.jpeg"),
+        os.path.join("static", "logo.jpg"),
+        os.path.join("static", "logo.jpeg"),
     ]
 
-    base = os.path.abspath(os.getcwd())
+    _debug(f"base cwd: {base}")
+    _debug(f"candidatos: {candidates}")
 
     for p in candidates:
+        if not p:
+            continue
         full = p if os.path.isabs(p) else os.path.join(base, p)
         if os.path.exists(full):
+            _debug(f"encontrado: {full}")
             return full
 
+    _debug("nenhum candidato encontrado")
     return None
 
 
-def exibir_logo(top=True, sidebar=True):
+def _render_image_any(path: str, target: str = "top", width: Optional[int] = None):
+    """
+    Renderiza imagem:
+    - PNG/JPG/JPEG → PIL (se disponível) ou bytes
+    - SVG          → bytes (Pillow não lê SVG)
+    """
+    try:
+        _, ext = os.path.splitext(path.lower())
+
+        if ext == ".svg":
+            with open(path, "rb") as f:
+                data = f.read()
+            if target == "sidebar":
+                st.sidebar.image(data)
+            else:
+                st.image(data, width=width)
+            return
+
+        if _HAS_PIL:
+            img = Image.open(path)
+            if target == "sidebar":
+                st.sidebar.image(img, use_container_width=True)
+            else:
+                st.image(img, use_container_width=(width is None), width=width)
+        else:
+            with open(path, "rb") as f:
+                data = f.read()
+            if target == "sidebar":
+                st.sidebar.image(data, use_container_width=True)
+            else:
+                st.image(data, use_container_width=(width is None), width=width)
+
+    except Exception as e:
+        # Fallback amigável (não quebra a página)
+        if target == "sidebar":
+            st.sidebar.markdown(f"**IOI‑RUN** _(logo indisponível: {e})_")
+        else:
+            st.markdown(f"### IOI‑RUN _(logo indisponível: {e})_")
+
+
+def exibir_logo(top: bool = True, sidebar: bool = True, top_width: Optional[int] = None):
     """
-    Exibe a logo na parte superior (top) e/ou na sidebar.
-    Se a logo não existir, mostra texto fallback limpo sem quebrar a execução.
+    Exibe a logo no topo e/ou na sidebar.
+    - top_width: largura opcional (px) para a imagem do topo.
     """
     logo_path = _resolve_logo_path()
 
     if not logo_path:
-        # Fallback seguro
+        # Fallback textual — evita exceções se o arquivo não existir
         if top:
             st.markdown("### IOI‑RUN")
         if sidebar:
             st.sidebar.markdown("### IOI‑RUN")
         return
 
-    try:
-        logo = Image.open(logo_path)
-    except Exception as e:
-        if top:
-            st.markdown(f"### IOI‑RUN (logo indisponível: {e})")
-        if sidebar:
-            st.sidebar.markdown("### IOI‑RUN")
-        return
-
-    # -----------------------
-    # LOGO NO TOPO (CENTRADA)
-    # -----------------------
+    # TOPO (centralizado)
     if top:
-        col1, col2, col3 = st.columns([2, 3, 2])
-        with col2:
-            st.image(logo, use_column_width=True)
+        try:
+            col1, col2, col3 = st.columns([2, 3, 2])
+        except Exception:
+            col1 = col2 = col3 = None
+
+        if col2:
+            with col2:
+                _render_image_any(logo_path, target="top", width=top_width)
+        else:
+            _render_image_any(logo_path, target="top", width=top_width)
 
-    # -----------------------
-    # LOGO NO SIDEBAR
-    # -----------------------
+    # SIDEBAR
     if sidebar:
-        st.sidebar.image(logo, use_column_width=True)
+        _render_image_any(logo_path, target="sidebar")