Update database.py

database.py

@@ -156,9 +156,169 @@ class DatabaseManager:
                 print("Factores de riesgo insertados")
 
             conn.commit()
+
+            # Cargar datos de demo si la BD está recién creada (sin pacientes)
+            n_patients = conn.execute("SELECT COUNT(*) FROM Patients").fetchone()[0]
+            if n_patients == 0:
+                self._seed_demo_data(conn)
+
         except Exception as e:
             print(f"Error insertando datos por defecto: {e}")
 
+    def _seed_demo_data(self, conn):
+        """Inserta datos de demostración al arrancar con una BD vacía."""
+        import random
+        from datetime import datetime, timedelta
+
+        print("=== Insertando datos de demostración ===")
+
+        def rand_consult_date(days_max=180):
+            days  = random.randint(0, days_max)
+            hours = random.randint(7, 18)
+            mins  = random.randint(0, 59)
+            return (datetime.now() - timedelta(days=days, hours=hours, minutes=mins)
+                    ).strftime("%Y-%m-%d %H:%M:%S")
+
+        # ── Usuarios de prueba ────────────────────────────────────────────
+        demo_users = [
+            ("dr_ramirez",  "doctor123", "Doctor"),
+            ("dr_perez",    "doctor123", "Doctor"),
+            ("dr_martinez", "doctor123", "Doctor"),
+            ("supervisor",  "admin456",  "Admin"),
+        ]
+        user_ids = {}
+        for username, password, role in demo_users:
+            cur = conn.execute(
+                "INSERT OR IGNORE INTO Users (username, password, role) VALUES (?,?,?)",
+                (username, self.hash_password(password), role)
+            )
+            row = conn.execute("SELECT userID FROM Users WHERE username=?", (username,)).fetchone()
+            user_ids[username] = row["userID"]
+        conn.commit()
+        print(f"  + {len(demo_users)} usuarios de prueba creados")
+
+        doctor_names = ["dr_ramirez", "dr_perez", "dr_martinez"]
+        doctor_ids   = [user_ids[n] for n in doctor_names]
+
+        # ── Pacientes ─────────────────────────────────────────────────────
+        patients_data = [
+            ("Juan Carlos Medina Castillo",    "1958-03-14", "M", "Tipo 2"),
+            ("María Elena Reyes Sánchez",      "1965-07-22", "F", "Tipo 2"),
+            ("Roberto Antonio Pérez Núñez",    "1972-11-05", "M", "Tipo 1"),
+            ("Carmen Altagracia Díaz López",   "1949-01-30", "F", "Tipo 2"),
+            ("Luis Fernando Vargas Torres",    "1980-09-18", "M", "Tipo 2"),
+            ("Ana Milagros Santos Herrera",    "1955-04-12", "F", "Tipo 2"),
+            ("Pedro Emilio Guzmán Rosario",    "1963-12-28", "M", "Tipo 1"),
+            ("Josefa Alicia Mora Taveras",     "1970-06-03", "F", "Gestacional"),
+            ("Francisco Javier Marte Féliz",   "1945-08-17", "M", "Tipo 2"),
+            ("Rosa María Almonte Encarnación", "1968-02-09", "F", "Tipo 2"),
+            ("Miguel Ángel Cabrera Paulino",   "1976-10-25", "M", "MODY"),
+            ("Esperanza del Carmen Pichardo",  "1952-05-31", "F", "Tipo 2"),
+            ("Carlos Daniel Tejada Ureña",     "1988-03-07", "M", "Tipo 1"),
+            ("Yolanda Beatriz Novas Acosta",   "1961-09-14", "F", "Tipo 2"),
+            ("Ramón Ernesto Jiménez Clase",    "1957-07-19", "M", "Tipo 2"),
+            ("Gloria Inés Santana Bido",       "1973-11-22", "F", "Tipo 2"),
+            ("Héctor Manuel Ogando Matos",     "1966-04-08", "M", "Tipo 2"),
+            ("Milagros Concepción Báez Rijo",  "1982-01-16", "F", "Tipo 1"),
+            ("Víctor Hugo Severino Cepeda",    "1948-08-04", "M", "Tipo 2"),
+            ("Luz Marina Polanco Espinal",     "1959-06-27", "F", "Tipo 2"),
+            ("Andrés Felipe Castillo Grullón", "1990-12-11", "M", "Tipo 1"),
+            ("Natividad Eugenia Pena Arias",   "1944-03-23", "F", "Tipo 2"),
+            ("Domingo Rafael Flores Medina",   "1967-10-01", "M", "Otro"),
+            ("Sandra Margarita Cruz Lizardo",  "1978-05-18", "F", "Tipo 2"),
+            ("Eugenio Antonio Lora Ferreira",  "1953-09-30", "M", "Tipo 2"),
+        ]
+        patient_ids = []
+        for i, (name, birth, gender, diab) in enumerate(patients_data):
+            cur = conn.execute(
+                "INSERT INTO Patients (createdByUserID, name, birthDate, gender, diabetesType) VALUES (?,?,?,?,?)",
+                (doctor_ids[i % len(doctor_ids)], name, birth, gender, diab)
+            )
+            patient_ids.append(cur.lastrowid)
+        conn.commit()
+        print(f"  + {len(patient_ids)} pacientes creados")
+
+        # ── Factores de riesgo ────────────────────────────────────────────
+        rf_map = {r["name"]: r["riskFactorID"]
+                  for r in conn.execute("SELECT riskFactorID, name FROM RiskFactors").fetchall()}
+
+        patient_risk_factors = {
+            "Hipertensión":            [0, 3, 4, 5, 8, 9, 13, 14, 16, 18, 19, 21, 24],
+            "Diabetes Tipo 1":         [2, 6, 12, 17, 20],
+            "Diabetes Tipo 2":         [0, 1, 3, 4, 5, 7, 8, 9, 11, 13, 14, 15, 16, 18, 19, 21, 22, 23, 24],
+            "Obesidad":                [0, 3, 8, 9, 14, 18, 21, 24],
+            "Tabaquismo":              [6, 8, 14, 18, 22],
+            "Sedentarismo":            [1, 3, 5, 9, 13, 19, 21],
+            "Antecedentes Familiares": [0, 2, 4, 6, 10, 12, 15, 17, 20],
+            "Edad Avanzada":           [3, 8, 18, 21],
+            "Colesterol Alto":         [0, 4, 8, 13, 16, 18, 24],
+            "Nefropatía":              [3, 8, 18],
+        }
+        rf_count = 0
+        for rf_name, indices in patient_risk_factors.items():
+            rf_id = rf_map.get(rf_name)
+            if rf_id is None:
+                continue
+            for idx in indices:
+                if idx < len(patient_ids):
+                    conn.execute(
+                        "INSERT OR IGNORE INTO PatientsRiskFactors (patientID, riskFactorID) VALUES (?,?)",
+                        (patient_ids[idx], rf_id)
+                    )
+                    rf_count += 1
+        conn.commit()
+        print(f"  + {rf_count} factores de riesgo asignados")
+
+        # ── Consultas ─────────────────────────────────────────────────────
+        consultations = [
+            (0,  92.3, True,  "Lesiones hemorrágicas visibles en polo posterior. Evaluación urgente con retinólogo."),
+            (0,  87.1, True,  "Seguimiento: persisten microaneurismas. Control en 3 meses."),
+            (3,  95.8, True,  "Retinopatía diabética no proliferativa severa. Derivación inmediata."),
+            (3,  88.4, True,  "Exudados duros periféricos. Edema macular presente."),
+            (5,  79.2, True,  "Microaneurismas múltiples en arcada temporal superior."),
+            (8,  96.1, True,  "Retinopatía proliferativa. Neovascularización de disco. Requiere fotocoagulación."),
+            (8,  91.7, True,  "Post-fotocoagulación. Reducción de neovascularización. Seguimiento mensual."),
+            (13, 84.5, True,  "Hemorragias en llama en cuadrante nasal inferior."),
+            (14, 88.9, True,  "Exudados algodonosos y microaneurismas. RDNP moderada."),
+            (18, 93.2, True,  "Retinopatía diabética avanzada. Tracción vitreorretiniana incipiente."),
+            (21, 90.6, True,  "Microaneurismas y hemorragias puntiformes bilaterales."),
+            (24, 82.3, True,  "RDNP leve. Primera detección. Inicio de seguimiento cada 6 meses."),
+            (1,  91.5, False, "Retina sin signos de retinopatía. Control anual programado."),
+            (1,  94.2, False, "Sin cambios respecto a control anterior. Fondo de ojo normal."),
+            (2,  88.7, False, "Retina en buen estado. Se refuerza control glucémico."),
+            (4,  96.3, False, "Sin evidencia de retinopatía. Paciente con buen control metabólico."),
+            (6,  85.1, False, "Fondo de ojo sin alteraciones. Seguimiento semestral indicado."),
+            (7,  92.8, False, "Retina normal. Paciente en primer trimestre, control mensual."),
+            (9,  89.4, False, "Sin signos de RD. Se mejora el control de HbA1c."),
+            (10, 93.7, False, "Sin alteraciones retinianas. Paciente joven con buen pronóstico."),
+            (11, 87.6, False, "Retina sin lesiones. Control semestral."),
+            (12, 91.2, False, "Sin retinopatía. Visión borrosa ocasional descartada como causa retiniana."),
+            (15, 94.8, False, "Fondo de ojo normal. Mantener HbA1c < 7%."),
+            (16, 86.3, False, "Sin signos de RD. Control en 6 meses."),
+            (17, 93.1, False, "Retina sin alteraciones. Control glucémico óptimo."),
+            (19, 88.9, False, "Sin retinopatía. Se orienta sobre importancia del ejercicio físico."),
+            (20, 95.4, False, "Fondo de ojo normal. Primer control, paciente de reciente diagnóstico."),
+            (23, 90.1, False, "Retina sin lesiones. Excelente adherencia al tratamiento."),
+        ]
+        for pat_idx, confidence, has_dr, notes in consultations:
+            if pat_idx >= len(patient_ids):
+                continue
+            pid = patient_ids[pat_idx]
+            doc_id = conn.execute(
+                "SELECT createdByUserID FROM Patients WHERE patientID=?", (pid,)
+            ).fetchone()["createdByUserID"]
+            conn.execute(
+                """INSERT INTO Consultations
+                   (patientID, createdByUserID, diabeticRetinopathy, notes,
+                    confidence, rawOutput, consultationDate)
+                   VALUES (?,?,?,?,?,?,?)""",
+                (pid, doc_id, 1 if has_dr else 0, notes,
+                 confidence, confidence / 100.0, rand_consult_date())
+            )
+        conn.commit()
+        print(f"  + {len(consultations)} consultas creadas")
+        print("=== Datos de demostración listos ===")
+
     # ------------------------------------------------------------------ #
     #  UTILIDADES
     # ------------------------------------------------------------------ #
