Update app.py

app.py

@@ -86,6 +86,55 @@ codes = """001 - Pendaftaran Kendaraan
 009 - Penangguhan atau Deklarasi Perubahan Penggunaan Kendaraan""".split("\n")
 
 
+examples = [
+    {"code": "001", "examples": [
+        "Pendaftaran kendaraan dengan plat B 1234 XYZ",
+        "Daftar baru untuk plat kendaraan D 5678 ABC",
+        "Registrasi kendaraan plat AB 9102 DEF"
+    ]},
+    {"code": "002", "examples": [
+        "Pembaruan data untuk kendaraan plat B 2345 GHI",
+        "Update informasi kendaraan dengan plat L 6789 JKL",
+        "Perubahan data pemilik plat DK 1122 MNO"
+    ]},
+    {"code": "003", "examples": [
+        "Alih kepemilikan kendaraan dengan plat F 3344 PQR",
+        "Proses balik nama untuk kendaraan plat W 5566 STU",
+        "Perubahan pemilik pada kendaraan plat AD 7788 VWX"
+    ]},
+    {"code": "004", "examples": [
+        "Laporan kehilangan dokumen untuk kendaraan plat B 9900 YZA",
+        "Hilangnya plat nomor kendaraan D 1122 BCD",
+        "Dokumen kendaraan dengan plat AB 3344 EFG dilaporkan hilang"
+    ]},
+    {"code": "005", "examples": [
+        "Pembayaran pajak kendaraan dengan plat B 5566 HIJ",
+        "Kelola pajak tahunan kendaraan plat L 7788 KLM",
+        "Pajak kendaraan dengan plat DK 9900 NOP dibayar"
+    ]},
+    {"code": "006", "examples": [
+        "Pemeriksaan fisik kendaraan dengan plat F 1122 QRS",
+        "Verifikasi kendaraan plat W 3344 TUV",
+        "Cek kondisi kendaraan dengan plat AD 5566 WXY"
+    ]},
+    {"code": "007", "examples": [
+        "Pendaftaran kendaraan impor dengan plat B 7788 ZAB",
+        "Registrasi armada baru dengan plat D 9900 CDE",
+        "Kendaraan ekspor plat AB 1122 FGH didaftarkan"
+    ]},
+    {"code": "008", "examples": [
+        "Pelaporan terkait kendaraan dengan plat B 3344 IJK",
+        "Bantuan atas kendaraan plat L 5566 LMN",
+        "Laporkan masalah kendaraan dengan plat DK 7788 OPQ"
+    ]},
+    {"code": "009", "examples": [
+        "Penangguhan penggunaan kendaraan dengan plat F 9900 RST",
+        "Deklarasi perubahan fungsi kendaraan plat W 1122 UVW",
+        "Penggunaan baru untuk kendaraan plat AD 3344 XYZ"
+    ]}
+]
+
+
 vehicle_tax_info = {
     "B 1234 BCA": {
         "no_rangka": "1237191234",
@@ -178,7 +227,8 @@ model_ids = [
 model_id = model_ids[-1]
 model = SentenceTransformer(model_id, trust_remote_code=True)
 
-codes_emb = model.encode([x[6:] for x in codes])
+# codes_emb = model.encode([x[6:] for x in codes])
+codes_emb = model.encode([x["examples"] for x in examples]).mean(dim=-1)
 
 
 def censor_middle(number, num_to_hide=4):
@@ -450,7 +500,8 @@ def reload(chosen_model_id):
     if chosen_model_id != model_id:
         model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
         model_id = chosen_model_id
-        codes_emb = model.encode([x[6:] for x in codes])
+        # codes_emb = model.encode([x[6:] for x in codes])
+        codes_emb = model.encode([x["examples"] for x in examples]).mean(dim=-1)
         return f"Model {chosen_model_id} has been succesfully loaded!"
     return f"Model {chosen_model_id} is ready!"