Update extract.py

extract.py

@@ -24,4 +24,42 @@ def take_webdata(url):
         if wd:
             wd.quit()
 
-    return Image.open(BytesIO(screenshot)) , page_title
+    return Image.open(BytesIO(screenshot)) , page_title
+
+
+def get_vehicle_info(plate_number: str):
+    # Configure headless Chrome
+    options = Options()
+    options.add_argument("--headless")
+    options.add_argument("--disable-gpu")
+    options.add_argument("--no-sandbox")
+
+    # Path to chromedriver (adjust if needed)
+    driver = webdriver.Chrome(options=options)
+
+    try:
+        # Step 1: Open the site
+        driver.get("https://www.jambisamsat.net/infopkb.html")
+        time.sleep(2)
+
+        # Step 2: Find the input and enter plate number
+        input_element = driver.find_element(By.NAME, "nopol")
+        input_element.send_keys(plate_number)
+
+        # Step 3: Submit the form
+        submit_button = driver.find_element(By.CSS_SELECTOR, 'input[type="submit"]')
+        submit_button.click()
+        time.sleep(2)
+
+        driver.implicitly_wait(5)
+        
+        page_title = driver.title
+        screenshot = driver.get_screenshot_as_png()
+
+        return Image.open(BytesIO(screenshot)) , page_title
+
+    except Exception as e:
+        return f"Error occurred: {e}"
+
+    finally:
+        driver.quit()