Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -145,7 +145,7 @@ def df_to_dict_list(df: pd.DataFrame) -> list:
|
|
| 145 |
|
| 146 |
@app.get("/kepler", response_model=DatasetListResponse)
|
| 147 |
async def get_kepler_exoplanets(
|
| 148 |
-
limit: int = Query(
|
| 149 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 150 |
koi_disposition: Optional[str] = Query(None, description="Filtrar por disposición (CONFIRMED, FALSE POSITIVE, CANDIDATE)")
|
| 151 |
):
|
|
@@ -172,8 +172,11 @@ async def get_kepler_exoplanets(
|
|
| 172 |
if koi_disposition:
|
| 173 |
df = df[df['koi_disposition'] == koi_disposition]
|
| 174 |
|
| 175 |
-
# Aplicar paginación
|
| 176 |
-
|
|
|
|
|
|
|
|
|
|
| 177 |
|
| 178 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 179 |
columns_to_include = [
|
|
@@ -229,7 +232,7 @@ async def get_cumulative_exoplanet_by_name(kepoi_name: str):
|
|
| 229 |
|
| 230 |
@app.get("/k2", response_model=DatasetListResponse)
|
| 231 |
async def get_k2_exoplanets(
|
| 232 |
-
limit: int = Query(
|
| 233 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 234 |
disposition: Optional[str] = Query(None, description="Filtrar por disposición")
|
| 235 |
):
|
|
@@ -245,8 +248,11 @@ async def get_k2_exoplanets(
|
|
| 245 |
if disposition:
|
| 246 |
df = df[df['disposition'] == disposition]
|
| 247 |
|
| 248 |
-
# Aplicar paginación
|
| 249 |
-
|
|
|
|
|
|
|
|
|
|
| 250 |
|
| 251 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 252 |
columns_to_include = [
|
|
@@ -302,7 +308,7 @@ async def get_k2_exoplanet_by_name(pl_name: str):
|
|
| 302 |
|
| 303 |
@app.get("/tess", response_model=DatasetListResponse)
|
| 304 |
async def get_tess_exoplanets(
|
| 305 |
-
limit: int = Query(
|
| 306 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 307 |
tfopwg_disp: Optional[str] = Query(None, description="Filtrar por disposición del grupo de trabajo")
|
| 308 |
):
|
|
@@ -318,8 +324,11 @@ async def get_tess_exoplanets(
|
|
| 318 |
if tfopwg_disp:
|
| 319 |
df = df[df['tfopwg_disp'] == tfopwg_disp]
|
| 320 |
|
| 321 |
-
# Aplicar paginación
|
| 322 |
-
|
|
|
|
|
|
|
|
|
|
| 323 |
|
| 324 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 325 |
columns_to_include = [
|
|
|
|
| 145 |
|
| 146 |
@app.get("/kepler", response_model=DatasetListResponse)
|
| 147 |
async def get_kepler_exoplanets(
|
| 148 |
+
limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
|
| 149 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 150 |
koi_disposition: Optional[str] = Query(None, description="Filtrar por disposición (CONFIRMED, FALSE POSITIVE, CANDIDATE)")
|
| 151 |
):
|
|
|
|
| 172 |
if koi_disposition:
|
| 173 |
df = df[df['koi_disposition'] == koi_disposition]
|
| 174 |
|
| 175 |
+
# Aplicar paginación solo si limit está definido
|
| 176 |
+
if limit is not None:
|
| 177 |
+
df_page = df.iloc[offset:offset + limit]
|
| 178 |
+
else:
|
| 179 |
+
df_page = df.iloc[offset:]
|
| 180 |
|
| 181 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 182 |
columns_to_include = [
|
|
|
|
| 232 |
|
| 233 |
@app.get("/k2", response_model=DatasetListResponse)
|
| 234 |
async def get_k2_exoplanets(
|
| 235 |
+
limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
|
| 236 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 237 |
disposition: Optional[str] = Query(None, description="Filtrar por disposición")
|
| 238 |
):
|
|
|
|
| 248 |
if disposition:
|
| 249 |
df = df[df['disposition'] == disposition]
|
| 250 |
|
| 251 |
+
# Aplicar paginación solo si limit está definido
|
| 252 |
+
if limit is not None:
|
| 253 |
+
df_page = df.iloc[offset:offset + limit]
|
| 254 |
+
else:
|
| 255 |
+
df_page = df.iloc[offset:]
|
| 256 |
|
| 257 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 258 |
columns_to_include = [
|
|
|
|
| 308 |
|
| 309 |
@app.get("/tess", response_model=DatasetListResponse)
|
| 310 |
async def get_tess_exoplanets(
|
| 311 |
+
limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
|
| 312 |
offset: int = Query(0, ge=0, description="Offset para paginación"),
|
| 313 |
tfopwg_disp: Optional[str] = Query(None, description="Filtrar por disposición del grupo de trabajo")
|
| 314 |
):
|
|
|
|
| 324 |
if tfopwg_disp:
|
| 325 |
df = df[df['tfopwg_disp'] == tfopwg_disp]
|
| 326 |
|
| 327 |
+
# Aplicar paginación solo si limit está definido
|
| 328 |
+
if limit is not None:
|
| 329 |
+
df_page = df.iloc[offset:offset + limit]
|
| 330 |
+
else:
|
| 331 |
+
df_page = df.iloc[offset:]
|
| 332 |
|
| 333 |
# Seleccionar columnas relevantes (solo con ≤50% nulos)
|
| 334 |
columns_to_include = [
|