Update app.py

app.py

@@ -762,33 +762,46 @@ def get_data(nopol: str):
 async def get_data_json(body: InputData):
     message = body.message
 
-    data_kendaraan, total_tagihan, rincians_pkb, rincians_swd = get_vehicle_info_bs4(message)
-    data_kendaraan["no_polisi"] = nopol
+    if len(message) > 15:  
+        pattern = r'\b([A-Za-z]{1,2})\s?(\d{2,4})\s?([A-Za-z]{1,3})\b'
+        matches = re.findall(pattern, message)
+        message = [" ".join(x).upper() for i,x in enumerate(matches)]
+    else:
+        message = [message]
+    
+    string = ""
+
+    for m in message:
+    
+        data_kendaraan, total_tagihan, rincians_pkb, rincians_swd = get_vehicle_info_bs4(m)
+        data_kendaraan["no_polisi"] = nopol
+    
+        string += "\n---*DATA KENDARAAN:*\n"
+        string += "\n".join([f"{k.replace('_', ' ')} \t\t\t: {v}" for k,v in data_kendaraan.items()])
+        string += "\n\n---"
+    
+        string += "*RINCIAN:*\n"
+        string += "".join([f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" for x in total_tagihan])
+    
+        string += "\n*Rincian PKB:*\n"
+        string+= "".join(
+            [
+                f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" 
+                for x in rincians_pkb 
+                if str(x.get("total", 0)) != "0"
+            ]
+        )
+    
+        string += "\n*Rincian SWDKLLJ:*\n"
+        string += "".join(
+            [
+                f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" 
+                for x in rincians_swd 
+                if str(x.get("total", 0)) != "0"
+            ]
+        )
 
-    string = "*DATA KENDARAAN:*\n"
-    string += "\n".join([f"{k.replace("_", " ").upper()} \t\t\t: {v}" for k,v in data_kendaraan.items()])
-    string += "\n\n---"
-
-    string += "*RINCIAN:*\n"
-    string += "".join([f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" for x in total_tagihan])
-
-    string += "\n*Rincian PKB:*\n"
-    string+= "".join(
-        [
-            f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" 
-            for x in rincians_pkb 
-            if str(x.get("total", 0)) != "0"
-        ]
-    )
-
-    string += "\n*Rincian SWDKLLJ:*\n"
-    string+= "".join(
-        [
-            f"{x['jenis']}:\nPokok: Rp{x['pokok']}\nDenda: Rp{x['denda']}\nTotal: Rp{x['total']}\n---" 
-            for x in rincians_swd 
-            if str(x.get("total", 0)) != "0"
-        ]
-    )
+        string += "\n\n"
 
     return {
         "code":200,