исправление ошибки

app.py

@@ -28,7 +28,7 @@ locale = load_locale(st.session_state.lang)
 st.markdown(f"<style>{load_css()}</style>", unsafe_allow_html=True)
 
 # Получение уникальных значений для фильтров
-moods = [locale["filter_all"]] + get_unique_mood(df)
+moods = [locale['filter_all']] + get_unique_mood(df)
 all_tags = get_unique_tags(df)
 
 # Рендеринг сайдбара с фильтрами
@@ -38,11 +38,11 @@ selected_filters = render_sidebar(cookie, locale, all_langs, moods, all_tags)
 filtered_df = apply_filters(
     locale,
     df=df,
-    search_query=selected_filters["search_query"],
-    absurdity_range=selected_filters["absurdity_range"],
-    selected_mood=selected_filters["selected_mood"],
-    selected_tags=selected_filters["selected_tags"],
-    sort_option=selected_filters["sort_option"]
+    search_query=selected_filters['search_query'],
+    absurdity_range=selected_filters['absurdity_range'],
+    selected_mood=selected_filters['selected_mood'],
+    selected_tags=selected_filters['selected_tags'],
+    sort_option=selected_filters['sort_option']
 )
 
 # Основной контент

data_loader.py

@@ -6,14 +6,14 @@ import streamlit as st
 def load_data():
     """Загрузка и кэширование набора данных"""
     dataset = load_dataset("loim/ru_fantasy_characters")
-    return pd.DataFrame(dataset["train"])
+    return pd.DataFrame(dataset['train'])
 
 @st.cache_data
 def get_unique_mood(_df):
     """Получение уникальных значений настроения"""
-    return sorted(_df["mood"].unique().tolist())
+    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()})
+    return sorted({tag for tags in _df['tags'] for tag in tags.split()})

ui_components.py

