feat: adding predicted ecg

src/application/dto/prediction_dto.py

@@ -22,6 +22,13 @@ class PredictionResponse(BaseModel):
     ppg_signal_id: str = Field(description="UUID of the source PPG signal.")
     predicted_sbp: float = Field(description="Predicted Systolic Blood Pressure (mmHg).")
     predicted_dbp: float = Field(description="Predicted Diastolic Blood Pressure (mmHg).")
+    predicted_ecg: Optional[list] = Field(
+        default=None,
+        description=(
+            "Synthetic ECG signal windows produced by CardioGAN. "
+            "Format: list of segments, each segment is a list of float samples (224 samples @ 125 Hz)."
+        ),
+    )
     mean_arterial_pressure: float = Field(description="Computed MAP: DBP + (SBP-DBP)/3 (mmHg).")
     pulse_pressure: float = Field(description="SBP - DBP (mmHg).")
     hypertension_stage: str = Field(
@@ -42,6 +49,7 @@ class PredictionResponse(BaseModel):
                 "ppg_signal_id": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
                 "predicted_sbp": 118.5,
                 "predicted_dbp": 76.2,
+                "predicted_ecg": [[0.12, -0.05, 0.33], [0.09, -0.11, 0.28]],
                 "mean_arterial_pressure": 90.3,
                 "pulse_pressure": 42.3,
                 "hypertension_stage": "Normal",

src/application/use_cases/get_prediction_history.py

@@ -88,11 +88,13 @@ class GetPredictionHistoryUseCase:
                 ppg_signal_id=p.ppg_signal_id,
                 predicted_sbp=p.predicted_sbp,
                 predicted_dbp=p.predicted_dbp,
+                predicted_ecg=p.predicted_ecg,
                 mean_arterial_pressure=p.mean_arterial_pressure,
                 pulse_pressure=p.pulse_pressure,
                 hypertension_stage=p.hypertension_stage,
                 model_version=p.model_version,
                 inference_time_ms=p.inference_time_ms,
+                sa_log=p.sa_log,
                 created_at=p.created_at,
             )
             for p in predictions

src/infrastructure/model/mock_model_service.py

@@ -177,6 +177,7 @@ class MockModelService(ModelService):
             ppg_signal_id=ppg_signal_id,
             predicted_sbp=round(sbp, 1),
             predicted_dbp=round(dbp, 1),
+            predicted_ecg=fake_ecg.tolist(),
             model_version=self.model_version,
             inference_time_ms=round(elapsed_ms, 2),
             sa_log=sa_log,

src/interface/api/routes/prediction_routes.py

@@ -147,6 +147,7 @@ async def get_prediction_by_signal(
         ppg_signal_id=prediction.ppg_signal_id,
         predicted_sbp=prediction.predicted_sbp,
         predicted_dbp=prediction.predicted_dbp,
+        predicted_ecg=prediction.predicted_ecg,
         mean_arterial_pressure=prediction.mean_arterial_pressure,
         pulse_pressure=prediction.pulse_pressure,
         hypertension_stage=prediction.hypertension_stage,