Update app.py

app.py

@@ -40,21 +40,25 @@ codes = """001 - Vehicle Registration (New)
 033 - Other/Undetected""".split("\n")
 
 
+
+
 model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
+codes_emb = model.encode(codes)
 
 def respond(
     message,
     history: list[tuple[str, str]],
 ):
+    global codes_emb
+    
     # messages = [{"role": "system", "content": system_message}]
     pattern = r'\b([A-Z]{1,2})\s?(\d{4})\s?([A-Z]{3})\b'
-    pattern = r'\b([A-Z]{1,2})\s?(\d{4})\s?([A-Z]{3})\b'
+    pattern = r'\b([A-Z]{1,2})\s?(\d{4})\s?([A-Z]{1,3})\b'
 
     matches = re.findall(pattern, message)
 
     plate_numbers = ", ".join([" ".join(x) for i,x in enumerate(matches)])
     
-    codes_emb = model.encode(codes)
     text_emb = model.encode(message)
     scores = cos_sim(codes_emb, text_emb)[:,0]
 
@@ -88,10 +92,21 @@ def respond(
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
-demo = gr.ChatInterface(
-    respond,
-)
+# demo = gr.ChatInterface(
+#     respond,
+# )
+
+
+with gr.Blocks() as demo:
+    # Add header title and description
+    gr.Markdown("# List of Request Numbers")
+    gr.Markdown("\n".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)
 
 if __name__ == "__main__":
     demo.launch()