Update app.py

app.py

@@ -7,11 +7,13 @@ from fastapi.middleware.cors import CORSMiddleware
 from mcp.orchestrator import orchestrate_search, answer_ai_question
 from mcp.schemas import UnifiedSearchInput, UnifiedSearchResult
 from mcp.workspace import get_workspace, save_query
+from mcp.knowledge_graph import build_knowledge_graph
 from pathlib import Path
 import pandas as pd
 from fpdf import FPDF
 import asyncio
 import plotly.express as px
+import streamlit.components.v1 as components
 
 ROOT_DIR = Path(__file__).resolve().parent
 LOGO_PATH = ROOT_DIR / "assets" / "logo.png"
@@ -121,6 +123,13 @@ def render_ui():
             fig = px.histogram(pub_years, nbins=10, title="Publication Year Distribution")
             st.plotly_chart(fig)
 
+        # --- INTERACTIVE KNOWLEDGE GRAPH ---
+        st.markdown("### 🗺️ Knowledge Graph Explorer")
+        kg_html_path = build_knowledge_graph(results["papers"], results["umls"], results["drug_safety"])
+        with open(kg_html_path, 'r', encoding='utf-8') as f:
+            kg_html = f.read()
+        components.html(kg_html, height=600)
+
         # Export as CSV
         if results["papers"]:
             df = pd.DataFrame(results["papers"])