Upload folder using huggingface_hub

home.py

@@ -26,6 +26,7 @@ from plot_helper import (
     MAX_CHARTS_IN_PAGE,
     NUM_COLS,
     build_trend_figure,
+    render_vitals_plot_layout,
     reset_trends,
     update_trends,
 )
@@ -278,7 +279,7 @@ def format_patient_info(patient):
 
 async def load_patient_bundle(
     patient_id: str,
-) -> Tuple[pd.DataFrame, List[str], Dict[str, Any], List[Dict[str, Any]]]:
+):
     """
     Given patient_id, return:
     - reports_df
@@ -286,7 +287,9 @@ async def load_patient_bundle(
     - meta dict with patient basics
     - final_reports list
     - run_stats list
-    - vitals
+    - vitals history
+    - latest vitals
+    - vitals plots
     """
     if not patient_id:
         return pd.DataFrame(), [], {}, []
@@ -302,6 +305,7 @@ async def load_patient_bundle(
     run_stats = await db.get_run_stats_by_patient(patient_id)
     latest_vitals = await render_latest_vitals_card_layout(patient_id)
     vitals_history = await db.get_vitals_by_patient(patient_id)
+    vitals_plots = await render_vitals_plot_layout(patient_id)
 
     reports_df = flatten_reports_v2(reports)
     test_names = trends_index(trends)
@@ -322,6 +326,7 @@ async def load_patient_bundle(
         run_stats,
         vitals_history,
         latest_vitals,
+        vitals_plots,
     )
 
 
@@ -605,18 +610,39 @@ def build_home_page(
                                     label="Unit", placeholder="e.g., bpm, cm"
                                 )
 
-                            save_vitals_btn = gr.Button("Save Readings", variant="huggingface", scale=0)
+                            save_vitals_btn = gr.Button(
+                                "Save Readings", variant="huggingface", scale=0
+                            )
                             save_vitals_status = gr.Markdown()
                         # Latest Vitals
                         with gr.Row():
                             latest_vitals_cards = [
                                 gr.Label(visible=False) for _ in range(20)
                             ]
-                        # Historical Vitals
-                        vitals_df = gr.DataFrame(
-                            headers=["Date", "Name", "Value", "Unit", "AI Status"],
-                            interactive=False,
-                        )
+                        with gr.Row():
+                            # Historical Vitals
+                            with gr.Column(scale=1):
+                                vitals_df = gr.DataFrame(
+                                    headers=[
+                                        "Date",
+                                        "Name",
+                                        "Value",
+                                        "Unit",
+                                        "AI Status",
+                                    ],
+                                    interactive=False,
+                                )
+                            with gr.Column(scale=2):
+                                vitals_plots = []
+                                rows = math.ceil(
+                                    20 / 2
+                                )  # default num_cols_per_row columns per row
+                                for r in range(rows):
+                                    with gr.Row():
+                                        for c in range(
+                                            2
+                                        ):  # num_cols_per_row cols per row
+                                            vitals_plots.append(gr.Plot(visible=True))
                     with gr.Tab("🧪 Tests", id=2):
                         reports_df = gr.DataFrame(
                             headers=[
@@ -794,6 +820,7 @@ def build_home_page(
                     run_stats,
                     vitals_history,
                     latest_vitals,
+                    vitals_plots,
                 ) = await load_patient_bundle(patient_id)
                 (*trends, page, page_info, prev_btn, next_btn) = await update_trends(
                     patient_id=patient_id, page=0, num_cols=NUM_COLS
@@ -815,6 +842,7 @@ def build_home_page(
                         value=flatten_vitals(vitals_history),
                     ),
                     *latest_vitals,
+                    *vitals_plots,
                     *trends,
                     page,
                     page_info,
@@ -841,6 +869,7 @@ def build_home_page(
                         value=[],
                     ),
                     gr.update(),
+                    gr.update(),
                     *trends,
                     page,
                     page_info,
@@ -895,6 +924,7 @@ def build_home_page(
                 run_stats_df,
                 vitals_df,
                 *latest_vitals_cards,
+                *vitals_plots,
                 *chart_outputs,
                 current_page,
                 page_info,
@@ -990,7 +1020,12 @@ def build_home_page(
                 vitals_custom_value,
                 vitals_custom_unit,
             ],
-            outputs=[save_vitals_status, vitals_df, *latest_vitals_cards],
+            outputs=[
+                save_vitals_status,
+                vitals_df,
+                *latest_vitals_cards,
+                *vitals_plots,
+            ],
         )
 
         # open modal and set states
@@ -1019,6 +1054,7 @@ def build_home_page(
                 run_stats_df,
                 vitals_df,
                 *latest_vitals_cards,
+                *vitals_plots,
                 *chart_outputs,
                 current_page,
                 page_info,

modules/db.py

@@ -10,7 +10,7 @@ from motor.motor_asyncio import AsyncIOMotorGridFSBucket
 from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorGridFSBucket
 import os
 from dotenv import load_dotenv
-
+import pandas as pd
 
 class SheamiDB:
     def __init__(self, uri: str = None, db_name: str = "sheami"):
@@ -741,3 +741,29 @@ class SheamiDB:
         else:
             vitals = {}
         return vitals
+
+    async def get_vitals_trends_by_patient(self,patient_id):
+        docs = await self.vitals.aggregate([
+            {"$match": {"patient_id": ObjectId(patient_id)}},
+            {"$unwind": "$readings"},
+            {"$project": {
+                "date": 1,
+                "name": "$readings.name",
+                "value": "$readings.value",
+                "unit": "$readings.unit"
+            }},
+            {"$sort": {"date": 1}}
+        ]).to_list(length=None)
+
+        df = pd.DataFrame(docs)
+        trend_docs = []
+        if "name" in df:
+            for vital_name, group in df.groupby("name"):
+                trend_docs.append({
+                    "test_name": vital_name,
+                    "trend_data": group[["date", "value"]].to_dict("records"),
+                    "unit": group["unit"].iloc[0],  # assume consistent unit
+                    # TODO: optional: enrich with AI-based reference ranges
+                    "test_reference_range": {}
+                })
+        return trend_docs

plot_helper.py

@@ -7,6 +7,7 @@ import numpy as np
 import pandas as pd
 from common import get_db
 from ui import disable_component
+import plotly.express as px
 
 MAX_CHARTS_IN_PAGE = 40
 NUM_COLS = 4
@@ -108,7 +109,7 @@ def build_trend_figure(trend_doc: Dict[str, Any]) -> Figure:
         title=f"{trend_doc.get('test_name','')}",
     )
     fig.update_yaxes(autorange=True)
-    fig.update_xaxes(autorange=True)    
+    fig.update_xaxes(autorange=True)
     return sanitize_plotly_figure(fig)
 
 
@@ -216,3 +217,22 @@ def next_page(page, figures_len):
 
 def prev_page(page):
     return max(page - 1, 0)
+
+
+async def render_vitals_plot_layout(patient_id):
+    docs = await get_db().get_vitals_trends_by_patient(patient_id)
+    figures = [build_trend_figure(doc) for doc in docs if doc]
+    # Pad/truncate to exactly 20 charts
+    if len(figures) > 20:
+        figures = figures[:20]
+    elif len(figures) < 20:
+        while len(figures) < 20:
+            empty_fig = Figure()
+            empty_fig.update_layout(
+                title="No Data",
+                xaxis=dict(visible=False),
+                yaxis=dict(visible=False),
+                margin=dict(l=30, r=20, t=40, b=30),
+            )
+            figures.append(empty_fig)
+    return [gr.Plot(value=fig, label=fig.layout.title.text) for fig in figures]

ui.py

@@ -905,4 +905,4 @@ async def render_latest_vitals_card_layout(patient_id: str):
     while len(cards) < 20:
         cards.append(gr.Label(value="-", label="", visible=True))
 
-    return cards
+    return cards