Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import logging, os, re, sys, threading
+import logging, os, sys, threading
 
 from custom_utils import connect_to_database, rag_ingestion, rag_retrieval, rag_inference
 
@@ -30,16 +30,14 @@ def invoke(openai_api_key, prompt, rag_option):
         else:
             ### Pre-retrieval processing: index filter
             ### Post-retrieval processing: result filter
-            """
             match_stage = {
                 "$match": {
-                    "address.country": re.compile(r"United States"),
-                    "accommodates": { "$gt": 1, "$lt": 3}
+                    "accommodates": { "$gt": 1, "$lt": 3},
+                    "bedrooms": { "$gt": 0, "$lt": 2}
                 }
             }
     
             additional_stages = [match_stage]
-            """
             ###
             
             search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)