Upload app.py

app.py

@@ -29,6 +29,7 @@ import pathlib
 from pathlib import Path
 from matplotlib.font_manager import FontProperties
 import seaborn as sns
+from time import sleep
 
 os.environ["OPENAI_API_KEY"] = os.environ['user_token']
 openai.api_key = os.environ['user_token']
@@ -54,17 +55,17 @@ radio_2 = col2.radio(label='模式选择', options=[
 
 # with tab2:
 def upload_file(uploaded_file):
-    # uploaded_file = st.file_uploader(
-    #     "选择一个文件", type=(["csv", "xlsx", "xls"]))
     if uploaded_file is not None:
         filename = uploaded_file.name
         st.write(filename) ## print out the whole file name to validate.
         try:
             if '.csv' in filename:
                 csv_file = pd.read_csv(uploaded_file)
+                csv_file.to_csv('./upload.csv', encoding='utf-8', index=False)
                 st.write(csv_file[:3])  # 这里只是显示文件，后面需要定位文件所在的绝对路径。
             else:
                 xls_file = pd.read_excel(uploaded_file)
+                xls_file.to_csv('./upload.csv', index=False)
                 st.write(xls_file[:3])
         except Exception as e:
             st.write(e)
@@ -73,14 +74,29 @@ def upload_file(uploaded_file):
         temp_dir = tempfile.TemporaryDirectory()
         # ! working.
         uploaded_file_path = pathlib.Path(temp_dir.name) / uploaded_file_name
-        with open(uploaded_file_path, 'wb') as output_temporary_file:
-            # output_temporary_file.write(uploaded_file.read())
+        with open('./upload.csv', 'wb') as output_temporary_file:
             # ! 必须用这种格式读入内容，然后才可以写入temporary文件夹中。
+            # output_temporary_file.write(uploaded_file.getvalue())
             output_temporary_file.write(uploaded_file.getvalue())
-            st.write(uploaded_file_path)  # * 可以查看文件是否真实存在，然后是否可以
-            output_temporary_file.close()
+            # st.write(uploaded_file_path)  # * 可以查看文件是否真实存在，然后是否可以
+            
+            # output_temporary_file.close()
+            # new_file = tempfile.NamedTemporaryFile(delete=False)
+            # new_file.write(uploaded_file.getvalue())
 
-            return uploaded_file_path
+            # os.write(output_temporary_file, uploaded_file.getvalue())
+            # temp_file = tempfile.NamedTemporaryFile(dir=str(uploaded_file_path), delete=False)
+
+
+        st.write('Now file saved successfully.')
+        # sleep(1000)
+        # fd = os.open(uploaded_file_path, os.O_WRONLY)
+        # os.write(fd, uploaded_file.getvalue())
+        # os.close(fd)
+
+        # return uploaded_file_path
+        # return new_file
+        # return new_file
 
 
 bing_search_api_key = os.environ['bing_api_key']
@@ -259,10 +275,10 @@ async def text_mode():
 
 async def data_mode(uploaded_file_path):
     print('数据分析模式启动！')
-
+    uploaded_file_path = './upload.csv'
     ## TODO: validate the existence of uploaded file. 
-    st.write(uploaded_file_path)
-    tmp1 = pd.read_csv(str(uploaded_file_path))
+    # st.write(f"passed file path in data_mode: {uploaded_file_path}")
+    tmp1 = pd.read_csv('./upload.csv')
     st.write(tmp1[:5])
 
     # Initialize chat history