Update models/responses.py

models/responses.py

@@ -1,11 +1,148 @@
-from typing import Dict, List, Optional
-from pydantic import BaseModel
-
-class PredictResponse(BaseModel):
-    label: str
-    probabilities: Optional[Dict[str, float]] = None
-    recognized: List[str]
-    unknown: List[str]
-    missing: List[str]
-    feature_order: List[str]
-    engineered: Dict[str, Optional[float]]
+from typing import Dict, List, Optional
+from pydantic import BaseModel
+
+class PredictResponse(BaseModel):
+    label: str
+    probabilities: Optional[Dict[str, float]] = None
+    recognized: List[str]
+    unknown: List[str]
+    missing: List[str]
+    feature_order: List[str]
+    engineered: Dict[str, Optional[float]]
+
+
+# Modelos para los datasets de NASA
+class ExoplanetCumulative(BaseModel):
+    """Modelo para exoplanetas del dataset Cumulative (Kepler)"""
+    kepid: Optional[int] = None
+    kepoi_name: Optional[str] = None
+    kepler_name: Optional[str] = None
+    koi_disposition: Optional[str] = None
+    koi_pdisposition: Optional[str] = None
+    koi_score: Optional[float] = None  # Score de confianza (0-1)
+    
+    # 11 características esenciales
+    koi_period: Optional[float] = None  # Periodo orbital (días)
+    koi_duration: Optional[float] = None  # Duración del tránsito (horas)
+    koi_depth: Optional[float] = None  # Profundidad del tránsito (ppm)
+    koi_impact: Optional[float] = None  # Parámetro de impacto
+    koi_prad: Optional[float] = None  # Radio del planeta (radios terrestres)
+    koi_slogg: Optional[float] = None  # Gravedad superficial estelar (log10(cm/s²))
+    koi_sma: Optional[float] = None  # Semi-eje mayor (AU)
+    koi_smet: Optional[float] = None  # Metalicidad estelar [Fe/H]
+    koi_srad: Optional[float] = None  # Radio estelar (radios solares)
+    koi_steff: Optional[float] = None  # Temperatura efectiva estelar (K)
+    koi_snr: Optional[float] = None  # Relación señal-ruido
+    
+    # Características orbitales adicionales
+    koi_eccen: Optional[float] = None  # Excentricidad orbital
+    koi_incl: Optional[float] = None  # Inclinación orbital (grados)
+    koi_dor: Optional[float] = None  # Distancia planeta-estrella / radio estelar
+    koi_ror: Optional[float] = None  # Radio planeta / radio estrella
+    
+    # Características estelares adicionales
+    koi_smass: Optional[float] = None  # Masa estelar (masas solares)
+    koi_kepmag: Optional[float] = None  # Magnitud en banda Kepler
+    
+    # Flujo y temperatura
+    koi_insol: Optional[float] = None  # Flujo de insolación (flujo terrestre)
+    koi_teq: Optional[float] = None  # Temperatura de equilibrio (K)
+    
+    # Coordenadas
+    ra: Optional[float] = None  # Ascensión recta
+    dec: Optional[float] = None  # Declinación
+
+
+class ExoplanetK2(BaseModel):
+    """Modelo para exoplanetas del dataset K2"""
+    pl_name: Optional[str] = None
+    hostname: Optional[str] = None
+    k2_name: Optional[str] = None
+    epic_hostname: Optional[str] = None
+    disposition: Optional[str] = None
+    discoverymethod: Optional[str] = None
+    disc_year: Optional[int] = None
+    
+    # IDs y nombres alternativos
+    hd_name: Optional[str] = None  # ID del catálogo Henry Draper
+    hip_name: Optional[str] = None  # ID del catálogo Hipparcos
+    tic_id: Optional[str] = None  # TESS Input Catalog ID
+    gaia_id: Optional[str] = None  # Gaia DR2/DR3 ID
+    
+    # Datos del planeta para visualización
+    pl_orbper: Optional[float] = None  # Periodo orbital (días)
+    pl_orbsmax: Optional[float] = None  # Semi-eje mayor (AU)
+    pl_rade: Optional[float] = None  # Radio del planeta (radios terrestres)
+    pl_bmasse: Optional[float] = None  # Masa del planeta (masas terrestres)
+    pl_bmassj: Optional[float] = None  # Masa del planeta (masas jovianas)
+    pl_dens: Optional[float] = None  # Densidad del planeta (g/cm³)
+    pl_orbeccen: Optional[float] = None  # Excentricidad orbital
+    pl_eqt: Optional[float] = None  # Temperatura de equilibrio (K)
+    pl_insol: Optional[float] = None  # Flujo de insolación (flujo terrestre)
+    pl_orbincl: Optional[float] = None  # Inclinación orbital (grados)
+    
+    # Datos estelares
+    st_teff: Optional[float] = None  # Temperatura efectiva estelar (K)
+    st_rad: Optional[float] = None  # Radio estelar (radios solares)
+    st_mass: Optional[float] = None  # Masa estelar (masas solares)
+    st_met: Optional[float] = None  # Metalicidad estelar [Fe/H]
+    st_logg: Optional[float] = None  # Gravedad superficial estelar
+    st_dens: Optional[float] = None  # Densidad estelar (g/cm³)
+    st_lum: Optional[float] = None  # Luminosidad estelar (luminosidades solares)
+    st_age: Optional[float] = None  # Edad estelar (Gyr)
+    
+    # Datos del sistema
+    sy_dist: Optional[float] = None  # Distancia al sistema (parsecs)
+    sy_vmag: Optional[float] = None  # Magnitud V
+    sy_kmag: Optional[float] = None  # Magnitud K
+    sy_gaiamag: Optional[float] = None  # Magnitud Gaia
+    
+    # Coordenadas
+    ra: Optional[float] = None
+    dec: Optional[float] = None
+
+
+class ExoplanetTOI(BaseModel):
+    """Modelo para exoplanetas del dataset TESS Objects of Interest (TOI)"""
+    toi: Optional[float] = None
+    tid: Optional[int] = None  # TESS Input Catalog ID
+    ctoi_alias: Optional[str] = None
+    tfopwg_disp: Optional[str] = None  # Disposición del grupo de trabajo
+    pl_pnum: Optional[int] = None  # Número de planetas en el sistema
+    
+    # Datos del planeta para visualización
+    pl_orbper: Optional[float] = None  # Periodo orbital (días)
+    pl_rade: Optional[float] = None  # Radio del planeta (radios terrestres)
+    pl_eqt: Optional[float] = None  # Temperatura de equilibrio (K)
+    pl_insol: Optional[float] = None  # Flujo de insolación (flujo terrestre)
+    pl_trandep: Optional[float] = None  # Profundidad del tránsito (ppm)
+    pl_trandur: Optional[float] = None  # Duración del tránsito (horas) - DEPRECATED
+    pl_trandurh: Optional[float] = None  # Duración del tránsito (horas)
+    pl_imppar: Optional[float] = None  # Parámetro de impacto
+    pl_orbsmax: Optional[float] = None  # Semi-eje mayor (AU)
+    
+    # Datos estelares
+    st_teff: Optional[float] = None  # Temperatura efectiva estelar (K)
+    st_rad: Optional[float] = None  # Radio estelar (radios solares)
+    st_mass: Optional[float] = None  # Masa estelar (masas solares)
+    st_met: Optional[float] = None  # Metalicidad estelar [Fe/H]
+    st_logg: Optional[float] = None  # Gravedad superficial estelar
+    st_dist: Optional[float] = None  # Distancia (parsecs)
+    st_tmag: Optional[float] = None  # Magnitud TESS
+    
+    # Movimiento propio
+    st_pmra: Optional[float] = None  # Movimiento propio en ascensión recta (mas/yr)
+    st_pmdec: Optional[float] = None  # Movimiento propio en declinación (mas/yr)
+    
+    # Coordenadas
+    ra: Optional[float] = None
+    dec: Optional[float] = None
+    rastr: Optional[str] = None  # Ascensión recta (string)
+    decstr: Optional[str] = None  # Declinación (string)
+
+
+class DatasetListResponse(BaseModel):
+    """Respuesta para listado de exoplanetas"""
+    total: int
+    count: int
+    data: List[Dict]