рейтинговая система

app.py

@@ -1,29 +1,35 @@
 import streamlit as st
-from streamlit_cookies_controller import CookieController
+from streamlit_local_storage import LocalStorage
+from database import init_supabase
 from utils import apply_filters, apply_sort
 from locales import load_locale, lang_codes
 from data_loader import load_data, get_unique_mood, get_unique_tags
 from ui_components import render_sidebar, render_main_content, load_css
-import time
 
 DEFAULT_LANG = "en"
 
 
 # Настройки страницы
 st.set_page_config(layout="wide", page_icon="🧙", page_title="Fantasy Characters Explorer")
-cookie = CookieController()
+localS = LocalStorage()
 
 # Загрузка данных
+database = init_supabase()
 df = load_data()
 
-st.session_state.lang = cookie.get("lang")
+st.session_state.lang = localS.getItem("lang")
 all_langs = lang_codes()
 if st.session_state.lang not in all_langs:
     st.session_state.lang = (st.context.locale.split("-", 1)[0] if st.context.locale else DEFAULT_LANG)
-    cookie.set("lang", st.session_state.lang, secure=True, same_site=None)
+    localS.setItem("lang", st.session_state.lang)
 
 locale = load_locale(st.session_state.lang)
 
+if 'rated' not in st.session_state:
+    st.session_state.rated = localS.getItem("rated")
+    if st.session_state.rated is None:
+        st.session_state.rated = []
+
 # Загрузка стилей
 st.markdown(f"<style>{load_css()}</style>", unsafe_allow_html=True)
 
@@ -32,7 +38,7 @@ moods = [locale['filter_all']] + get_unique_mood(df)
 all_tags = get_unique_tags(df)
 
 # Рендеринг сайдбара с фильтрами
-selected_filters = render_sidebar(cookie, locale, all_langs, moods, all_tags)
+selected_filters = render_sidebar(localS, locale, all_langs, moods, all_tags)
 
 # Применение фильтров
 filtered_df = apply_filters(
@@ -48,4 +54,4 @@ sorted_df = apply_sort(locale, filtered_df, selected_filters['sort_option'])
 
 
 # Основной контент
-render_main_content(locale, sorted_df)
+render_main_content(locale, database, localS, sorted_df)

data_loader.py

@@ -1,12 +1,19 @@
-from datasets import load_dataset
+import os
+import hashlib
 import pandas as pd
 import streamlit as st
 
+def create_hash(row):
+    combined = str(row['name']) + str(row['short_story'])
+    return hashlib.sha256(combined.encode()).hexdigest()
+
 @st.cache_data
 def load_data():
     """Загрузка и кэширование набора данных"""
-    dataset = load_dataset("loim/ru_fantasy_characters")
-    return pd.DataFrame(dataset['train'])
+    df = pd.read_json(os.getenv("DATASET_PATH"))
+    df['hash'] = df.apply(create_hash, axis=1)
+
+    return df
 
 @st.cache_data
 def get_unique_mood(_df):

database.py

@@ -0,0 +1,15 @@
+import os
+import streamlit as st
+from supabase import create_client
+
+@st.cache_resource
+def init_supabase():
+    url = os.getenv("SUPABASE_URL")
+    key = os.getenv("SUPABASE_KEY")
+    return create_client(url, key)
+
+def get_hash(database, table, hash):
+    return database.table(table).select("*").eq("hash", hash).execute()
+
+def upsert_data(database, table, data):
+    return database.table(table).upsert(data).execute()

locales/en.json

@@ -28,5 +28,8 @@
 	"rpt_card_desc": "Appearance:",
 	"rpt_card_story": "Personality:",
 	"rpt_card_style": "Speech:",
-	"rpt_card_msg": "First message:"
+	"rpt_card_msg": "First message:",
+	"rating_text": "Rating: 👍 {} | 👎 {}",
+	"rating_like": "👍 Like",
+	"rating_dislike": "👎 Dislike"
 }

locales/ru.json

@@ -28,5 +28,8 @@
 	"rpt_card_desc": "Внешность:",
 	"rpt_card_story": "Характер:",
 	"rpt_card_style": "Речь:",
-	"rpt_card_msg": "Первая реплика:"
+	"rpt_card_msg": "Первая реплика:",
+	"rating_text": "Рейтинг: 👍 {} | 👎 {}",
+	"rating_like": "👍 Нравится",
+	"rating_dislike": "👎 Не нравится"
 }

