Update app.py

app.py

@@ -749,7 +749,8 @@ async def greet_json(body: InputData):
 @app.get("/get_data")
 def get_data(nopol: str):
     data_kendaraan, total_tagihan, rincians_pkb, rincians_swd = get_vehicle_info_bs4(nopol)
-    data_kendaraan["no_polisi"] = nopol
+    if data_kendaraan:
+        data_kendaraan["no_polisi"] = nopol
     return {
         "data_kendaraan":data_kendaraan,
         "rincian_total_tagihan":total_tagihan,
@@ -774,6 +775,10 @@ async def get_data_json(body: InputData):
     for m in message:
     
         data_kendaraan, total_tagihan, rincians_pkb, rincians_swd = get_vehicle_info_bs4(m)
+        if not data_kendaraan:
+            string += "\n------\nError: Kendaraan dengan Plat Nomor {m} tidak ditemukan!\n------\n\n"
+            continue
+            
         data_kendaraan["no_polisi"] = m
     
         string += "\n---*DATA KENDARAAN:*\n"