Merge pull request #105 from Sairam90/main

blackary · Aug 8, 2024 · 2c1d6e4 · 2c1d6e4
2 parents 18492be + c4c34e5
commit 2c1d6e4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/st_pages/__init__.py b/src/st_pages/__init__.py
@@ -3,6 +3,7 @@
 import json
 from dataclasses import dataclass
 from pathlib import Path
+from typing import cast
 
 import streamlit as st
 import toml
@@ -75,7 +76,10 @@ def _get_nav_from_toml(
 
         for page in pages:
             if page.is_section:
-                current_section = f"{translate_icon(page.icon)} {page.name}"
+                if page.icon is not None:
+                    current_section = f"{translate_icon(page.icon)} {page.name}"
+                else:
+                    current_section = cast(str, page.name)
                 if page.name in st.session_state[HIDE_PAGES_KEY]:
                     sections_to_drop.append(current_section)
                 continue