Update app.py

app.py

@@ -181,6 +181,18 @@ model = SentenceTransformer(model_id, trust_remote_code=True)
 codes_emb = model.encode([x[6:] for x in codes])
 
 
+def censor_middle(number, num_to_hide=4):
+  number_str = str(number)
+  middle_index = len(number_str) // 2
+  start_index = middle_index - num_to_hide // 2
+  end_index = middle_index + num_to_hide // 2
+
+  censored_part = "*" * num_to_hide
+  censored_number = number_str[:start_index] + censored_part + number_str[end_index:]
+
+  return censored_number
+
+
 def get_calculation(request_code, plate_number):
     print(request_code, plate_number, "GET CALC")
     calc = detail_perhitungan.get(request_code)
@@ -196,7 +208,7 @@ def get_calculation(request_code, plate_number):
     result_detail = request_code + " - " + calc.get("name")
 
     # out = "=============================================="
-    # out += "\nWajib Pajak ingin melakukan proses berikut:\n"
+    # out += "\n\nWajib Pajak ingin melakukan proses berikut:\n"
     # out = ""
     
     return result, str(description), result_detail
@@ -210,24 +222,25 @@ def build_outputs(results, descriptions, result_details, plate_number):
     vehicle = vehicle_tax_info.get(plate_number)
 
     out = "--------------------------------------------------"
-    out = "\nPlate Number: " + plate_number + "\n"
-    out += "\n".join([k + " : " + str(v) for  k,v in vehicle.items()])
-    out += "\n----------------------------------------------"
+    out = "\n\nNomor Polisi: " + plate_number + "\n"
+    out += "\n".join([k + " : " + str(v) if k != "no_rangka" else k + " : " + censor_middle(v) for  k,v in vehicle.items()])
+    # out += "\n----------------------------------------------"
     
     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)):
         harga_jual = str(vehicle.get("harga_jual"))
-        out += f"{i+1}. {detail}\nDetail perhitungan: {desc.replace('harga_jual', harga_jual)} = {res}\n"
+        out += f"{i+1}. {detail}\nRumus: {desc}\nDetail perhitungan: {desc.replace('harga_jual', harga_jual)} = {res}\n"
 
-    out += "----------------------------\nEstimasi biaya: "
+    # out += "----------------------------\nEstimasi biaya: "
+    out += "\n\n---------------------------\n\nEstimasi Biaya:"
 
     if len(results) > 1:
         out += " + ".join([f"Rp{x}" for x in results])
         out += f" = {sum(results)}"
     else:
         out += str(results[0])
-    out += "\n----------------------------\n\n--------------------------------"
+    # out += "\n----------------------------\n\n--------------------------------"
     return out
     
 
@@ -279,12 +292,19 @@ def respond(
             request_numbers.append(undetected_code)
 
         request_numbers_copy = request_numbers
+        request_details_copy = request_details
 
         request_numbers = "\n".join(request_numbers)
         request_details = "\n".join(request_details)
         request_scores = "\n".join(request_scores)
+
+        # if len(request_numbers_copy) > 0:
+        #     for code, detail in zip(request_numbers_copy, request_details_copy):
+        #         kode_mohon = detail_perhitungan.get(code)
+        #         formula = kode_mohon.get("formula")
+        #         description = inspect.getsource(formula).split(":", 2)[-1].strip()
         
-        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
+        # 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 = [], [], []