@@ -439,51 +599,6 @@ class DatabaseManager:
         finally:
             conn.close()
 
-    def get_consultation_by_id(self, consultation_id: int, user_id: int, role: str) -> Optional[Dict]:
-        """Devuelve una consulta completa con datos del paciente y factores de riesgo.
-        Los doctores solo pueden ver sus propias consultas."""
-        conn = self.get_connection()
-        try:
-            query = """
-                SELECT
-                    c.consultationID, c.patientID, c.createdByUserID,
-                    c.diabeticRetinopathy, c.notes, c.consultationDate,
-                    c.imagePath, c.confidence, c.rawOutput,
-                    p.name        AS patientName,
-                    p.birthDate   AS birthDate,
-                    p.gender      AS gender,
-                    p.diabetesType AS diabetesType,
-                    u.username    AS doctorName
-                FROM Consultations c
-                JOIN Patients p ON c.patientID = p.patientID
-                JOIN Users    u ON c.createdByUserID = u.userID
-                WHERE c.consultationID = ?
-            """
-            params = [consultation_id]
-            if role != "Admin":
-                query += " AND c.createdByUserID = ?"
-                params.append(user_id)
-
-            row = conn.execute(query, params).fetchone()
-            if not row:
-                return None
-
-            result = self._serialize(row)
-
-            # Cargar factores de riesgo del paciente
-            rf_rows = conn.execute(
-                """SELECT rf.riskFactorID, rf.name, rf.description
-                   FROM RiskFactors rf
-                   JOIN PatientsRiskFactors prf ON rf.riskFactorID = prf.riskFactorID
-                   WHERE prf.patientID = ?""",
-                (result["patientID"],)
-            ).fetchall()
-            result["riskFactors"] = [self._serialize(r) for r in rf_rows]
-
-            return result
-        finally:
-            conn.close()
-
     def get_patient_consultations(self, patient_id: int) -> List[Dict]:
         conn = self.get_connection()
         try:
@@ -529,12 +644,7 @@ class DatabaseManager:
             offset = (page - 1) * per_page
 
             rows = conn.execute(
-                f"""SELECT c.*,
-                           p.name        AS patientName,
-                           p.birthDate   AS birthDate,
-                           p.gender      AS gender,
-                           p.diabetesType AS diabetesType,
-                           u.username    AS doctorName
+                f"""SELECT c.*, p.name as patientName, u.username as doctorName
                     {base} {where}
                     ORDER BY c.consultationDate DESC
                     LIMIT ? OFFSET ?""",