Update src/visualization.py

src/visualization.py

@@ -1,3 +1,15 @@
+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")
@@ -364,18 +376,6 @@ 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 components.html."""
     import streamlit.components.v1 as components