Update extract.py

extract.py

@@ -3,7 +3,7 @@ from selenium.common.exceptions import WebDriverException
 from bs4 import BeautifulSoup
 import time
 
-def extract_domain_data(domain_url):
+def extract_data(user_input, mode):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
@@ -13,7 +13,7 @@ def extract_domain_data(domain_url):
         wd = webdriver.Chrome(options=options)
         wd.set_window_size(1080, 720)
         # Construir la URL de búsqueda
-        url_busqueda = f"https://app.neilpatel.com/es/traffic_analyzer/keywords?domain={domain_url}&lang=es&locId=2724&mode=domain"
+        url_busqueda = f"https://app.neilpatel.com/es/traffic_analyzer/keywords?domain={user_input}&lang=es&locId=2724&mode={mode}"
         wd.get(url_busqueda)
         time.sleep(15)  # Espera 15 segundos para que la página se cargue completamente
         
@@ -78,4 +78,4 @@ def extract_domain_data(domain_url):
 
     # Parsear el texto filtrado
     datos_parseados = parsear_texto(lineas_filtradas)
-    return datos_parseados
+    return datos_parseados