@@ -21,7 +21,7 @@ def render_sidebar(cookie, locale, all_langs, moods, all_tags):
                 cookie.set("lang", st.session_state.lang_select)
 
         st.selectbox(
-            locale["filter_lang"],
+            locale['filter_lang'],
             all_langs,
             index=all_langs.index(st.session_state.lang),
             on_change=on_language_change,
@@ -30,15 +30,15 @@ def render_sidebar(cookie, locale, all_langs, moods, all_tags):
 
         st.markdown("---")
 
-        st.title(locale["filter_title"])
-        search_query = st.text_input(locale["filter_search"], "")
+        st.title(locale['filter_title'])
+        search_query = st.text_input(locale['filter_search'], "")
         sort_option = st.selectbox(
-            locale["filter_sort_title"],
-            locale["filter_sort"]
+            locale['filter_sort_title'],
+            locale['filter_sort']
         )
-        absurdity_range = st.slider(locale["filter_absurd"], 0, 10, (0, 10))
-        selected_mood = st.selectbox(locale["filter_mood"], moods)
-        selected_tags = st.multiselect(locale["filter_tags"], all_tags)
+        absurdity_range = st.slider(locale['filter_absurd'], 0, 10, (0, 10))
+        selected_mood = st.selectbox(locale['filter_mood'], moods)
+        selected_tags = st.multiselect(locale['filter_tags'], all_tags)
     
     return {
         "search_query": search_query,
@@ -52,14 +52,14 @@ def render_sidebar(cookie, locale, all_langs, moods, all_tags):
 def generate_rpg_card(locale, row):
     """Генерация RPG-карточки персонажа"""
     return f"""
-**{locale["rpt_card_world"]}** {row['world']}
+**{locale['rpt_card_world']}** {row['world']}
 
-**{locale["rpt_card_name"]}** {row['name']}
-**{locale["rpt_card_desc"]}** {row['description']}
-**{locale["rpt_card_story"]}** {row['short_story']}
-**{locale["rpt_card_style"]}** {row['style']}
+**{locale['rpt_card_name']}** {row['name']}
+**{locale['rpt_card_desc']}** {row['description']}
+**{locale['rpt_card_story']}** {row['short_story']}
+**{locale['rpt_card_style']}** {row['style']}
 
-**{locale["rpt_card_msg"]}** 
+**{locale['rpt_card_msg']}** 
 > {row['first_message']}
 """
 
@@ -72,16 +72,16 @@ def render_character_card(locale, row):
         
         col1, col2 = st.columns([1, 3])
         with col1:
-            st.markdown(f"**{locale["char_world"]}** {row['world']}")
-            st.markdown(f"**{locale["char_mood"]}** `{row['mood']}`")
-            st.markdown(f"**{locale["char_absurd"]}** `{row['absurdity']}/10`")
-            st.markdown(f"**{locale["char_tags"]}** `{' '.join(row['tags'].split())}`")
+            st.markdown(f"**{locale['char_world']}** {row['world']}")
+            st.markdown(f"**{locale['char_mood']}** `{row['mood']}`")
+            st.markdown(f"**{locale['char_absurd']}** `{row['absurdity']}/10`")
+            st.markdown(f"**{locale['char_tags']}** `{' '.join(row['tags'].split())}`")
         
         with col2:
-            st.markdown(f"**{locale["char_desc"]}** {row['description']}")
-            st.markdown(f"**{locale["char_style"]}** *{row['style']}*")
+            st.markdown(f"**{locale['char_desc']}** {row['description']}")
+            st.markdown(f"**{locale['char_style']}** *{row['style']}*")
         
-        with st.expander(locale["char_rpg_card"], expanded=False):
+        with st.expander(locale['char_rpg_card'], expanded=False):
             rpg_card = generate_rpg_card(locale, row)
             st.code(rpg_card, language="markdown")
         
@@ -92,7 +92,7 @@ def render_main_content(locale, filtered_df):
     st.title("🧙 Fantasy Characters Explorer")
     
     if len(filtered_df) == 0:
-        st.warning(locale["main_not_found"])
+        st.warning(locale['main_not_found'])
     else:
         # Инициализация состояния пагинации
         if 'page' not in st.session_state:
@@ -106,7 +106,7 @@ def render_main_content(locale, filtered_df):
         col1, col2, _ = st.columns([2, 3, 5])
         with col1:
             st.selectbox(
-                locale["main_counter"],
+                locale['main_counter'],
                 options=[5, 10, 20],
                 index=1,
                 key='per_page',
@@ -114,7 +114,7 @@ def render_main_content(locale, filtered_df):
             )
         with col2:
             st.number_input(
-                locale["main_page"],
+                locale['main_page'],
                 min_value=0,
                 max_value=total_pages-1,
                 value=st.session_state.page,
@@ -128,7 +128,7 @@ def render_main_content(locale, filtered_df):
         end_idx = min((st.session_state.page + 1) * st.session_state.per_page, len(filtered_df))
         
         # Отображение статистики
-        st.markdown(locale["main_found"].format(len(filtered_df), st.session_state.page, total_pages-1))
+        st.markdown(locale['main_found'].format(len(filtered_df), st.session_state.page, total_pages-1))
         
         # Рендеринг только видимых карточек
         for idx in range(start_idx, end_idx):
@@ -139,18 +139,18 @@ def render_main_content(locale, filtered_df):
         if total_pages > 1:
             cols = st.columns(4)
             with cols[0]:
-                if st.button(locale["pages_first"], disabled=(st.session_state.page == 0)):
+                if st.button(locale['pages_first'], disabled=(st.session_state.page == 0)):
                     st.session_state.page = 0
                     st.rerun()
             with cols[1]:
-                if st.button(locale["pages_before"], disabled=(st.session_state.page == 0)):
+                if st.button(locale['pages_before'], disabled=(st.session_state.page == 0)):
                     st.session_state.page -= 1
                     st.rerun()
             with cols[2]:
-                if st.button(locale["pages_next"], disabled=(st.session_state.page >= total_pages-1)):
+                if st.button(locale['pages_next'], disabled=(st.session_state.page >= total_pages-1)):
                     st.session_state.page += 1
                     st.rerun()
             with cols[3]:
-                if st.button(locale["pages_last"], disabled=(st.session_state.page == total_pages-1)):
+                if st.button(locale['pages_last'], disabled=(st.session_state.page == total_pages-1)):
                     st.session_state.page = total_pages-1
                     st.rerun()

utils.py

@@ -3,27 +3,27 @@ import streamlit as st
 @st.cache_data
 def get_search_mask(df, search_query):
     return (
-        df["world"].str.contains(search_query, case=False, na=False) |
-        df["name"].str.contains(search_query, case=False, na=False) |
-        df["short_story"].str.contains(search_query, case=False, na=False) |
-        df["description"].str.contains(search_query, case=False, na=False) |
-        df["style"].str.contains(search_query, case=False, na=False) |
-        df["first_message"].str.contains(search_query, case=False, na=False)
+        df['world'].str.contains(search_query, case=False, na=False) |
+        df['name'].str.contains(search_query, case=False, na=False) |
+        df['short_story'].str.contains(search_query, case=False, na=False) |
+        df['description'].str.contains(search_query, case=False, na=False) |
+        df['style'].str.contains(search_query, case=False, na=False) |
+        df['first_message'].str.contains(search_query, case=False, na=False)
     )
 
 @st.cache_data
 def apply_filters(locale, df, search_query, absurdity_range, selected_mood, selected_tags, sort_option):
     """Применение фильтров и сортировки"""
     filtered_df = df[
-        (df["absurdity"] >= absurdity_range[0]) & 
-        (df["absurdity"] <= absurdity_range[1])
+        (df['absurdity'] >= absurdity_range[0]) & 
+        (df['absurdity'] <= absurdity_range[1])
     ]
     
-    if selected_mood != locale["filter_all"]:
-        filtered_df = filtered_df[filtered_df["mood"] == selected_mood]
+    if selected_mood != locale['filter_all']:
+        filtered_df = filtered_df[filtered_df['mood'] == selected_mood]
     
     if selected_tags:
-        filtered_df = filtered_df[filtered_df["tags"].str.contains("|".join(selected_tags))]
+        filtered_df = filtered_df[filtered_df['tags'].str.contains("|".join(selected_tags))]
     
     if search_query:
         mask = get_search_mask(df, search_query)
@@ -31,10 +31,10 @@ def apply_filters(locale, df, search_query, absurdity_range, selected_mood, sele
     
     # Сортировка
     sort_columns = {
-        locale["filter_sort"][0]: ("name", True),
-        locale["filter_sort"][1]: ("name", False),
-        locale["filter_sort"][2]: ("absurdity", True),
-        locale["filter_sort"][3]: ("absurdity", False)
+        locale['filter_sort'][0]: ("name", True),
+        locale['filter_sort'][1]: ("name", False),
+        locale['filter_sort'][2]: ("absurdity", True),
+        locale['filter_sort'][3]: ("absurdity", False)
     }
     col, asc = sort_columns[sort_option]
     return filtered_df.sort_values(col, ascending=asc)