main.py

from web_scraper import WebScraper

scraper = WebScraper()

while True:
    print("\n📚 ADVANCED WEB SCRAPER")
    print("1. Scrape Books")
    print("2. Search Book")
    print("3. Filter by Price")
    print("4. Sort by Price")
    print("5. Exit")

    choice = input("Enter your choice: ").strip()

    if choice == "1":
        try:
            pages = int(input("Enter number of pages: "))

            if pages <= 0:
                print("Pages must be greater than 0. ❌")
                continue

            scraper.show_books(pages)

        except ValueError:
            print("Please enter a valid number. ❌")

    elif choice == "2":
        keyword = input("Enter book name: ").strip()

        if not keyword:
            print("Search cannot be empty. ❌")
            continue

        try:
            pages = int(input("Enter number of pages: "))

            if pages <= 0:
                print("Pages must be greater than 0. ❌")
                continue

            scraper.search_book(keyword, pages)

        except ValueError:
            print("Please enter a valid number. ❌")

    elif choice == "3":
        try:
            max_price = float(input("Enter maximum price: "))
            pages = int(input("Enter number of pages: "))

            if max_price < 0 or pages <= 0:
                print("Please enter valid values. ❌")
                continue

            books = scraper.filter_by_price(max_price, pages)

            print(f"\n===== BOOKS UNDER £{max_price:.2f} =====")

            if not books:
                print("No books found.")
            else:
                for book in books:
                    print(
                        f"{book['title']} → "
                        f"£{book['price']:.2f}"
                    )

        except ValueError:
            print("Please enter valid numbers. ❌")

    elif choice == "4":
        try:
            pages = int(input("Enter number of pages: "))

            if pages <= 0:
                print("Pages must be greater than 0. ❌")
                continue

            books = scraper.get_books(pages)
            sorted_books = scraper.sort_by_price(books)

            print("\n===== 5 CHEAPEST BOOKS =====")

            for book in sorted_books[:5]:
                print(
                    f"{book['title']} → "
                    f"£{book['price']:.2f}"
                )

        except ValueError:
            print("Please enter a valid number. ❌")

    elif choice == "5":
        print("Goodbye! 👋")
        break

    else:
        print("Invalid choice!")