urfu-pe / run.py
Alex
changed readme.md and run.py for hf
37b635d
import streamlit as st
from pathlib import Path
def st_page_rename(pages_name: dict[str, str]) -> None:
"""
Переименование страниц в сайдбаре
Временный хак, так как другие способы не сработали
:param pages_name: Список py файлов и имен
"""
from streamlit.source_util import get_pages as st_get_pages
from streamlit.source_util import _on_pages_changed as st_on_pages_changed
pages = st_get_pages("")
for page_k, page_v in pages.items():
script_path = Path(page_v["script_path"])
for page_name_k, page_name_v in pages_name.items():
name_path = Path(page_name_k)
if Path.samefile(script_path, name_path):
page_v["page_name"] = page_name_v
st_on_pages_changed.send()
def read_readme() -> str:
"""
Чтение файла README.md
:return: Текст
"""
text = Path("README.md").read_text(encoding='utf-8')
return text[text.find('#'):]
def main_app() -> None:
"""
Запуск основного приложения
"""
st_page_rename({"run.py": "Главная страница",
"pages/page_one.py": "Генератор аудио",
"pages/page_two.py": "Описание изображения"})
st.markdown(read_readme(), unsafe_allow_html=True)
# Запуск через streamlit
main_app()