Update app.py

app.py

@@ -6,10 +6,12 @@ from fastapi import FastAPI
 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 pathlib import Path
 import pandas as pd
 from fpdf import FPDF
 import asyncio
+import plotly.express as px
 
 ROOT_DIR = Path(__file__).resolve().parent
 LOGO_PATH = ROOT_DIR / "assets" / "logo.png"
@@ -60,7 +62,26 @@ def generate_pdf(papers):
 def render_ui():
     st.set_page_config(page_title="MedGenesis AI", layout="wide")
 
-    # Header with logo and branding
+    # --- SIDEBAR WORKSPACE ---
+    with st.sidebar:
+        st.header("🗂️ Your Workspace")
+        saved_queries = get_workspace()
+        if saved_queries:
+            for i, item in enumerate(saved_queries, 1):
+                with st.expander(f"{i}. {item['query']}"):
+                    st.write("**AI Summary:**", item["result"]["ai_summary"])
+                    st.write("**First Paper:**", item["result"]["papers"][0]["title"] if item["result"]["papers"] else "None")
+                    df = pd.DataFrame(item["result"]["papers"])
+                    st.download_button(
+                        label="📥 Download as CSV",
+                        data=df.to_csv(index=False),
+                        file_name=f"workspace_query_{i}.csv",
+                        mime="text/csv",
+                    )
+        else:
+            st.info("Run a search and save it here!")
+
+    # --- MAIN APP HEADER ---
     col1, col2 = st.columns([0.15, 0.85])
     with col1:
         if LOGO_PATH.exists():
@@ -85,6 +106,7 @@ def render_ui():
         with st.spinner("Thinking... Gathering and analyzing data across 5 systems..."):
             results = asyncio.run(orchestrate_search(query))
             st.success("Search complete! 🎉")
+            save_query(query, results)  # Save to workspace
 
     if results:
         # Papers
@@ -93,6 +115,12 @@ def render_ui():
             st.markdown(f"**{i}. [{paper['title']}]({paper['link']})**  \n*{paper['authors']}* ({paper['source']})")
             st.markdown(f"<div style='font-size: 0.9em; color: gray'>{paper['summary']}</div>", unsafe_allow_html=True)
 
+        # --- INTERACTIVE VISUALIZATION: Publication Year Histogram ---
+        pub_years = [p["published"] for p in results["papers"] if p.get("published")]
+        if pub_years:
+            fig = px.histogram(pub_years, nbins=10, title="Publication Year Distribution")
+            st.plotly_chart(fig)
+
         # Export as CSV
         if results["papers"]:
             df = pd.DataFrame(results["papers"])