allinaigc commited on
Commit
0189dcd
·
1 Parent(s): 20a3ee5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,6 +49,7 @@ import PyPDF2 ## read the local_KB PDF file.
49
  # import localKB_construct
50
  import save_database_info
51
  from datetime import datetime
 
52
 
53
  os.environ["OPENAI_API_KEY"] = os.environ['user_token']
54
  openai.api_key = os.environ['user_token']
@@ -618,7 +619,7 @@ def upload_file(uploaded_file):
618
  # st.write(upload_file)
619
  localKB_construct.process_file(uploaded_file, username)
620
  ## 在屏幕上展示当前知识库的信息,包括名字和加载日期。
621
- save_database_info.save_database_info(f'./{username}/database_name.csv', filename, str(datetime.now().strftime("%Y-%m-%d %H:%M")))
622
  st.markdown('新知识库解析成功,请务必刷新页面,然后开启对话 🔃')
623
  # spinner = st.empty()
624
 
 
49
  # import localKB_construct
50
  import save_database_info
51
  from datetime import datetime
52
+ import pytz
53
 
54
  os.environ["OPENAI_API_KEY"] = os.environ['user_token']
55
  openai.api_key = os.environ['user_token']
 
619
  # st.write(upload_file)
620
  localKB_construct.process_file(uploaded_file, username)
621
  ## 在屏幕上展示当前知识库的信息,包括名字和加载日期。
622
+ save_database_info.save_database_info(f'./{username}/database_name.csv', filename, str(datetime.now(pytz.timezone('Asia/Shanghai')).strftime("%Y-%m-%d %H:%M")))
623
  st.markdown('新知识库解析成功,请务必刷新页面,然后开启对话 🔃')
624
  # spinner = st.empty()
625