natasha

app.py

@@ -156,9 +156,10 @@ def rerank_and_boost(query, df, I, rerank_scores, k=10):
 
 
 # --- Local method
-stopwords = ["хочу", "посмотреть", "смотреть", "нашел", "фильм", "покажи"]
+# stopwords = ["хочу", "посмотреть", "смотреть", "нашел", "фильм", "покажи"]
+stopwords = []
 
-query = "смешная комедия про детектива с животными"
+query = "хочу посмотреть фильм про кибер пиратов"
 query_cleaned = "query: " + "".join([word for word in query if word not in stopwords])
 query_vector = encoder.encode(query_cleaned, convert_to_numpy=True)
 query_vector = query_vector / np.linalg.norm(query_vector)
@@ -167,6 +168,17 @@ query_vector = query_vector.astype("float32").reshape(1, -1)
 # --- Search FAISS
 D, I = index.search(query_vector, k=10)
 
+
+# ---
+# ---
+# ---
+# ---
+# ---
+
+
+st.title(query)
+st.divider()
+
 col1, col2, col3 = st.columns(3)
 
 with col1:
@@ -197,7 +209,7 @@ with col3:
         title = df.iloc[idx]["title"]
         desc = df.iloc[idx]["description"]
         url = df.iloc[idx]["url"]
-        print(f"{title} | {score} | {boost}")
-        print(f"Description: {desc}")
-        print(f"URL: {url}")
-        print("")
+        st.write(f"{title} | {score} | {boost}")
+        st.write(f"Description: {desc}")
+        st.write(f"URL: {url}")
+        st.write("")