app search speed 2.0

app.py

@@ -2,9 +2,6 @@ import pandas as pd
 import streamlit as st
 import random
 
-if 'selected_series' not in st.session_state:
-    st.session_state.selected_series = None
-
 if 'button_clicked' not in st.session_state:
     st.session_state.button_clicked = False
 
@@ -21,10 +18,9 @@ def show_serial(number, score=None):
             st.caption(data.iloc[number, 3])
             if score:
                 st.write(f'{score[1]} metric: {score[0]:.4f}')
-        with col3:
             st.markdown(f'[Ссылка]({data.iloc[number, 0]})')
-            with st.popover('Описание'):
-                st.write(data.iloc[number, 4])
+        with col3:
+            st.text_area("Описание", value=data.iloc[number, 4], height=300, disabled=True)
 
 
 st.title('Рекомендатор сериалов')
@@ -51,7 +47,6 @@ with cols[3]:
 
 if st.button('Дай 10 случайных сериалов', use_container_width=True):
     st.session_state.button_clicked = True
-    st.session_state.selected_series = None
 
 if st.session_state.button_clicked:
     for _ in range(10):

pages/search.py

@@ -11,6 +11,12 @@ def load_model():
     return SentenceTransformer("paraphrase-multilingual-mpnet-base-v2")
 
 
+@st.cache_resource
+def get_shows():
+    data = pd.read_csv('data/data.csv')
+    return data['description'].tolist()
+
+
 @st.cache_resource
 def compute_index(_model, sentences):
     embeddings = _model.encode(sentences).astype('float32')
@@ -28,14 +34,7 @@ def compute_index(_model, sentences):
 
 
 model = load_model()
-data = pd.read_csv('data/data.csv')
-sentences = data['description'].tolist()
-
-# indices = {
-#     'L2': faiss.read_index("models/index_l2.faiss"),
-#     'Dot': faiss.read_index("models/index_dot.faiss"),
-#     'Cos': faiss.read_index("models/index_cosine.faiss")
-# }
+sentences = get_shows()
 
 index_l2, index_dot, index_cosine = compute_index(model, sentences)
 indices = {
@@ -44,6 +43,12 @@ indices = {
     'Cos': index_cosine
 }
 
+# indices = {
+#     'L2': faiss.read_index("models/index_l2.faiss"),
+#     'Dot': faiss.read_index("models/index_dot.faiss"),
+#     'Cos': faiss.read_index("models/index_cosine.faiss")
+# }
+
 with st.form(key='submit'):
     cols = st.columns(2)
     with cols[0]: