Extract

from smooth import SmoothClient

client = SmoothClient()

with client.session() as session:
    session.goto("https://news.ycombinator.com")

    result = session.extract(
        schema={
            "type": "array",
            "items": {
                "type": "object",
                "properties": {
                    "title": {"type": "string", "description": "The title of the story"},
                    "url": {"type": "string", "description": "Link to the story"},
                    "points": {"type": "integer", "description": "Number of upvotes"}
                }
            }
        },
        prompt="Extract the top 5 stories from the front page"
    )

    for story in result.output:
        print(f"{story['title']} - {story['points']} points")

with client.session() as session:
    session.goto("https://example.com/product/12345")

    result = session.extract(
        schema={
            "type": "object",
            "properties": {
                "name": {"type": "string", "description": "Product name"},
                "price": {"type": "number", "description": "Price in USD"},
                "rating": {"type": "number", "description": "Average rating out of 5"},
                "reviews_count": {"type": "integer", "description": "Number of reviews"}
            }
        },
        prompt="Extract the main product details from this page"
    )

    product = result.output
    print(f"{product['name']}: ${product['price']} ({product['rating']} stars)")

with client.session() as session:
    session.goto("https://example.com/search?q=laptops")

    result = session.extract(
        schema={
            "type": "array",
            "items": {
                "type": "object",
                "properties": {
                    "name": {"type": "string", "description": "Product name"},
                    "price": {"type": "number", "description": "Price in USD"},
                    "specs": {"type": "string", "description": "Key specifications"}
                }
            }
        },
        prompt="Extract the first 10 laptops from the search results"
    )

    for laptop in result.output:
        print(f"{laptop['name']}: ${laptop['price']}")

with client.session() as session:
    session.run_task(
        task="Search for one-way flights from NYC to LA on March 15",
        url="https://www.google.com/travel/flights"
    )

    result = session.extract(
        schema={
            "type": "array",
            "items": {
                "type": "object",
                "properties": {
                    "airline": {"type": "string", "description": "Airline name"},
                    "departure": {"type": "string", "description": "Departure time"},
                    "arrival": {"type": "string", "description": "Arrival time"},
                    "price": {"type": "number", "description": "Price in USD"}
                }
            }
        },
        prompt="Extract only non-stop flights under $300, sorted by price from lowest to highest"
    )

    for flight in result.output:
        print(f"{flight['airline']}: ${flight['price']} ({flight['departure']} - {flight['arrival']})")

with client.session() as session:
    session.goto("https://shop.example.com/category/electronics")

    result = session.extract(
        schema={
            "type": "array",
            "items": {
                "type": "object",
                "properties": {
                    "name": {"type": "string", "description": "Product name"},
                    "price": {"type": "number", "description": "Price in USD"},
                    "in_stock": {"type": "boolean", "description": "Whether the item is available"}
                }
            }
        },
        prompt="Extract all products that are in stock"
    )

    # Find the cheapest in-stock item
    products = result.output
    if products:
        cheapest = min(products, key=lambda p: p['price'])
        session.run_task(f"Add '{cheapest['name']}' to cart")