Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import re
+import inspect
 
 from sentence_transformers import SentenceTransformer
 from sentence_transformers.util import cos_sim
@@ -41,48 +42,123 @@ codes = """001 - Vehicle Registration (New)
 undetected = "099 - Other/Undetected"
 
 
-codes = """001 - Pendaftaran Kendaraan (Baru)
-002 - Pembaruan Pendaftaran Kendaraan
-003 - Alih Kepemilikan Kendaraan
-004 - Pembatalan Pendaftaran Kendaraan
-005 - Penggantian Sertifikat Pendaftaran Kendaraan yang Hilang
-006 - Pembaruan Perubahan Alamat
-007 - Koreksi Data Kendaraan
-008 - Koreksi Nama Kepemilikan
-009 - Pembayaran Pajak Kendaraan
-010 - Proses Denda Keterlambatan Pembayaran
-011 - Pembaruan Jenis/Spesifikasi Kendaraan
-012 - Pembayaran Pajak Kendaraan Melalui E-Samsat
-013 - Penerbitan STNK (Sertifikat Pendaftaran Kendaraan)
-014 - Pembaruan STNK
-015 - Pemeriksaan Kelayakan Jalan Kendaraan Bermotor
-016 - Pembaruan Nomor Plat Kendaraan
-017 - Penggantian Plat yang Hilang
-018 - Pendaftaran Ekspor Kendaraan
-019 - Pendaftaran Impor Kendaraan
-020 - Pendaftaran Kendaraan Armada
-021 - Pembaruan Pendaftaran Kendaraan Massal
-022 - Bantuan Asuransi Kendaraan
-023 - Pelaporan Kecelakaan Kendaraan
-024 - Deklarasi Perubahan Penggunaan Kendaraan (misalnya, pribadi ke komersial)
-025 - Verifikasi Dokumen Hukum
-026 - Alih Kepemilikan Kendaraan Warisan
-027 - Penangguhan Sementara STNK
-028 - Pembaruan Dokumen Bukti Kepemilikan
-029 - Pemeriksaan Riwayat Kepemilikan Kendaraan
-030 - Permintaan Perhitungan Ulang Pajak Kendaraan
-031 - Permohonan Pembebasan Pajak (untuk kasus khusus)
-032 - Alih Kepemilikan Kendaraan Pemilik yang Meninggal""".split("\n")
+# codes = """001 - Pendaftaran Kendaraan (Baru)
+# 002 - Pembaruan Pendaftaran Kendaraan
+# 003 - Alih Kepemilikan Kendaraan
+# 004 - Pembatalan Pendaftaran Kendaraan
+# 005 - Penggantian Sertifikat Pendaftaran Kendaraan yang Hilang
+# 006 - Pembaruan Perubahan Alamat
+# 007 - Koreksi Data Kendaraan
+# 008 - Koreksi Nama Kepemilikan
+# 009 - Pembayaran Pajak Kendaraan
+# 010 - Proses Denda Keterlambatan Pembayaran
+# 011 - Pembaruan Jenis/Spesifikasi Kendaraan
+# 012 - Pembayaran Pajak Kendaraan Melalui E-Samsat
+# 013 - Penerbitan STNK (Sertifikat Pendaftaran Kendaraan)
+# 014 - Pembaruan STNK
+# 015 - Pemeriksaan Kelayakan Jalan Kendaraan Bermotor
+# 016 - Pembaruan Nomor Plat Kendaraan
+# 017 - Penggantian Plat yang Hilang
+# 018 - Pendaftaran Ekspor Kendaraan
+# 019 - Pendaftaran Impor Kendaraan
+# 020 - Pendaftaran Kendaraan Armada
+# 021 - Pembaruan Pendaftaran Kendaraan Massal
+# 022 - Bantuan Asuransi Kendaraan
+# 023 - Pelaporan Kecelakaan Kendaraan
+# 024 - Deklarasi Perubahan Penggunaan Kendaraan (misalnya, pribadi ke komersial)
+# 025 - Verifikasi Dokumen Hukum
+# 026 - Alih Kepemilikan Kendaraan Warisan
+# 027 - Penangguhan Sementara STNK
+# 028 - Pembaruan Dokumen Bukti Kepemilikan
+# 029 - Pemeriksaan Riwayat Kepemilikan Kendaraan
+# 030 - Permintaan Perhitungan Ulang Pajak Kendaraan
+# 031 - Permohonan Pembebasan Pajak (untuk kasus khusus)
+# 032 - Alih Kepemilikan Kendaraan Pemilik yang Meninggal""".split("\n")
+
+codes = """001 - Pendaftaran Kendaraan
+002 - Pembaruan Data Kendaraan
+003 - Alih atau Pembatalan Kepemilikan
+004 - Penggantian Dokumen atau Plat yang Hilang
+005 - Pembayaran dan Pengelolaan Pajak Kendaraan
+006 - Pemeriksaan dan Verifikasi Kendaraan
+007 - Pendaftaran Kendaraan Ekspor, Impor, atau Armada
+008 - Pelaporan dan Bantuan Terkait Kendaraan
+009 - Penangguhan atau Deklarasi Perubahan Penggunaan Kendaraan"""
+
+
+vehicle_tax_info = {
+    "B 1234 BCA": {
+        "no_rangka": "1237191234",
+        "type": "SUV",
+        "tanggal": "23 Desember 2024",
+        "status": "Belum Bayar",
+        "harga_jual": 150000  # In Rupiah
+    },
+    "B 5678 XYZ": {
+        "no_rangka": "9876543210",
+        "type": "Sedan",
+        "tanggal": "15 Januari 2025",
+        "status": "Belum Bayar",
+        "harga_jual": 300000  # In Rupiah
+    },
+    "D 3456 DEF": {
+        "no_rangka": "4561237890",
+        "type": "MPV",
+        "tanggal": "10 Februari 2025",
+        "status": "Sudah Bayar",
+        "harga_jual": 250000  # In Rupiah
+    }
+}
+
+# Table for detail calculations (perhitungan)
+detail_perhitungan = {
+    "001": {
+        "name": "Pendaftaran Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.1  # Example formula: 10% of harga_jual
+    },
+    "002": {
+        "name": "Pembaruan Data Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.05  # Example formula: 5% of harga_jual
+    },
+    "003": {
+        "name": "Alih Kepemilikan Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.1  # Example formula: 10% of harga_jual
+    },
+    "004": {
+        "name": "Penggantian Dokumen atau Plat yang Hilang",
+        "formula": lambda harga_jual: harga_jual * 0.03  # Example formula: 3% of harga_jual
+    },
+    "005": {
+        "name": "Pembayaran dan Pengelolaan Pajak Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.12  # Example formula: 12% of harga_jual
+    },
+    "006": {
+        "name": "Pemeriksaan dan Verifikasi Kendaraan",
+        "formula": lambda harga_jual: 100000  # Example formula: 2% of harga_jual
+    },
+    "007": {
+        "name": "Pendaftaran Kendaraan Ekspor, Impor, atau Armada",
+        "formula": lambda harga_jual: harga_jual * 0.15  # Example formula: 15% of harga_jual
+    },
+    "008": {
+        "name": "Pelaporan dan Bantuan Terkait Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.04  # Example formula: 4% of harga_jual
+    },
+    "009": {
+        "name": "Penangguhan atau Deklarasi Perubahan Penggunaan Kendaraan",
+        "formula": lambda harga_jual: harga_jual * 0.06  # Example formula: 6% of harga_jual
+    }
+}
 
 undetected = "099 - Lainnya/Tidak Terdeteksi"
 
 model_ids = [
-    "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
     "Alibaba-NLP/gte-multilingual-base",
     "BAAI/bge-m3",
     "sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
     "intfloat/multilingual-e5-small",
-    "sentence-transformers/distiluse-base-multilingual-cased-v2"
+    "sentence-transformers/distiluse-base-multilingual-cased-v2",
+    "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
 ]
 # model_id = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
 # model_id = "Alibaba-NLP/gte-multilingual-base"