requirements.txt

@@ -1,4 +1,5 @@
+pandas
+hashlib
+supabase
 streamlit
-streamlit_cookies_controller
-datasets
-pandas
+streamlit-local-storage

ui_components.py

@@ -1,4 +1,6 @@
 import streamlit as st
+from database import get_hash, upsert_data
+
 
 @st.cache_resource
 def load_css():
@@ -6,7 +8,7 @@ def load_css():
     with open("assets/styles.css") as f:
         return f.read()
 
-def render_sidebar(cookie, locale, all_langs, moods, all_tags):
+def render_sidebar(localS, locale, all_langs, moods, all_tags):
     """Рендеринг боковой панели с фильтрами"""
     with st.sidebar:
         st.markdown(
@@ -18,7 +20,7 @@ def render_sidebar(cookie, locale, all_langs, moods, all_tags):
 
         def on_language_change():
             if st.session_state.lang_select != st.session_state.lang:
-                cookie.set("lang", st.session_state.lang_select, secure=True, same_site=None)
+                localS.setItem("lang", st.session_state.lang_select)
 
         st.selectbox(
             locale['filter_lang'],
@@ -52,21 +54,21 @@ 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']}** 
-> {row['first_message']}
+{locale['rpt_card_msg']} {row['first_message']}
 """
 
 # Кэшированный рендеринг карточек
 @st.cache_data
 def render_character_card(locale, row):
     with st.container():
+        st.markdown("---")
         st.markdown(f"### {row['name']}")
         st.caption(f"*{row['short_story']}*")
         
@@ -84,10 +86,35 @@ def render_character_card(locale, row):
         with st.expander(locale['char_rpg_card'], expanded=False):
             rpg_card = generate_rpg_card(locale, row)
             st.code(rpg_card, language="markdown")
+
+def render_character_rating(locale, database, localS, row):
+    current_hash = row['hash']
+
+    rating_data, _ = get_hash(database, "rating", current_hash)
+    rating_data = rating_data[1][0] if rating_data[1] else {"hash": current_hash, "likes": 0, "dislikes": 0}
+
+    # Проверяем, оценивали ли уже этот хэш
+    if current_hash in st.session_state.rated and (rating_data['likes'] != 0 or rating_data['dislikes']):
+        # Показываем текущий рейтинг
+        st.write(locale["rating_text"].format(rating_data['likes'], rating_data['dislikes']))
+    else:
+        # Показываем кнопки для оценки
+        if st.button(locale["rating_like"], key=f"like_{current_hash}"):
+            rating_data["likes"] += 1
+            upsert_data(database, "rating", rating_data)
+            st.session_state.rated.append(current_hash)
+            localS.setItem("rated", st.session_state.rated)
+            st.rerun()
         
-        st.markdown("---")
+        if st.button(locale["rating_dislike"], key=f"dislike_{current_hash}"):
+            rating_data["dislikes"] += 1
+            upsert_data(database, "rating", rating_data)
+            st.session_state.rated.append(current_hash)
+            localS.setItem("rated", st.session_state.rated)
+            st.rerun()
+
 
-def render_main_content(locale, filtered_df):
+def render_main_content(locale, database, localS, filtered_df):
     """Рендеринг основного контента с пагинацией"""
     st.title("🧙 Fantasy Characters Explorer")
     
@@ -99,7 +126,7 @@ def render_main_content(locale, filtered_df):
             st.session_state.page = 0
         
         # Настройки пагинации
-        per_page = 10  # Персонажей на странице
+        per_page = 5  # Персонажей на странице
         total_pages = max(1, (len(filtered_df) // per_page) + (1 if len(filtered_df) % per_page else 0))
         if st.session_state.page >= total_pages:
             st.session_state.page = 0
@@ -110,7 +137,7 @@ def render_main_content(locale, filtered_df):
             st.selectbox(
                 locale['main_counter'],
                 options=[5, 10, 20],
-                index=1,
+                index=0,
                 key='per_page',
                 on_change=lambda: st.session_state.update(page=0)
             )
@@ -136,6 +163,9 @@ def render_main_content(locale, filtered_df):
         for idx in range(start_idx, end_idx):
             row = filtered_df.iloc[idx]
             render_character_card(locale, row)
+            render_character_rating(locale, database, localS, row)
+
+        st.markdown("---")
         
         # Кнопки навигации
         if total_pages > 1: