Update extract.py

extract.py

@@ -116,8 +116,8 @@ def scrape_vehicle(driver):
     except Exception as e:
         print("Gagal parsing det_swd:", e)
 
-    rincians = [total_tagihan, rincians_pkb, rincians_swd]
-    return data_kendaraan, rincians
+    # rincians = [total_tagihan, rincians_pkb, rincians_swd]
+    return [data_kendaraan, total_tagihan, rincians_pkb, rincians_swd]
 
 
 def get_vehicle_info(plate_number: str):
@@ -163,14 +163,14 @@ def get_vehicle_info(plate_number: str):
         
         time.sleep(1)
 
-        data_kendaraan, rincian = scrape_vehicle(driver)
+        data = scrape_vehicle(driver)
 
         print(data_kendaraan, rincian)
 
         page_title = driver.title
         screenshot = driver.get_screenshot_as_png()
 
-        return Image.open(BytesIO(screenshot)) , page_title
+        return Image.open(BytesIO(screenshot)) , page_title, data
 
     except WebDriverException as e:
         return Image.new('RGB', (1, 1)), page_title