1. 調整法務AI layout

2. 可以存或回覆預設版本prompt
3. 更新 unstructure all docs

app.py

@@ -41,7 +41,7 @@ from langchain.vectorstores import Chroma
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 
-from langchain.document_loaders import DirectoryLoader
+from langchain.document_loaders import DirectoryLoader, UnstructuredAPIFileLoader
 from langchain.document_loaders import UnstructuredFileLoader
 
 
@@ -456,6 +456,7 @@ def content_summary(split_documents):
 def pdf_summary(file_name):
     print("file_name: "+file_name)
     loader = UnstructuredFileLoader(file_name)
+
     document = loader.load()
     text_splitter = RecursiveCharacterTextSplitter(
         chunk_size=1000,
@@ -860,42 +861,94 @@ def gradio_run():
                     ], label="訊息範例", inputs=tmp_msg)
 
         with gr.Tab("法務AI幫手"):
+            legal_path = "./data/"
+            quotation_file = "legal_quotation_prompt.txt"
+            contract_file = "legal_contract_prompt.txt"
+
+            def load_prompt_from_file(typeString):
+                if typeString == "保密合約":
+                    _path_string = legal_path + contract_file
+                else:
+                    _path_string = legal_path + quotation_file
+                f = open(_path_string, 'r', encoding="utf-8")
+                return_string= f.read()
+                f.close()
+                return return_string
+            def save_func(typeString, prompt_string):
+                if typeString == "保密合約":
+                    _path_string = legal_path + contract_file
+                else:
+                    _path_string = legal_path + quotation_file
+                f = open(_path_string, "w", encoding="utf-8")
+                f.write(prompt_string)
+                f.close()
+            def restore_func(typeString):
+                if typeString == "保密合約":
+                    content_string = default_legal_contract_prompt
+                else:
+                    content_string = default_legal_quotation_prompt
+                save_func(typeString, content_string)
+                return content_string
             def change_prompt(inputString):
                 global prompt_string
                 prompt_string = inputString
                 return inputString
-
+            gr.Markdown("""
+                ### 面版說明:
+                操作介面全部都在左側, 右側是摘要內容.
+                ### 操作步驟                
+                1. 選擇摘要的類型: 選 `保密合約` 或 `報價單`
+                2. 微調prompt內容: 直接點選 `prompt對話框` 修改文字內容
+                3. 上傳檔案: 支援PDF/doc/docx 等格式   
+                                 
+            """)
+            gr.Markdown("""
+            ---
+            """)
             with gr.Row():
                 with gr.Column(scale=2):
+                    contract_type = gr.Radio(choices=["報價單","保密合約"],
+                                             label="1. 請選擇摘要類型",
+                                             info="選擇不一樣的摘要類型,會改變下方的prompt 內容",
+                                             type="value",
+                                             value="報價單",
+                                             interactive=True)
+
                     gr.Markdown("""
-                    ### 操作方式:
-                    
-                    1. 可以選擇任何合約或是報價單, 目前僅支援text/pdf/docx/csv 格式; 上傳後即可使用下方的 prompt 來做摘要
-                    2. 如果要修改prompt, 直接點選 `prompt 對話框` 修改即可; 若有覺得摘要後內容不足, 不夠清楚, 可以在prompt 裡修正或新增定義. 新增完再上傳一次檔案即可
-                    3. 再煩請覺得更好的prompt 內容給Change; 若後續有需要長時間微調prompt, 會再加上可以線上儲存 prompt 功能.
-                    4. 下方有prompt 範例, 可以直接點選就會更改prompt 內容.
-                    
+                    ---
                     """)
-                    file_name_field = gr.Textbox(max_lines=1, label="1. 上傳檔案", placeholder="目前沒有上傳檔案")
-                    upload_button = gr.UploadButton("請上傳保密合約或報價單(可接受text, pdf, docx, csv 格式)",
-                                                file_types=["text", ".pdf", ".csv", ".docx"], file_count="multiple")
+                    _firstString = load_prompt_from_file("報價單")
+                    prompt_textbox = gr.Textbox(_firstString,
+                                                lines=20,
+                                                max_lines=20,
+                                                label="2. Prompt",
+                                                interactive=True)
+                    prompt_textbox.change(change_prompt, inputs=prompt_textbox)
+                    with gr.Row():
+                        with gr.Column():
+                            saveBtn = gr.Button("保存現有Prompt")
+                        with gr.Column():
+                            restoreBtn = gr.Button("回覆預設Prompt")
+
                     gr.Markdown("""
-                    
-                    
+                    ---
                     """)
