Update extract.py

extract.py

@@ -155,10 +155,10 @@ def get_vehicle_info(plate_number: str):
         scroll_height = driver.execute_script("return document.body.scrollHeight")
         driver.set_window_size(1920, scroll_height + 200)  # force full-page height
 
-        button = wait.until(EC.element_to_be_clickable((By.ID, "show_det_pkb")))
+        button = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.ID, "show_det_pkb")))
         button.click()
         
-        button = wait.until(EC.element_to_be_clickable((By.ID, "show_det_swd")))
+        button = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.ID, "show_det_swd")))
         button.click()
         
         time.sleep(1)