Update src/streamlit_app.py

src/streamlit_app.py

@@ -554,12 +554,36 @@ if search_clicked or (query and len(query.strip()) > 0):
                 page_no = metadata.get("page_no", "N/A")
                 title = metadata.get("title")                
 
-                
-                st.markdown("#### [{i}] DOC : {doc_id}  |  Page: {page_no} | Title {title}".format(i=i, doc_id=doc_id, page_no=page_no, title=title))
+                # st.markdown("#### [{i}] DOC : {doc_id}  |  Page: {page_no} | Title {title}".format(i=i, doc_id=doc_id, page_no=page_no, title=title))
+                st.markdown(
+                    "#### [{i}] DOC : <span style='color:green;'>{doc_id}</span> | Page: {page_no} | Title: {title}".format(
+                        i=i, doc_id=doc_id, page_no=page_no, title=title
+                    ),
+                    unsafe_allow_html=True
+                )
+
 
                 html_content = markdown2.markdown(text_content)
-                st.markdown(f'<div style="background: #303336; padding: 1rem; border-radius: 8px; margin: 1rem 0; line-height: 1.6; color: white;">{html_content}</div>', unsafe_allow_html=True)
+                # st.markdown(f'<div style="background: #303336; padding: 1rem; border-radius: 8px; margin: 1rem 0; line-height: 1.6; color: white;">{html_content}</div>', unsafe_allow_html=True)
+                
+                # Split into lines for preview
+                lines = text_content.split("\n")
+                preview_lines = 5
+                preview_text = "\n".join(lines[:preview_lines])
+                html_preview = markdown.markdown(preview_text)
+                
+                # Show preview
+                st.markdown(
+                    f'<div style="background: #303336; padding: 1rem; border-radius: 8px; line-height: 1.6; color: white; max-height: 7em; overflow: hidden;">{html_preview}</div>',
+                    unsafe_allow_html=True
+                )
                 
+                # Expandable full content
+                with st.expander("🔽 Expand Full Document", expanded=False):
+                    st.markdown(
+                        f'<div style="background: #303336; padding: 1rem; border-radius: 8px; line-height: 1.6; color: white;">{html_content}</div>',
+                        unsafe_allow_html=True
+                    )
                 
                 # Expandable full metadata
                 doc_label = "Source" if st.session_state.chat_mode else "Result"