-                    prompt_textbox = gr.Textbox(default_legal_quotation_prompt, lines=20, max_lines=20, label="2. Prompt", interactive=True)
-                    prompt_textbox.change(change_prompt, inputs=prompt_textbox)
+
+                    file_name_field = gr.Textbox(max_lines=1, label="3. 上傳檔案", placeholder="目前沒有上傳檔案")
+
+                    #event
+                    saveBtn.click(save_func, inputs=[contract_type, prompt_textbox],)
+                    restoreBtn.click(restore_func, inputs=contract_type, outputs=prompt_textbox)
+                    upload_button = gr.UploadButton("請上傳保密合約或報價單(可接受text, pdf, docx, csv 格式)",
+                                                file_types=["text", ".pdf", ".csv", ".docx", ".doc"], file_count="multiple")
+                    contract_type.change(fn=load_prompt_from_file, inputs=contract_type, outputs=prompt_textbox)
+
                 with gr.Column(scale=3):
                     summary_text = gr.Textbox()
-                    summary_text.label = "3. AI 摘要:"
+                    summary_text.label = "AI 摘要:"
                     summary_text.change = False
-                    summary_text.lines = 40
-                    summary_text.max_lines = 40
-                    pass
-
-
-            gr.Examples([default_legal_contract_prompt, default_legal_quotation_prompt], label="Prompt 範例",
-                        inputs=prompt_textbox, outputs=prompt_textbox, fn=change_prompt)
+                    summary_text.lines = 38
+                    summary_text.max_lines = 38
 
             upload_button.upload(upload_large_file, upload_button, file_name_field).\
                 then(change_prompt,inputs=prompt_textbox).\

data/legal_contract_prompt.txt

@@ -0,0 +1,20 @@
+
+你是一位超級助理, 十分擅長從大量文字中擷取摘要.
+以下用 ''' 包含的是保密合約的內容，幫我生成一份2,000個中文字以內保密合約摘要，摘要需要包含以下項目：
+1.背景: 介紹對方公司的背景、為什麼要跟該公司簽訂保密合約
+2.目的: 要與對方交換什麼資料, 資料內容與範圍
+3.合約期間:保密合約的時間範圍
+4.提前解約條款: 發生什麼樣的條件就會要提前解約
+5.保密期間: 保密的時間範圍
+6.管轄法院: 如有爭端,雙方同意的管轄法院是哪個法院
+
+AI 風險評估: 希望AI 可以評估該資料交換是否有高風險的疑慮; 評估準測:
+高風險: 涉及到營業秘密的內容
+中風險: 沒有營業秘密, 但有涉及敏感資料(足以辨識個人的訊息)
+低風險: 僅涉及作業面向的訊息
+
+保密合約:
+'''
+{text}
+'''
+

data/legal_quotation_prompt.txt

@@ -0,0 +1,23 @@
+
+你是一位超級助理, 十分擅長從大量文字中擷取摘要.
+以下用 ''' 包含的是報價單的內容，幫我生成一份2,000個中文字以內報價單摘要，摘要需要包含以下項目：
+
+1. 標的名稱: 報價單中所列出的產品或服務的名稱。
+2. 價格: 報價單中所列出的每個產品或服務的價格, 一定要有正確的幣別與金額數字.
+3. 付款內容: 報價單中所列出的付款方式和相關內容, 包括訂金, 交貨款和保留款的金額和支付方式; 除了各款項的交付百分比, 也需要有正確的金額與幣別.
+4. 交貨時間: 報價單中所列出的產品或服務的交付的日期或時間範圍。
+5. 保固(英文為Warranty): 請摘要報價單中所有關於保固內容.
+6. 維修費用：報價單中所列出的產品或服務的維修費用或相關條款, 有任何維修的金額請一定要列出.
+7. 貿易條件(Trade Term)
+8. 其他注意事項：報價單中所列出的其他重要事項或注意事項。
+
+請根據報價單的內容, 生成一份清晰明確的摘要, 條列式地把摘要列出, 確保所有項目都被包含在內. 如果內容超過三句話, 請以子項目的方式逐一列舉出來.
+
+請注意，生成的摘要應該是簡潔且易於理解的, 要詳細條列出內容, 不可產生 "依其他文件說明" 等說明方式.
+在報價單裡沒有找到符合的資訊, 你被允許回答 "無相關資料".
+
+報價單內容:
+
+'''
+{text}
+'''

requirements.txt

@@ -9,8 +9,7 @@ chromadb
 google-search-results
 tabulate
 pydantic==1.10.8
-unstructured
 hnswlib
 gradio_client==0.2.7
 tiktoken
-unstructured[docx]
+unstructured[all-docs]