Update app.py

app.py

@@ -26,6 +26,44 @@ df_bobot = pd.read_sql(f"SELECT * FROM {db_user}.bobot", mydb)
 df_bobot['BOBOT'] = [float(x.replace(",",".")) for x in df_bobot['BOBOT']]#.astype(float)
 df_bobot["JENIS_KENDARAAN"] = [x[:-2] + x[-1] if x[-2] == "." else x for x in df_bobot["JENIS_KENDARAAN"]]
 
+dict_swdkllj = {
+    "A":3000,
+    "B":23000,
+    "C1":35000,
+    "C2":83000,
+    "DP":35000,
+    "DU":73000,
+    "EP":153000,
+    "EU":90000,
+    "F":163000
+}
+
+def find_swdkllj(row):
+    cc = int(row["jumlah_cc"])
+    nm = str(row["nm_jenis_kb"]).upper()
+    
+    if nm in ["AMBULANCE", "DAMKAR", "MOBIL JENAZAH"]:
+        return "A"
+    if nm in ["ALAT BERAT"]:
+        return "B"
+    if nm in ["SEDAN", "JEEP", "MINIBUS"]:
+        return "DP"
+    
+    if "SPD. MOTOR" in nm:
+        if cc <= 50:
+            return "A"
+        elif cc <= 250:
+            return "C1"
+        else:
+            return "C2"
+    
+    if "PICK UP" in nm:
+        if cc <= 2400:
+            return "DP"
+
+df_sample["gol"] = df_sample.apply(find_swdkllj, axis=1)
+
+
 df_sample["pnbp_stnk"] = df_sample["nm_jenis_kb"].apply(
     lambda x: 100000 if x == "SPD. MOTOR R2" else 200000
 )
@@ -843,7 +881,10 @@ def respond_pkb(
               
           if scores[1] > w_avg:
               d = datetime.now().date().year - vehicle["tg_akhir_pkb"].values[0].year
-              swdkllj = 35000
+              if d < 1:
+                  d = 1
+              # swdkllj = 35000
+              swdkllj = dict_swdkllj.get(vehicle["gol"].values[0])
 
               out += f"\nPKB \t\t: {int(pkb*d):,}"
               out += f"\nSWDKLLJ \t\t: {int(swdkllj*d):,}"
@@ -858,9 +899,6 @@ def respond_pkb(
               out_k += f"\nKalkulasi : Rp{int(pkb):,} + Rp{int(swdkllj):,} + Rp{int(pnbp_stnk):,} + Rp{int(pnbp_tnkb):,}"
               out_k += f"\nTotal Pembayaran : Rp{int(pkb + swdkllj + pnbp_stnk + pnbp_tnkb):,}\n"
               
-              if d < 1:
-                  d == 1
-              
               rincian += f"|{int(pkb*d):,}|0|{int(pkb*d):,}|PKB|\n" + f"|{int(swdkllj*d):,}|0|{int(swdkllj*d):,}|SWDKLLJ|\n\n"
               rincian_pkb = "### RINCIAN PKB:\n\n| POKOK | DENDA | TOTAL |\n|-:|-:|-:|\n"
               rincian_swdkllj = "### RINCIAN SWDKLLJ:\n\n| POKOK | DENDA | TOTAL |\n|-:|-:|-:|\n"