修正統計資料內容

app.py

@@ -323,6 +323,9 @@ def get_hr_files():
     else:
         return make_markdown_table(files)
 
+def get_be_csv_files():
+    return make_markdown_table(glob.glob(root_file_path+believe_source_path+"/*.csv"))
+
 def update_hr_km(files):
     file_paths = [file.name for file in files]
     dest_file_path=root_file_path+hr_source_path
@@ -336,6 +339,20 @@ def update_hr_km(files):
     save_sqlite(sqlite_key, [Path(file_path).name for file_path in file_paths])
     return get_hr_files()
 
+def update_be_csv_km(files):
+    file_paths = [file.name for file in files]
+    dest_file_path=root_file_path+believe_source_path
+    if not os.path.exists(dest_file_path):
+        os.makedirs(dest_file_path)
+
+    for file in file_paths:
+        shutil.copy(file, dest_file_path)
+
+    if files == None:
+        return ''
+    else:
+        return make_markdown_table(files)
+
 def clear_all_collection(collection_name):
     pass
 
@@ -355,6 +372,9 @@ def clear_hr_datas():
     delete_sql(sqlite_key)
     return get_hr_files()
 
+def clear_be_csv_datas():
+    all_files_under_diretory(root_file_path+believe_source_path)
+
 def num_of_collection(collection_name):
     client = get_chroma_client(collection_name)
     number = client.get_collection(collection_name).count()
@@ -558,7 +578,7 @@ def read_logs():
 def pandas_analysis(prompt_str, message, chat_history):
     dir_path = f"{root_file_path}{believe_source_path}/*.csv"
     res = glob.glob(dir_path)
-    df = pd.concat(map(pd.read_csv, res))
+    df = pd.concat((pd.read_csv(f, encoding='unicode_escape') for f in res), ignore_index=True)
     local_deploy_id= "text-davinci-003"
     local_model_name = "text-davinci-003"
     llm = AzureOpenAI(
@@ -713,16 +733,10 @@ def gradio_run():
                             audio_html = gr.HTML(htm_audio, visible=False)
                         with gr.Column():
                             isAudio = gr.Checkbox(label="是否要有語音", info="要開啟語音嗎?查詢時間會增長")
-                            gr.Markdown("""
-                            ### AI 虛擬客服:
-                            * 這是一個實驗性質的AI 客服
-                            * 講話超過15秒就不會產生,正在要求放寬限制
-                            * 想要放誰的頭像都可以, 要放董事長也可以.
-                            * 訂閱制(有效時間 6/13~7/13)
-                            """)
+
 
                     with gr.Row():
-                        chatbot = gr.Chatbot(value=[], elem_id="chatbot").style(height=400)
+                        chatbot = gr.Chatbot(value=[], elem_id="chatbot").style(height=600)
                     with gr.Row():
                         with gr.Column(scale=5):
                             msg = gr.Textbox(
@@ -770,14 +784,9 @@ def gradio_run():
 建議先詢問欄位後, 後續再構思其他問題
                         """)
 
-                invField = gr.Textbox(visible=False)
-                dir_path = f"{root_file_path}{believe_source_path}/*.csv"
-                res = glob.glob(dir_path)
-                gr.Examples(res, label="資料庫檔案", inputs=invField, examples_per_page=4)
             with gr.Row():
                 with gr.Column():
-
-                    tmp_chatbot = gr.Chatbot(value=[], elem_id="tmp_chatbot").style(height=500)
+                    tmp_chatbot = gr.Chatbot(value=[], elem_id="tmp_chatbot").style(height=700)
                     with gr.Row():
                         with gr.Column(scale=5):
                             tmp_msg = gr.Textbox(show_label=False,placeholder="輸入你的問題",)
@@ -785,24 +794,27 @@ def gradio_run():
                             tmp_clear = gr.Button("清除對話")
                 with gr.Column():
                     prompt_textbox = gr.Textbox("""
-你是一位專業HR專家以及資料科學家,提供給你的資料是現有的人員資料表.
+你是一位資料科學家,提供給你的資料是2023年每一週的click 次數.
 有下列定義:
 
-1. 
+1. 欄位 23W01 代表2023年的第一個星期; 23W02 代表2023年的第二個星期
 
 請以中文回答我下面的問題:{message}
                             """, lines=10, label="Prompt:有{chat_history}及{message}, 請至少保留{message}變數",interactive=True, max_lines=10)
-                    console = gr.Textbox(lines=11, label="Console", max_lines=11)
+                    be_csv_file_list=gr.Textbox(get_be_csv_files, label="CSV Files", placeholder="沒有任何檔案存在", max_lines=10, lines=10)
+                    upload_button = gr.UploadButton("上傳統計資料(.csv格式)",file_types=[".csv"],file_count="multiple")
+                    upload_button.upload(update_be_csv_km, inputs=upload_button, outputs=be_csv_file_list).then(get_be_csv_files,outputs=be_csv_file_list)
+
+                    cleanDataBtn = gr.Button(value="刪除所有csv 資料")
+                    cleanDataBtn.click(clear_be_csv_datas, outputs=be_csv_file_list).then(get_be_csv_files, outputs=be_csv_file_list)
                 tmp_msg.submit(pandas_analysis, [prompt_textbox, tmp_msg, tmp_chatbot], [tmp_msg, tmp_chatbot],queue=True)
                 tmp_clear.click(lambda: None, None, tmp_chatbot, queue=False)
             with gr.Row():
                 gr.Examples([
-                    '你有什麼欄位?',
-                    '資料裡有屬於台灣(TW)的員工有多少位?',
-                    '全台灣的員工, 每人每天問五個問題, 1個月花費多少錢?',
-                    '如果龍華廠區的員工每人每天問3個問題,台灣員工每人每天問7個問題, 請問這樣一個月多少錢?'
+                    '你有什麼欄位?'
                     ], label="訊息範例", inputs=tmp_msg)
-
+            with gr.Row():
+                console = gr.Textbox(lines=11, label="Console", max_lines=11)
         demo.queue(concurrency_count=10)
         lunch_style(demo,console)