Update src/visualization.py

src/visualization.py

@@ -1,15 +1,3 @@
-import streamlit as st
-import json
-import networkx as nx
-import pandas as pd
-from typing import Dict, List, Any, Optional, Set, Tuple
-import plotly.graph_objects as go
-import plotly.express as px
-import matplotlib.pyplot as plt
-import matplotlib.colors as mcolors
-from collections import defaultdict
-import math
-
 def display_reasoning_trace(query: str, retrieved_docs: List[Dict], answer: str, ontology_manager):
     """Display an enhanced trace of how ontological reasoning was used to answer the query."""
     st.subheader("🧠 Ontology-Enhanced Reasoning")
@@ -376,26 +364,24 @@ def explain_ontology_advantages(entity_mentions: List[Dict], relationship_mentio
     
     return advantages# src/visualization.py
 
+import streamlit as st
+import json
+import networkx as nx
+import pandas as pd
+from typing import Dict, List, Any, Optional, Set, Tuple
+import plotly.graph_objects as go
+import plotly.express as px
+import matplotlib.pyplot as plt
+import matplotlib.colors as mcolors
+from collections import defaultdict
+import math
+
 def render_html_in_streamlit(html_content: str):
-    """Display HTML content in Streamlit using an iframe."""
-    import base64
-    
-    # Encode the HTML content
-    encoded_html = base64.b64encode(html_content.encode()).decode()
-    
-    # Create an iframe with the data URL
-    iframe_html = f"""
-        <iframe 
-            srcdoc="{encoded_html}" 
-            width="100%" 
-            height="600px" 
-            frameborder="0" 
-            allowfullscreen>
-        </iframe>
-    """
+    """Display HTML content in Streamlit using components.html."""
+    import streamlit.components.v1 as components
     
-    # Display the iframe
-    st.markdown(iframe_html, unsafe_allow_html=True)
+    # Directly render the HTML using components.html
+    components.html(html_content, height=600, scrolling=True)
 
 
 def display_ontology_stats(ontology_manager):