Update utility.py

utility.py

@@ -281,12 +281,13 @@ def read_datalake(user_phone, user_question):
     sales_list = [doc.to_dict() for doc in sales_ref.stream()]
     
     inventory_df = pd.DataFrame(inventory_list)
+    inventory_df = inventory_df[inventory_df.columns not in ['created_at', 'intent', 'transaction_type']]
     sales_df = pd.DataFrame(sales_list)
     # return f"{inventory_df} \n\n {sales_df}"
 
-    # lake = SmartDatalake([inventory_df, sales_df], config={"llm": llm, "custom_whitelisted_dependencies":["ast"], "response_parser": FlaskResponse, "enable_cache": False, "save_logs": False})
-    # response = lake.chat(user_question)
-    response = inventory_df
+    lake = SmartDatalake([inventory_df, sales_df], config={"llm": llm, "custom_whitelisted_dependencies":["ast"], "response_parser": FlaskResponse, "enable_cache": False, "save_logs": False})
+    response = lake.chat(user_question)
+    # response = inventory_df
 
     return response