Update app.py

app.py

@@ -36,27 +36,27 @@ def invoke(openai_api_key, prompt, rag_option):
 
         if (RAG_INGESTION):
             rag_ingestion(collection)
-
-        """
-        print("777")
-        search_path = "address.country"
-
-        print("888")
-        # Create a match stage
-        match_stage = {
-            "$match": {
-               search_path: re.compile(r"United States"),
-               "accommodates": { "$gt": 1, "$lt": 3}
+            return
+        else:
+            """
+            print("777")
+            search_path = "address.country"
+    
+            print("888")
+            # Create a match stage
+            match_stage = {
+                "$match": {
+                   search_path: re.compile(r"United States"),
+                   "accommodates": { "$gt": 1, "$lt": 3}
+                }
             }
-        }
-
-        print("999")
-        additional_stages = [match_stage]
-        """
-
-        print("000")
-        #result = handle_user_query(openai_api_key, query, db, collection, additional_stages)
-        return handle_user_prompt(openai_api_key, prompt, db, collection)
+    
+            print("999")
+            additional_stages = [match_stage]
+            """
+    
+            #result = handle_user_query(openai_api_key, query, db, collection, additional_stages)
+            return rag_retrieval(openai_api_key, prompt, db, collection)
 
 gr.close_all()