from typing import Union
from fastapi import FastAPI
from src.extraction_normal import main
from pydantic import BaseModel, Field

app = FastAPI()

class Product(BaseModel):
    product: str = Field("Producto a buscar", example = "carros")
    pages: Union[int, str] = Field("Number of pages to scrape", example = "all")

@app.post("/product")
async def get_data(product: Product):
    print(product)
    data = main(product=product.product,
                      pages= product.pages)
    return data

@app.get("/heart-beat")
async def service_health():
    return {"ok"}