Update main.py

main.py

@@ -74,10 +74,10 @@ def bot():
     user_id = request.json.get("user_id")
     user_question = request.json.get("user_question")
     load_dotenv()
-    inventory_ref = db.collection("users").document(user_id).collection('inventory')
+    inventory_ref = db.collection("test_users").document(user_id).collection('inventory')
 
 
-    sales_ref = db.collection("users").document(user_id).collection('sales')
+    sales_ref = db.collection("test_users").document(user_id).collection('sales')
 
     inventory_list = []
     for doc in inventory_ref.stream():
@@ -92,7 +92,7 @@ def bot():
     inventory_df = pd.DataFrame(inventory_list)
     sales_df = pd.DataFrame(sales_list)
 
-    lake = SmartDatalake([inventory_df, sales_df], config={"llm":llm, "response_parser":FlaskResponse, "enable_cache": False})
+    lake = SmartDatalake([inventory_df, sales_df], config={"llm":llm, "response_parser":FlaskResponse, "enable_cache": False, "enable_logging":False})
     response = lake.chat(user_question)
 
     print(user_question)