add tab

pages/recommendations.py

@@ -84,7 +84,7 @@ with col1:
     query = st.text_area('Введите запрос, чтобы получить рекомендации', f'{text}', height=95)
 with col2:
     number = st.number_input(
-        "Сколько книг найти?", value=10
+        "Сколько книг найти?", value=3
 )
     find_button = st.button('Найти', key='find_button', use_container_width=True)
 
@@ -99,25 +99,24 @@ if find_button and query:
                 with col2:
                     st.subheader(f"{row['title']}")
                     st.write(f"**Автор:** {row['author']}")
-                    st.write("---")
-                    st.write(row['annotation'])
-                    col1, col2 = st.columns(2)
-                    with col1:
+                    tab1, tab2 = st.tabs(['Аннотация', 'Краткое содержание'])
+                    with tab1:
+                        st.write(row['annotation'])
                         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(f"**Ссылка:** {row['page_url']}")
+                    with tab2:
+                        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("---")
                 st.write("---")