from pocketflow import Flow
from selenium_web_automation_tools.custom_scraper import CustomScraper

# Create a flow
flow = Flow("My Automation Flow")

# Create a scraper node
scraper = CustomScraper(
    name="MyScraper",
    url="https://example.com",
    description="Scrape the main title"
)

# Add to flow
flow.add_node(scraper)

# Run
flow.run(shared={})

from pocketflow import Flow
from selenium_web_automation_tools.custom_inputter import CustomInputter

# Create a flow
flow = Flow("Login Automation")

# Create an inputter node
inputter = CustomInputter(
    name="LoginForm",
    url="https://example.com/login",
    input_data={
        "username": "myuser",
        "password": "mypassword"
    }
)

# Set as start node
flow = Flow(start=inputter)

# Run
flow.run(shared={})