data_loader.py

from datasets import load_dataset
import pandas as pd
import streamlit as st

@st.cache_data
def load_data():
    """Загрузка и кэширование набора данных"""
    dataset = load_dataset("loim/ru_fantasy_characters")
    return pd.DataFrame(dataset["train"])

@st.cache_data
def get_unique_mood(_df):
    """Получение уникальных значений настроения"""
    return sorted(_df["mood"].unique().tolist())

@st.cache_data
def get_unique_tags(_df):
    """Получение уникальных тегов"""
    return sorted({tag for tags in _df["tags"] for tag in tags.split()})