ChenyuRabbitLove commited on
Commit
4e14d61
·
1 Parent(s): 5094d0c

feat/change write mode

Browse files
Files changed (1) hide show
  1. utils/chatbot.py +2 -3
utils/chatbot.py CHANGED
@@ -68,9 +68,8 @@ class Chatbot:
68
  self.upload_state = "done"
69
 
70
  def __write_into_db(self, service, db: pd.DataFrame, cur_content: pd.DataFrame):
71
- # db = pd.concat([db, cur_content], ignore_index=True)
72
- # db.to_csv(f"{self.uid}_knowledge_base.csv", index=False)
73
- cur_content.to_csv(f"{self.uid}_knowledge_base.csv", index=False)
74
  media = MediaFileUpload(f"{self.uid}_knowledge_base.csv", resumable=True)
75
  request = (
76
  service.files()
 
68
  self.upload_state = "done"
69
 
70
  def __write_into_db(self, service, db: pd.DataFrame, cur_content: pd.DataFrame):
71
+ db = pd.concat([db, cur_content], ignore_index=True)
72
+ db.to_csv(f"{self.uid}_knowledge_base.csv", index=False)
 
73
  media = MediaFileUpload(f"{self.uid}_knowledge_base.csv", resumable=True)
74
  request = (
75
  service.files()