@@ -95,6 +171,48 @@ model = SentenceTransformer(model_id, trust_remote_code=True)
 
 codes_emb = model.encode([x[6:] for x in codes])
 
+
+def get_calculation(request_code, plate_number):
+    calc = detail_perhitungan.get(request_code)
+    vechicle = vehicle_tax_info.get(plate_number)
+    
+    if vehicle == None or calc == None:
+        return None, None, None
+        
+    harga_jual = vechicle.get("harga_jual")
+    formula = calc.get("formula")
+    result = formula(harga_jual)
+    description = inspect.getsource(formula).split(":")[-1].strip()
+    result_detail = request_code + " - " + calc.get("name")
+
+    # out = "=============================================="
+    # out += "\nWajib Pajak ingin melakukan proses berikut:\n"
+    # out = ""
+    
+    return result, description, result_detail
+
+
+def build_output(result, description, result_detail, plate_number):
+    return build_outputs([result], [description], [result_detail], plate_number)
+
+
+def build_outputs(results, descriptions, result_details, plate_number):
+    out = "=============================================="
+    out += f"\nWajib Pajak dengan NoPol {plate_number} ingin melakukan proses berikut:\n"
+
+    for i, (res,desc,detail) in enumerate(zip(results, descriptions, result_details)):
+        out += f"{i+1}. {detail}\nDetail perhitungan:\n{desc.replace("harga_jual", str(res))}\n"
+
+    out += "Estimasi biaya: "
+
+    if len(results) > 1:
+        out += " + ".join([f"Rp{x}" for x in results])
+        out += f" = {sum(results)}"
+    else:
+        out += str(results[0])
+    return out
+    
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -121,7 +239,9 @@ def respond(
 
     matches = re.findall(pattern, message)
 
-    plate_numbers = ", ".join([" ".join(x) for i,x in enumerate(matches)]).upper()
+    plates = [" ".join(x).upper() for i,x in enumerate(matches)]
+
+    plate_numbers = ", ".join(plates)
     
     text_emb = model.encode(message)
     scores = cos_sim(codes_emb, text_emb)[:,0]
@@ -143,8 +263,25 @@ def respond(
         request_numbers = "\n".join(request_numbers)
         request_details = "\n".join(request_details)
         request_scores = "\n".join(request_scores)
+        
+        out = "Request code number:\n" + request_numbers + "\n\nRequest detail:\n" + request_details + f"\n\nConfidence score:\n{request_scores}" + "\n\nPlate numbers: " + plate_numbers
+
+        for plate in plates:
+            results = descriptions = result_details = []
+            for code in request_numbers:
+                result, description, result_detail = get_calculation(code, plate)
+
+                if result != None:
+                    results.append(result)
+                    descriptions.append(description)
+                    result_details.append(result_detail)
+
+            out += "\n\n" + build_outputs(results, descriptions, result_details, plate)
+
+        return out
+
+        # result, description, result_detailget_calculation(request_code, plate_number)
 
-        return "Request code number:\n" + request_numbers + "\n\nRequest detail:\n" + request_details + f"\n\nConfidence score:\n{request_scores}" + "\n\nPlate numbers: " + plate_numbers
 
 
     s_max = scores.argmax()
@@ -156,8 +293,23 @@ def respond(
         request_code = codes[scores.argmax()]
         # "{:.2f}".format(a)
 
-    return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + f"\nConfidence score: {round(scores[s_max].tolist(),3)}" + "\nPlate numbers: " + plate_numbers
+    out = "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + f"\nConfidence score: {round(scores[s_max].tolist(),3)}" + "\nPlate numbers: " + plate_numbers
+
+    for plate in plates:
+        results = descriptions = result_details = []
+        result, description, result_detail = get_calculation(code, plate)
+
+        if result != None:
+            results.append(result)
+            descriptions.append(description)
+            result_details.append(result_detail)
 
+        out += "\n\n" + build_outputs(results, descriptions, result_details, plate)
+
+    return out
+    
+    # if vehicle_tax_info.get(request_detail)
+    
     # for val in history:
     #     if val[0]:
     #         messages.append({"role": "user", "content": val[0]})
@@ -209,6 +361,8 @@ with gr.Blocks() as demo:
     gr.Markdown("<br>".join(codes) + "<br>" + undetected)
     gr.Markdown("# Valid License Plate Number Criteria:")
     gr.Markdown("(1-2 letters) (4 numbers) (1-3 letters)")
+    gr.Markdown("# Example Valid Plate Numbers")
+    gr.Markdown("<br>".join(vehicle_tax_info.keys()))
 
 
     gr.Markdown("# Choose & Load Model:")