Update app.py

app.py

@@ -5,7 +5,7 @@ from custom_utils import connect_to_database, rag_ingestion, rag_retrieval, rag_
 
 lock = threading.Lock()
 
-RAG_INGESTION = True
+RAG_INGESTION = False
 
 RAG_OFF      = "Off"
 RAG_NAIVE    = "Naive RAG"
@@ -64,48 +64,48 @@ def invoke(openai_api_key, prompt, rag_option):
             additional_stages = [projection_stage]
             """
             ###
-            review_average_stage = {
-                "$addFields": {
-                    "averageReviewScore": {
-                        "$divide": [
-                            {
-                                "$add": [
-                                    "$review_scores.review_scores_accuracy",
-                                    "$review_scores.review_scores_cleanliness",
-                                    "$review_scores.review_scores_checkin",
-                                    "$review_scores.review_scores_communication",
-                                    "$review_scores.review_scores_location",
-                                    "$review_scores.review_scores_value",
-                                ]
-                            },
-                            6  # Divide by the number of review score types to get the average
-                        ]
-                    },
-                    # Calculate a score boost factor based on the number of reviews
-                    "reviewCountBoost": "$number_of_reviews"
-                }
-            }
+            #review_average_stage = {
+            #    "$addFields": {
+            #        "averageReviewScore": {
+            #            "$divide": [
+            #                {
+            #                    "$add": [
+            #                        "$review_scores.review_scores_accuracy",
+            #                        "$review_scores.review_scores_cleanliness",
+            #                        "$review_scores.review_scores_checkin",
+            #                        "$review_scores.review_scores_communication",
+            #                        "$review_scores.review_scores_location",
+            #                        "$review_scores.review_scores_value",
+            #                    ]
+            #                },
+            #                6  # Divide by the number of review score types to get the average
+            #            ]
+            #        },
+            #        # Calculate a score boost factor based on the number of reviews
+            #        "reviewCountBoost": "$number_of_reviews"
+            #    }
+            #}
 
-            weighting_stage = {
-                "$addFields": {
-                    "combinedScore": {
-                        # Example formula that combines average review score and review count boost
-                        "$add": [
-                            {"$multiply": ["$averageReviewScore", 0.9]},  # Weighted average review score
-                            {"$multiply": ["$reviewCountBoost", 0.1]}   # Weighted review count boost
-                        ]
-                    }
-                }
-            }
+            #weighting_stage = {
+            #    "$addFields": {
+            #        "combinedScore": {
+            #            # Example formula that combines average review score and review count boost
+            #            "$add": [
+            #                {"$multiply": ["$averageReviewScore", 0.9]},  # Weighted average review score
+            #                {"$multiply": ["$reviewCountBoost", 0.1]}   # Weighted review count boost
+            #            ]
+            #        }
+            #    }
+            #}
 
             # Apply the combinedScore for sorting
-            sorting_stage_sort = {
-                "$sort": {"combinedScore": -1}  # Descending order to boost higher combined scores
-            }
+            #sorting_stage_sort = {
+            #    "$sort": {"combinedScore": -1}  # Descending order to boost higher combined scores
+            #}
 
-            additional_stages = [review_average_stage, weighting_stage, sorting_stage_sort]
+            #additional_stages = [review_average_stage, weighting_stage, sorting_stage_sort]
             ###
-            #additional_stages = []
+            additional_stages = []
             ###
             
             search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)