Update src/streamlit_app.py

src/streamlit_app.py

@@ -87,26 +87,25 @@ async def main():
 
     if user_input:
         st.session_state.messages.append({"role": "user", "content": user_input})
-    
+
         with st.chat_message("user"):
             st.markdown(user_input)
-    
-            with st.chat_message("assistant"):
 
+        with st.chat_message("assistant"):
             AVAILABLE_SOURCES = list(set(
                 chunk["file_path"] for chunk in FULL_CHUNKS_DICT.values() if "file_path" in chunk
             ))
-    
+
             explicit_source = None
             is_specific = await is_query_about_specific_file(user_input)
             if is_specific:
                 guessed_file = await guess_relevant_file(user_input, AVAILABLE_SOURCES)
                 if guessed_file and guessed_file != "None":
                     explicit_source = guessed_file
-    
+
             matched_sources = []
             answer = ""
-    
+
             if explicit_source:
                 filtered_chunk_ids = [
                     chunk_id for chunk_id, chunk in FULL_CHUNKS_DICT.items()
@@ -114,8 +113,9 @@ async def main():
                 ]
                 ans_param = QueryParam(mode="mix", top_k=1)
                 answer = await rag.aquery(user_input, param=ans_param)
-    
+
                 st.markdown(f"🔍 **Focused on:** **{os.path.basename(explicit_source)}**")
+
                 matched_sources = [
                     {
                         "timestamp": chunk.get("timestamp"),
@@ -124,17 +124,17 @@ async def main():
                     for chunk_id, chunk in FULL_CHUNKS_DICT.items()
                     if chunk_id in filtered_chunk_ids and "timestamp" in chunk
                 ]
-    
+
             else:
                 ctx_param = QueryParam(mode="mix", only_need_context=True, top_k=3)
                 context_chunks = await rag.aquery(f"{user_input}\n<!--ctx-->", param=ctx_param)
-    
+
                 ans_param = QueryParam(mode="mix", top_k=3)
                 answer = await rag.aquery(user_input, param=ans_param)
-    
+
                 dc_chunks = extract_dc_chunks(context_chunks)[:3]
                 matched_sources = find_matches(dc_chunks, FULL_CHUNKS_DICT)
-    
+
             # Показываем источники перед ответом
             if matched_sources:
                 sources_md = "#### 📚 Sources:\n" + "\n".join(
@@ -143,7 +143,7 @@ async def main():
                 )
                 st.markdown(sources_md)
                 st.session_state.messages.append({"role": "assistant", "content": sources_md})
-    
+
             # Показываем ответ
             st.markdown(answer)
             st.session_state.messages.append({"role": "assistant", "content": answer})