Update app.py

app.py

@@ -7,6 +7,8 @@ import numpy as np
 import pandas as pd
 import os
 
+from extract import get_vehicle_info
+
 from datetime import datetime
 
 from sentence_transformers import SentenceTransformer
@@ -729,4 +731,15 @@ async def greet_json(body: InputData):
         "success":True,
         "message":"Request was successful",
         "data":{"results":{"text":out}},
+    }
+
+
+@app.get("/get_data")
+def get_data(nopol: str):
+    data_kendaraan, total_tagihan, rincians_pkb, rincians_swd = get_vehicle_info(nopol)
+    return {
+        "data_kendaraan":data_kendaraan,
+        "total_tagihan":total_tagihan,
+        "rincian_pkb":[x for x in rincians_pkb if any(x.values())]
+        "rincian_swd":[x for x in rincians_swd if any(x.values())]
     }