Update app.py

app.py

@@ -36,8 +36,7 @@ codes = """001 - Vehicle Registration (New)
 029 - Vehicle Ownership History Check
 030 - Vehicle Tax Recalculation Request
 031 - Tax Exemption Application (for special cases)
-032 - Deceased Owner’s Vehicle Ownership Transfer
-033 - Other/Undetected""".split("\n")
+032 - Deceased Owner’s Vehicle Ownership Transfer""".split("\n")
 
 
 
@@ -48,8 +47,11 @@ codes_emb = model.encode(codes)
 def respond(
     message,
     history: list[tuple[str, str]],
+    threshold
 ):
     global codes_emb
+
+    print(history)
     
     # messages = [{"role": "system", "content": system_message}]
     pattern = r'\b([A-Z]{1,2})\s?(\d{4})\s?([A-Z]{3})\b'
@@ -62,7 +64,12 @@ def respond(
     text_emb = model.encode(message)
     scores = cos_sim(codes_emb, text_emb)[:,0]
 
-    request_code = codes[scores.argmax()]
+    s_max = scores.argmax()
+
+    if scores[s_max] < threshold:
+        request_code = "033 - Other/Undetected"
+    else:
+        request_code = codes[scores.argmax()]
 
     return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + "\n Plate numbers: " + plate_numbers
 
@@ -103,10 +110,14 @@ with gr.Blocks() as demo:
     gr.Markdown("<br>".join(codes))
     gr.Markdown("# Valid License Plate Number Criteria:")
     gr.Markdown("(1-2 letter) (4 numbers) (1-3 letters)")
-    # gr.Markdown("This is a demo chat interface where you can ask questions and receive responses.")
 
     # Add chat interface
-    chat_interface = gr.ChatInterface(respond)
+    chat_interface = gr.ChatInterface(
+        respond, 
+        additional_inputs=[
+            gr.Number(0.5, label="threshold for Undetected", show_label=True, minimum=0., maximum=1.0, step=0.1)
+        ]
+    )
 
 if __name__ == "__main__":
     demo.launch()