Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,54 +1,65 @@
 import streamlit as st
-from rag import rag_chain, retriever  # ← добавьте экспорт retriever из rag.py
+from rag import rag_chain, retriever
 
-st.set_page_config(page_title="Поиск фильмов", layout="centered")
+# === Настройка страницы ===
+st.set_page_config(
+    page_title="Поиск фильмов",
+    page_icon="🎬",
+    layout="centered"
+)
+
+# === Заголовок и подзаголовок ===
 st.title("🎬 RAG-поиск по фильмам")
 st.caption("Спросите что-нибудь о фильмах — я найду похожие и расскажу!")
 
+# === Инициализация истории сообщений ===
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Отображение истории
+# === Отображение истории чата ===
 for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
-    if "posters" in msg:
-        cols = st.columns(min(5, len(msg["posters"])))
-        for i, url in enumerate(msg["posters"]):
-            if url:
+    with st.chat_message(msg["role"]):
+        st.write(msg["content"])
+        if "posters" in msg and msg["posters"]:
+            posters = [url.strip() for url in msg["posters"] if url.strip().startswith("http")]
+            cols = st.columns(min(5, len(posters)))
+            for i, url in enumerate(posters):
                 cols[i % len(cols)].image(url, width=130)
 
-# Ввод пользователя
-if prompt := st.chat_input("Например: 'Порекомендуй фильм про космос'"):
+# === Обработка нового запроса ===
+if prompt := st.chat_input("Например: «Порекомендуй фильм про космос»"):
+    # Добавляем сообщение пользователя
     st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
+    with st.chat_message("user"):
+        st.write(prompt)
 
-    # Получаем документы ДО генерации ответа (чтобы показать те же фильмы, что использовал LLM)
-    docs = retriever.invoke(prompt)
+    # Получаем релевантные документы
+    with st.spinner("🔍 Ищу подходящие фильмы..."):
+        docs = retriever.invoke(prompt)
 
-    # Извлекаем poster_url из метаданных
+    # Извлекаем постеры из метаданных
     posters = [
-        doc.metadata.get("poster_url", "")
+        doc.metadata.get("poster_url", "").strip()
         for doc in docs
-        if doc.metadata.get("poster_url")
+        if doc.metadata.get("poster_url", "").strip().startswith("http")
     ]
 
-    # Генерация ответа
-    with st.spinner("Ищу фильмы..."):
+    # Генерируем ответ через RAG
+    with st.spinner("🧠 Формулирую ответ..."):
         response = rag_chain.invoke(prompt)
 
-    # Сохраняем в историю с постерами
+    # Сохраняем ответ ассистента с постерами
     st.session_state.messages.append({
         "role": "assistant",
         "content": response,
         "posters": posters
     })
 
-    # Вывод ответа и постеров
-    st.chat_message("assistant").write(response)
-    if posters:
-        st.write("##### 🎞️ Найденные фильмы:")
-        cols = st.columns(min(5, len(posters)))
-        for i, url in enumerate(posters):
-            clean_url = url.strip()  # ← КЛЮЧЕВОЙ МОМЕНТ
-            if clean_url.startswith("http"):
-                cols[i % len(cols)].image(clean_url, width=130)
+    # Отображаем ответ
+    with st.chat_message("assistant"):
+        st.write(response)
+        if posters:
+            st.markdown("##### 🎞️ Найденные фильмы:")
+            cols = st.columns(min(5, len(posters)))
+            for i, url in enumerate(posters):
+                cols[i % len(cols)].image(url, width=130)