Update LLMInsights.py

LLMInsights.py

@@ -410,24 +410,23 @@ with tab1:
         st.markdown(f'<div style="max-height: 300px; overflow-y: auto; background-color:#f0f0f0; padding:10px; border-radius:10px;">{cleaned_context}</div>', unsafe_allow_html=True)
 
 
-        metadata_list = [doc.metadata for doc in response["source_documents"]]
+        # Augmented Knowledge Metadata Section
+        metadata_header = '<h2 style="color:#100170;">Augmented Knowledge Metadata</h2>'
+        metadata_container = '<div style="max-height: 300px; overflow-y: auto; background-color:#f0f0f0; padding:10px; border-radius:10px;">{}</div>'
+        
         formatted_metadata_list = []
         for i, metadata in enumerate(metadata_list, start=1):
-            formatted_metadata = f"<h2 style='color:#3366ff; font-size:16px;'>Metadata {i}:</h2><br>"
-            source = metadata.get('source', '').replace('\n', '')
-            title = metadata.get('title', '').replace('\n', '')
-            description = metadata.get('description', '').replace('\n', '')
-
-            formatted_metadata += f"<span style='color:#ff9900; font-weight:bold;'>Source:</span> <span style='color:#009900;'>{source}</span><br>"
-            formatted_metadata += f"<span style='color:#ff9900; font-weight:bold;'>Title:</span> <span style='color:#cc00cc;'>{title}</span><br>"
-            formatted_metadata += f"<span style='color:#ff9900; font-weight:bold;'>Description:</span> <span style='color:#ff0000;'>{description}</span><br><br>"
-            
-            formatted_metadata_list.append(formatted_metadata)
-
+            metadata_item = f"<h3 style='color:#3366ff;'>Metadata {i}:</h3>"
+            source = f"<span style='color:#ff9900; font-weight:bold;'>Source:</span> <span style='color:#009900;'>{metadata.get('source', '').replace('\n', '')}</span><br>"
+            title = f"<span style='color:#ff9900; font-weight:bold;'>Title:</span> <span style='color:#cc00cc;'>{metadata.get('title', '').replace('\n', '')}</span><br>"
+            description = f"<span style='color:#ff9900; font-weight:bold;'>Description:</span> <span style='color:#ff0000;'>{metadata.get('description', '').replace('\n', '')}</span><br><br>"
+            formatted_metadata_item = f"{metadata_item}{source}{title}{description}"
+            formatted_metadata_list.append(formatted_metadata_item)
+        
         metadata_text = '\n'.join(formatted_metadata_list)
-
-        st.markdown('<h1 style="color:#100170;font-size:24px;">Augmented Knowledge Metadata</h1>', unsafe_allow_html=True)
-        st.markdown(metadata_text, unsafe_allow_html=True)
+        formatted_metadata_section = metadata_header + metadata_container.format(metadata_text)
+        
+        st.markdown(formatted_metadata_section, unsafe_allow_html=True)