Update app.py

app.py

@@ -1,45 +1,54 @@
 import streamlit as st
 import time
-from playwright.sync_api import sync_playwright
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+from bs4 import BeautifulSoup
 import pandas as pd
 
-# Function to launch Playwright and scrape Fiverr for top 10 sellers
+# Function to launch Selenium and scrape Fiverr for top 10 sellers
 def fiverr_search(query):
-    # Start Playwright browser
-    with sync_playwright() as p:
-        browser = p.chromium.launch(headless=True)
-        page = browser.new_page()
-
-        # Go to Fiverr
-        page.goto("https://www.fiverr.com/")
-        
-        # Search for the given query
-        page.fill("input[name='search'] ", query)
-        page.click("button[data-testid='search-btn']")
-        time.sleep(3)  # Allow search results to load
-
-        # Extract top 10 sellers
-        seller_names = page.query_selector_all(".gig-card-layout .seller-name")
-        seller_titles = page.query_selector_all(".gig-card-layout .gig-title")
-        prices = page.query_selector_all(".gig-card-layout .price")
-
-        # Data collection
-        sellers = []
-        for i in range(min(10, len(seller_names))):
-            name = seller_names[i].inner_text()
-            title = seller_titles[i].inner_text()
-            price = prices[i].inner_text().strip()
-            sellers.append([name, title, price])
-
-        browser.close()
-
-        # Return seller data as a dataframe
-        return pd.DataFrame(sellers, columns=["Seller", "Title", "Price"])
-
-# Function to contact the seller
+    # Setup for Selenium WebDriver
+    options = webdriver.ChromeOptions()
+    options.add_argument("--headless")  # Run in headless mode
+
+    # Launch Chrome browser using WebDriver
+    driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
+    
+    # Go to Fiverr and search for the service
+    driver.get("https://www.fiverr.com/")
+    time.sleep(2)  # Let the page load
+    search_box = driver.find_element(By.NAME, "search")
+    search_box.send_keys(query)
+    search_box.submit()
+    time.sleep(3)  # Wait for results to load
+
+    # Parse the page with BeautifulSoup
+    soup = BeautifulSoup(driver.page_source, "html.parser")
+
+    # Extract top 10 sellers
+    seller_names = soup.find_all("span", class_="seller-name")
+    seller_titles = soup.find_all("div", class_="gig-title")
+    prices = soup.find_all("span", class_="price")
+
+    sellers = []
+    for i in range(min(10, len(seller_names))):
+        name = seller_names[i].get_text(strip=True)
+        title = seller_titles[i].get_text(strip=True)
+        price = prices[i].get_text(strip=True)
+        sellers.append([name, title, price])
+
+    # Close the browser
+    driver.quit()
+
+    # Return seller data as a dataframe
+    return pd.DataFrame(sellers, columns=["Seller", "Title", "Price"])
+
+# Function to send message to sellers (simulated for now)
 def send_message(seller_name, question):
     # Simulating sending a message to the seller on Fiverr
-    # For now, it's just logging the action (Real contact would need login session)
+    # Real contact would need login session, but this is just for testing.
     response = f"Message to {seller_name}: {question} (Sent successfully)"
     return response