final commit

pages/recommendations.py

@@ -6,6 +6,16 @@ from sentence_transformers import util
 from sentence_transformers import SentenceTransformer, util
 st.set_page_config(page_title="Custom Button Example", layout="wide")
 
+from langchain.chat_models.gigachat import GigaChat
+from langchain.prompts.chat import (
+    ChatPromptTemplate,
+    HumanMessagePromptTemplate,
+    SystemMessagePromptTemplate,
+)
+credentials = 'NDU5OTA0YzEtMWJkZi00OWE0LWEzZmEtNTZiNTg2MWI1ZjI3OjY5MGQ0NzRhLTExY2QtNDAzZi04ZGMwLWU5NDU1ZGQwMDY3Ng=='
+chat = GigaChat(model='GigaChat', credentials=credentials, verify_ssl_certs=False)
+
+
 @st.cache_resource
 def load_model_all_mpnet():
     return SentenceTransformer('all-mpnet-base-v2')
@@ -52,6 +62,7 @@ search = st.radio(
         "Используем 'all-mpnet-base-v2'",
         "Используем 'msmarco-roberta-base-v3'",
     ],
+    horizontal=True,
 )
 
 def params(search):
@@ -90,7 +101,23 @@ if find_button and query:
                     st.write(f"**Автор:** {row['author']}")
                     st.write("---")
                     st.write(row['annotation'])
-                    st.metric(label="Схожесть", value=f"{row['similarity_score']:.3f}")
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        st.metric(label="Схожесть", value=f"{row['similarity_score']:.3f}")
+                    with col2:
+                        with st.expander('Показать краткое содержание книги, используя GigaChat'):
+                            template = "ты умеешь кратко в несколько предложений описывать содержание книги по ее названию"
+                            system_message_prompt = SystemMessagePromptTemplate.from_template(template)
+                            human_template = "Кратко опиши содержание книги под названием: {book_title}"
+                            human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
+                            chat_prompt = ChatPromptTemplate.from_messages(
+                                [system_message_prompt, human_message_prompt]
+                            )
+                            formatted_prompt = chat_prompt.format_prompt(
+                                book_title=row['title']
+                            )
+                            response = chat(formatted_prompt.to_messages())
+                            st.write(response.content)
                     st.write(f"**Ссылка:** {row['page_url']}")
                 st.write("---")
     

requirements.txt

@@ -2,4 +2,5 @@ beautifulsoup4==4.12.3
 pandas==2.2.2
 Requests==2.32.3
 streamlit==1.37.0
-sentence-transformers==3.0.1
+sentence-transformers==3.0.1
+gigachain-community==0.2.11