Update main.py

main.py

@@ -85,25 +85,33 @@ def bot():
     user_id = request.json.get("user_id")
     user_question = request.json.get("user_question")
     load_dotenv()
-    inventory_ref = db.collection("test_users").document(user_id).collection('inventory')
+    
+    inventory_ref = db.collection("system_users").document(user_id).collection('inventory')
 
+    tasks_ref = db.collection("system_users").document(user_id).collection('tasks')
 
-    sales_ref = db.collection("test_users").document(user_id).collection('sales')
+    transactions_ref = db.collection("system_users").document(user_id).collection('transactions')
 
     inventory_list = []
     for doc in inventory_ref.stream():
         a = doc.to_dict()
         inventory_list.append(a)
 
-    sales_list = []
-    for doc in sales_ref.stream():
+    tasks_list = []
+    for doc in tasks_ref.stream():
         a = doc.to_dict()
-        sales_list.append(a)
-
+        tasks_list.append(a)
+        
+    transactions_list = []
+    for doc in transactions_ref.stream():
+        a = doc.to_dict()
+        transactions_list.append(a)
+        
     inventory_df = pd.DataFrame(inventory_list)
-    sales_df = pd.DataFrame(sales_list)
+    transactions_df = pd.DataFrame(transactions_list)
+    tasks_ref = pd.DataFrame(tasks_list)
 
-    lake = SmartDatalake([inventory_df, sales_df], config={"llm":llm, "response_parser":FlaskResponse, "enable_cache": False, "save_logs":False})
+    lake = SmartDatalake([inventory_df, transactions_df, tasks_df], config={"llm":llm, "response_parser":FlaskResponse, "enable_cache": False, "save_logs":False})
     response = lake.chat(user_question)
     print(user_question)