added extra page

This commit is contained in:
bolade
2025-11-24 19:37:28 +01:00
parent 580ad5d248
commit 8e8280bcb0
11 changed files with 1073 additions and 61 deletions
+26 -2
View File
@@ -18,6 +18,7 @@ class ContextGenerator:
self.pnoe_df = None
self.spirometry_df = None
self.seca_df = None
self.oxygenation_df = None
self.patient_info = {}
def load_data(
@@ -25,6 +26,7 @@ class ContextGenerator:
pnoe_path: str,
spirometry_path: str,
seca_path: Optional[str] = None,
oxygenation_path: Optional[str] = None,
):
"""Load all required datasets"""
self.pnoe_df = pd.read_csv(pnoe_path, delimiter=";")
@@ -33,6 +35,11 @@ class ContextGenerator:
self.seca_df = pd.read_excel(seca_path)
else:
self.seca_df = None
if oxygenation_path:
# Load muscle oxygenation data with skiprows to skip Train.Red metadata
self.oxygenation_df = pd.read_csv(oxygenation_path, skiprows=445)
else:
self.oxygenation_df = None
self._preprocess_pnoe_data()
def _preprocess_pnoe_data(self):
@@ -1375,8 +1382,25 @@ class ContextGenerator:
)
)
# Pages 12-17
for i in range(6):
# Page 12 - Muscle Oxygenation
contexts["page_12"] = {
"patient_name": self.patient_info["name"],
"page_number": 12,
}
# Generate muscle oxygenation chart if data is available
if graph_generator and self.oxygenation_df is not None:
try:
chart_str, metrics = graph_generator.generate_muscle_oxygenation_chart(
self.oxygenation_df, save_as_base64=True
)
contexts["page_12"]["muscle_oxygenation_chart"] = chart_str
contexts["page_12"].update(metrics)
except Exception as e:
print(f"Warning: Could not generate muscle oxygenation chart: {e}")
# Pages 13-17
for i in range(1, 6):
contexts[f"page_{i + 12}"] = {
"patient_name": self.patient_info["name"],
"page_number": i + 12,