Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,6 +8,8 @@ from pinecone_text.sparse import BM25Encoder
 import os
 import pickle
 import nltk
+import markdown2
+
 
 nltk.download("punkt", quiet=True)
 nltk.download("punkt_tab", quiet=True)
@@ -526,7 +528,7 @@ if search_clicked or (query and len(query.strip()) > 0):
                 
                 # Display AI response
                 # st.markdown(ai_response,unsafe_allow_html=True)
-                st.markdown(f'<div style="background: #303336; padding: 1rem; border-radius: 8px; margin: 1rem 0; line-height: 1.6;">{ai_response}</div>', unsafe_allow_html=True)
+                st.markdown(ai_response, unsafe_allow_html=True)
 
                 
                 
@@ -554,7 +556,8 @@ if search_clicked or (query and len(query.strip()) > 0):
                 
                 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(f'<div style="background: #303336; padding: 1rem; border-radius: 8px; margin: 1rem 0; line-height: 1.6;">{text_content}</div>', 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)
                 
                 
                 # Expandable full metadata