Update app.py

app.py

@@ -5,6 +5,8 @@ import numpy as np
 import pandas as pd
 import os
 
+from datetime import datetime
+
 from sentence_transformers import SentenceTransformer
 from sentence_transformers.util import cos_sim
 from sentence_transformers import CrossEncoder
@@ -24,6 +26,14 @@ 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"]]
 
+df_sample["pnbp_stnk"] = df["nm_jenis_kb"].apply(
+    lambda x: 100000 if x == "SPD. MOTOR R2" else 200000
+)
+
+df_sample["pnbp_tnkb"] = df["nm_jenis_kb"].apply(
+    lambda x: 60000 if x == "SPD. MOTOR R2" else 100000
+)
+
 codes = """001 - Vehicle Registration (New)
 002 - Vehicle Registration Renewal
 003 - Vehicle Ownership Transfer
@@ -794,6 +804,8 @@ def respond_pkb(
             
           v_type = vehicle["nm_jenis_kb"].values[0]
           nilai_jual = vehicle["nilai_jual"].values[0]
+          pnbp_stnk = vehicle["pnbp_stnk"].values[0]
+          pnbp_tnkb = vehicle["pnbp_tnkb"].values[0]
         
           bobot = df_bobot[df_bobot["JENIS_KENDARAAN"]==v_type]["BOBOT"].values[0]
         
@@ -811,15 +823,55 @@ def respond_pkb(
           if scores[scores_argsort[0]] < threshold:
               continue
 
+          header_rincian = "| POKOK | DENDA | TOTAL ||\n|-:|-:|-:|:-|\n"
+          rincian = "\n\n### RINCIAN:\n\n" + header_rincian
+
+          is_rincian = False
+
+          out_k = ""
+
           if scores[0] > w_avg:
-              out += "\nRumus Bea Balik Nama Kendaraan Bermotor (BBNKB) : TARIF x NJKB x PENGENAAN"
-              out += f"\nKalkulasi : {tarif_bbnkb*100}% x Rp{int(nilai_jual):,} x 100%"
-              out += f"\nTotal Pembayaran : Rp{int(bbnkb):,}\n"
+              out += f"\nBBNKB \t\t: {int(bbnkb):,}"
+              
+              out_k += "\nRumus Bea Balik Nama Kendaraan Bermotor (BBNKB) : TARIF x NJKB x PENGENAAN"
+              out_k += f"\nKalkulasi : {tarif_bbnkb*100}% x Rp{int(nilai_jual):,} x 100%"
+              out_k += f"\nTotal Pembayaran : Rp{int(bbnkb):,}\n"
+              rincian += f"|{int(bbnkb):,}|0|{int(bbnkb):,}|BBNKB|\n"
+              is_rincian = True
+          else:
+              bbnkb = 0
+              
           if scores[1] > w_avg:
-              out += "\nRumus Pokok Pajak Kendaraan Bermotor (PKB) : TARIF * NJKB * BOBOT * PENGENAAN"
-              out += f"\nKalkulasi : {tarif_pkb*100}% * Rp{int(nilai_jual):,} * {bobot} * 100%"
-              out += f"\nTotal Pembayaran : Rp{int(pkb):,}\n"
-
+              d = datetime.now().date().year - datetime.strptime(vehicle["tg_akhir_pkb"].values[0], "%Y-%m-%d").date().year
+              swdkllj = 35000
+
+              out += f"\nPKB \t\t: {int(pkb*d):,}"
+              out += f"\nSWDKLLJ \t\t: {int(swdkllj*d):,}"
+              out += f"\nPNBP STNK \t\t:{int(pnbp_stnk):,}"
+              out += f"\nPNBP TNKB \t\t:{int(pnbp_tnkb):,}"
+              out += f"\nTOTAL \t\t: {int(pkb + swdkllj + pnbp_stnk + pnbp_tnkb + bbnkb):,}"
+              
+              out_k += "\nRumus Pokok Pajak Kendaraan Bermotor (PKB) : TARIF * NJKB * BOBOT * PENGENAAN"
+              out_k += f"\nKalkulasi : {tarif_pkb*100}% * Rp{int(nilai_jual):,} * {bobot} * 100%"
+              out_k += f"\nTotal Pembayaran : Rp{int(pkb):,}\n"
+              out_k += "\nRumus Total PKB: PKB + SWDKLLJ + PNBP STNK + PNBP TNKB"
+              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"
+              for i in range(d):
+                  rincian_pkb += f"|{int(pkb):,}|0|{int(pkb):,}|PKB|\n"
+                  rincian_swdkllj += f"|{int(swdkllj):,}|0|{int(swdkllj):,}|PKB|\n"
+              rincian += rincian_pkb + "\n" + rincian_swdkllj
+              is_rincian = True
+          out += out_k
+          if is_rincian:
+              out += rincian
     if not out and scores[scores_argsort[0]] >= threshold:
         if scores[0] > w_avg:
             out += "\nRumus Bea Balik Nama Kendaraan Bermotor (BBNKB) : TARIF x NJKB x PENGENAAN"