Update app.py

app.py

@@ -47,7 +47,8 @@ codes_emb = model.encode(codes)
 def respond(
     message,
     history: list[tuple[str, str]],
-    threshold
+    threshold,
+    is_multiple
 ):
     global codes_emb
     global undetected
@@ -74,6 +75,24 @@ def respond(
     text_emb = model.encode(message)
     scores = cos_sim(codes_emb, text_emb)[:,0]
 
+    if is_multiple:
+        request_details = []
+        request_numbers = []
+        for i,score in enumerate(scores):
+            if score > threshold:
+                request_details.append(codes[i][6:])
+                request_numbers.append(codes[i][:3])
+
+        if not request_details:
+            request_details.append(undetected[6:])
+            request_numbers.append(undetected_code)
+
+        request_numbers = "\n".join(request_numbers)
+        request_details = "\n".join(request_details)
+
+        return "Request code number:\n" + request_numbers + "\nRequest detail:\n" + request_details + "\nPlate numbers: " + plate_numbers
+
+
     s_max = scores.argmax()
 
     if scores[s_max] < threshold:
@@ -82,7 +101,7 @@ def respond(
     else:
         request_code = codes[scores.argmax()]
 
-    return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + "\n Plate numbers: " + plate_numbers
+    return "Request code number: " + request_code[:3] + "\nRequest detail: " + request_code[6:] + "\nPlate numbers: " + plate_numbers
 
     # for val in history:
     #     if val[0]:
@@ -126,7 +145,8 @@ with gr.Blocks() as demo:
     chat_interface = gr.ChatInterface(
         respond, 
         additional_inputs=[
-            gr.Number(0.5, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1)
+            gr.Number(0.5, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1),
+            gr.Checkbox(label="multiple", info="Allow multiple request code numbers"),
         ]
     )