Update app.py

app.py

@@ -657,50 +657,50 @@ def main():
                     
     ############# Container for chat messages ##############
     with st.container():
-    # Display chat history
-    for i, (question, answer) in enumerate(st.session_state.chat_history):
-        answer_id = f"answer-{i}"
+        # Display chat history
+        for i, (question, answer) in enumerate(st.session_state.chat_history):
+            answer_id = f"answer-{i}"
     
-        # Custom HTML for the question with user avatar aligned to the right
-        st.markdown(f"""
-        <div style="display: flex; justify-content: flex-end; align-items: flex-start; margin-bottom: 20px;">
-            <span style="margin-right: 10px;">{question}</span>
-            <img src='data:image/png;base64,{st.session_state.user_selected_avatar}' style='width: 50px; height: 50px;'>
-        </div>
-        """, unsafe_allow_html=True)
-
-        # Display the response (only if it's not empty)
-        if response:
+            # Custom HTML for the question with user avatar aligned to the right
             st.markdown(f"""
-            <div id="{answer_id}" style="display: flex; align-items: flex-start; margin-bottom: 20px;">
-                <img src='data:image/png;base64,{ci_icon}' style='width: 50px; height: 50px; margin-right: 10px;'>
-                <span>{response}</span>
+            <div style="display: flex; justify-content: flex-end; align-items: flex-start; margin-bottom: 20px;">
+                <span style="margin-right: 10px;">{question}</span>
+                <img src='data:image/png;base64,{st.session_state.user_selected_avatar}' style='width: 50px; height: 50px;'>
             </div>
             """, unsafe_allow_html=True)
 
-        # Display metadata for unique source documents
-        if source_documents:
-            unique_sources = []
-            seen_urls = set()
-        
-            for doc in source_documents:
-                url = doc.metadata.get('url', 'N/A')
-                if url not in seen_urls:
-                    unique_sources.append(doc)
-                    seen_urls.add(url)
-        
-            if unique_sources:
+            # Display the response (only if it's not empty)
+            if response:
                 st.markdown(f"""
-                <div style="display: flex; align-items: flex-start; margin-bottom: 20px;">
+                <div id="{answer_id}" style="display: flex; align-items: flex-start; margin-bottom: 20px;">
                     <img src='data:image/png;base64,{ci_icon}' style='width: 50px; height: 50px; margin-right: 10px;'>
-                    <div>
-                    <b>Further Reading:</b><br>
+                    <span>{response}</span>
+                </div>
                 """, unsafe_allow_html=True)
 
-                for doc in unique_sources:
-                    st.markdown(f"- [{doc.metadata.get('title', 'No title available')}]({doc.metadata.get('url', '#')})")
-
-                st.markdown("</div></div>", unsafe_allow_html=True)
+            # Display metadata for unique source documents
+            if source_documents:
+                unique_sources = []
+                seen_urls = set()
+        
+                for doc in source_documents:
+                    url = doc.metadata.get('url', 'N/A')
+                    if url not in seen_urls:
+                        unique_sources.append(doc)
+                        seen_urls.add(url)
+        
+                if unique_sources:
+                    st.markdown(f"""
+                    <div style="display: flex; align-items: flex-start; margin-bottom: 20px;">
+                        <img src='data:image/png;base64,{ci_icon}' style='width: 50px; height: 50px; margin-right: 10px;'>
+                        <div>
+                        <b>Further Reading:</b><br>
+                    """, unsafe_allow_html=True)
+
+                    for doc in unique_sources:
+                        st.markdown(f"- [{doc.metadata.get('title', 'No title available')}]({doc.metadata.get('url', '#')})")
+
+                    st.markdown("</div></div>", unsafe_allow_html=True)
 
         # JavaScript to scroll to the latest answer
         if st.session_state.chat_history: