Update app.py

app.py

@@ -2,11 +2,24 @@ import gradio as gr
 import re
 import inspect
 import numpy as np
+import pandas as pd
 
 from sentence_transformers import SentenceTransformer
 from sentence_transformers.util import cos_sim
 from sentence_transformers import CrossEncoder
 
+import mysql.connector
+
+db_user = os.environ["DB_USER"] 
+
+mydb = mysql.connector.connect(
+  host=os.environ["DB_HOST"],
+  user=db_user,
+  password=os.environ["DB_PASSWORD"]
+)
+
+df_sample = pd.read_sql(f"SELECT * FROM {db_user}.mytable", mydb)
+df_bobot = pd.read_sql(f"SELECT * FROM {db_user}.bobot", mydb)
 
 codes = """001 - Vehicle Registration (New)
 002 - Vehicle Registration Renewal
@@ -724,6 +737,43 @@ def reload(chosen_model_id):
     return f"Model {chosen_model_id} is ready!"
     
 
+def respond_pkb(message, history: list[tuple[str, str]]):
+    tarif_pkb = 1.5/100
+    tarif_bbnkb = 0.1
+
+    pattern = r'\b([A-Za-z]{1,2})\s?(\d{2,4})\s?([A-Za-z]{1,3})\b'
+    
+    matches = re.findall(pattern, message)
+    
+    plates = [" ".join(x).upper() for i,x in enumerate(matches)]
+
+    out = ""
+    
+    for inp in plates:
+    
+        vehicle = df_sample[df_sample["no_polisi"] == inp.strip()].copy()
+        
+        if vehicle.shape[0] == 0:
+          out += f"\n\n---\nKendaraan {plate} Tidak Ditemukan\n"
+        else:
+          v_type = vehicle["nm_jenis_kb"].values[0]
+          nilai_jual = vehicle["nilai_jual"].values[0]
+        
+          bobot = df_bobot[df_bobot["JENIS KENDARAAN"]==v_type]["BOBOT"].values[0]
+        
+          bbnkb = tarif_bbnkb * nilai_jual * 1 # pengenaan
+          pkb = tarif_pkb * bobot * nilai_jual * 1 # pengenaan
+        
+          vehicle["nilai_jual"] = f"{int(nilai_jual):,}"
+          out += "\n\n---\nDetail Kendaraan:"
+          for k,v in vehicle.iloc[0].items():
+            out += f"\n{k} \t\t: {v}"
+          out += "\n=================================================================="
+          out += f"\nBea Balik Nama Kendaraan Bermotor (BBNKB) : {int(bbnkb):,}"
+          out += f"\nPajak Kendaraan Bermotor (PKB) : {int(pkb):,}\n"
+            
+    return out
+
 
 with gr.Blocks() as demo:
     # Add header title and description
@@ -750,20 +800,20 @@ with gr.Blocks() as demo:
 
     gr.Markdown("# Chatbot Interface:")
     chat_interface = gr.ChatInterface(
-        respond, 
-        additional_inputs=[
-            gr.Number(0.44, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1),
-            gr.Checkbox(True, label="multiple", info="Allow multiple request code numbers"),
-            gr.Number(3, label="maximum number of request codes", show_label=True, minimum=0, step=1, precision=0)
-        ],
-        # type="messages",
+        respond_pkb, 
         chatbot=gr.Chatbot(height=800),
-        # textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
-        # title="SamSat Virtual Assistant",
-        # description="Ask Yes Man any question",
-        # theme="soft",
-        # examples=["balik nama D 3456 DEF", "bayar pajak B 1234 BCA", "halo, selamat pagi!"],
-        # cache_examples=True,
+        # additional_inputs=[
+        #     gr.Number(0.44, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1),
+        #     gr.Checkbox(True, label="multiple", info="Allow multiple request code numbers"),
+        #     gr.Number(3, label="maximum number of request codes", show_label=True, minimum=0, step=1, precision=0)
+        # ],
+        # # type="messages",
+        # # textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
+        # # title="SamSat Virtual Assistant",
+        # # description="Ask Yes Man any question",
+        # # theme="soft",
+        # # examples=["balik nama D 3456 DEF", "bayar pajak B 1234 BCA", "halo, selamat pagi!"],
+        # # cache_examples=True,
     )
 
 if __name__ == "__main__":