Upload 2 files

app/models/ecg.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+from sqlalchemy import Column, DateTime, Float, Integer, String, func
+from sqlalchemy.orm import declarative_base
+
+from sqlalchemy import JSON as JSONType  # type: ignore
+
+Base = declarative_base()
+
+
+class ECGSample(Base):
+    """
+    Database model for storing ECG signals and associated metadata.
+    """
+
+    __tablename__ = "ecg_samples"
+
+    id = Column(Integer, primary_key=True, index=True)
+    patient_id = Column(String, index=True, nullable=False)
+    created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False)
+    signal = Column(JSONType, nullable=False)
+    label = Column(String, nullable=True)
+    score = Column(Float, nullable=True)
+    alert_level = Column(String, nullable=True)
+    hr = Column(Integer, nullable=True)
+
+    # Additional metadata fields
+    device_id = Column(String, nullable=True)
+    sampling_rate = Column(Float, nullable=True)

app/models/schemas.py

@@ -0,0 +1,35 @@
+from datetime import datetime
+from typing import List, Optional
+
+try:
+    from pydantic import BaseModel, Field, ConfigDict
+    _CONFIG_DICT_SUPPORTED = True
+except ImportError:  # pragma: no cover - Pydantic v1 fallback
+    from pydantic import BaseModel, Field  # type: ignore
+    _CONFIG_DICT_SUPPORTED = False
+
+
+class ECGInferenceRequest(BaseModel):
+    patient_id: str = Field(..., example="patient-123")
+    signal: List[float] = Field(..., min_items=1, example=[0.1, 0.2, 0.3])
+    device_id: Optional[str] = Field(None, example="device-abc")
+    sampling_rate: Optional[float] = Field(None, gt=0, example=250.0)
+    age: Optional[int] = Field(None, ge=0, example=70)
+    has_prior_stroke: Optional[bool] = Field(None, example=False)
+
+
+class ECGInferenceResponse(BaseModel):
+    patient_id: str
+    label: str
+    score: float
+    alert_level: str
+    hr: Optional[int] = None
+    sample_id: int
+    created_at: datetime
+    explanations: List[str] = Field(default_factory=list)
+
+    if _CONFIG_DICT_SUPPORTED:  # type: ignore
+        model_config = ConfigDict(from_attributes=True)  # type: ignore
+    else:
+        class Config:
+            orm_mode = True