added extra page
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user