Upload st_data_parser.py

st_data_parser.py

@@ -4,6 +4,8 @@
     2. 目前docx模块在huggingface的python3.10报错。暂时不支持docx文件。
     
 """
+##TODO： 完成txt文件的上传功能。
+
 # -*- coding: utf-8 -*-
 import numpy as np
 import pandas as pd
@@ -72,6 +74,20 @@ def parser(file):
         df = pd.read_json(json_file)
         file_content = df.to_string()
         # print('file_content:', file_content)
+        
+    elif '.txt' in file.name:
+        print('txt file detected')
+        
+        ### 本质上txt文件和json文件的处理方式是一样的。        
+        txt_file = file.getvalue() ## 在streamlit中获得上传文件的json文件内容。这里不能用file.read().
+        # txt_file = txt_file.decode('utf-8') ### working. 
+        txt_file = txt_file.encode('utf-8')
+        
+        ## 保障是utf-8的格式
+        # with open(file.name, 'r', encoding='utf-8') as f:
+        #     txt_file = f.read()
+        
+        file_content = txt_file
 
     return file_content