Upload app.py

app.py

@@ -15,6 +15,7 @@ from langchain.prompts import PromptTemplate
 from langchain.schema import Document
 import gradio as gr
 import re
+import time
 
 def initialize_llm(api_key):
     return ChatGroq(
@@ -55,9 +56,9 @@ sources = [
     "HouseTaxAct1130103.txt",
     "VehicleLicenseTaxAct1101230.txt",
     "TaxCollectionAct1101217.txt",
-    "AmusementTaxAct960523.txt",
     "StampTaxAct910515.txt",
-    "DeedTaxAct990505.txt"
+    "DeedTaxAct990505.txt",
+    "AmusementTaxAct960523.txt"
 ]
 
 documents = load_documents(sources)
@@ -65,7 +66,7 @@ print(f"\n成功載入 {len(documents)} 個網址或檔案")
 
 text_splitter = RecursiveCharacterTextSplitter(
     chunk_size=512,
-    chunk_overlap=50,
+    chunk_overlap=52,
     length_function=len,
     is_separator_regex=False,
     separators=["\n\n\n","\n\n", "\n", "。"]
@@ -146,6 +147,7 @@ def generate_insight_questions(query, api_key):
 
 def answer_question(query, api_key):
     try:
+        gr.Info("檢索地方稅知識庫中，請稍待片刻......")
         llm = initialize_llm(api_key)
         chain = create_chain(llm)
         result = chain.invoke({"query": query})
@@ -183,55 +185,66 @@ def convert_punctuation(text):
     return text.replace('?', '？').replace(',', '，').replace('!', '！').replace(' ', '　')
 
 def handle_interaction(query, api_key, state):
+    gr.Info("開始處理問題，請稍待片刻......")
+    start_time = time.time()
     if state is None:
         state = {"history": []}
     if not api_key:
-        api_key = os.getenv("Llama70B_Key")
+        api_key = os.getenv("YOUR_API_KEY")
     query = convert_punctuation(query)
     answer, insight_questions = answer_multiple_questions(query, api_key)
     state["history"].append((query, answer))
     while len(insight_questions) < 3:
         insight_questions.append("提供更多地方稅資訊")
+    end_time = time.time()
+    gr.Info(f"Model 已完成回覆，總執行時間： {(end_time - start_time):.2f} 秒。")
     return answer, insight_questions[0], insight_questions[1], insight_questions[2], state, query
 
 custom_css = """
-body {
-    background-color: #e8f5e9;
+.query-input {
+    background-color: #B7E0FF !important;
+    padding: 15px !important;
+    border-radius: 10px !important;
+    margin: 0 !important;
 }
-#answer-box textarea, #query-input textarea {
+.query-input textarea {
     font-size: 18px !important;
     background-color: #ffffff;
-    border: 1px solid #81c784;
+    border: 1px solid #f0f8ff;
     border-radius: 8px;
 }
-.center-text {
-    text-align: center !important;
-    color: #2e7d32 !important;
-}
-.gradio-container {
-    background-color: #c8e6c9 !important;
-    border-radius: 15px !important;
-    padding: 20px !important;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1) !important;
+.answer-box {
+    background-color: #FFF5CD !important;
+    padding: 10px !important;
+    border-radius: 10px !important;
+    margin: 0 !important;
 }
-.gr-button {
-    color: white !important;
-    border: none !important;
-    border-radius: 20px !important;
-    transition: all 0.3s ease !important;
-    font-weight: bold !important;
+.answer-box textarea {
+    font-size: 18px !important;
+    background-color: #ffffff;
+    border: 1px solid #f0f8ff;
+    border-radius: 8px;
 }
-.gr-button:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2) !important;
+.center-text {
+    text-align: center !important;
+    color: #ff4081;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
+    margin-bottom: 0 !important;
 }
 #submit-btn {
+    border-radius: 10px !important;
     background-color: #ff4081 !important;
+    color: white !important;
+    font-weight: bold !important;
+    transition: all 0.3s ease !important;
+    margin: 0 !important;
 }
 #submit-btn:hover {
     background-color: #f50057 !important;
+    transform: scale(1.05);
 }
 .insight-btn {
+    border-radius: 10px !important;
     background-color: #00bcd4 !important;
 }
 .insight-btn:hover {
@@ -243,24 +256,42 @@ body {
     border-radius: 10px !important;
 }
 .api-key-input {
-    max-width: 510px !important;
+    background-color: #FFCFB3 !important;
+    padding: 15px !important;
+    border-radius: 10px !important;
+    margin: 0 !important;
+}
+.clear-button {
+    color: white !important;
+    background-color: #000000 !important;
+    padding: 5px !important;
+    border-radius: 10px !important;
+    margin: 0 !important;
+}
+.clear-button:hover {
+    background-color: #000000 !important;
+    transform: scale(1.05);
 }
 """
 
 with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as iface:
-    gr.Markdown("# 地方稅知識庫系統 - 財政部財政資訊中心", elem_classes=["center-text"])
-    gr.Markdown("※ RAG-based 系統部署：江信宗，LLM：Llama-3.1-70B，以地方稅極少知識資料示範，僅供參考，準確資訊請依據地方稅稽徵機關回覆為準。", elem_classes=["center-text"])
+    gr.Markdown("""
+    # 地方稅知識庫系統 - 財政部財政資訊中心
+    > ### **※ RAG-based 系統部署：江信宗，LLM：Llama-3.1-70B，以地方稅極少知識資料示範，僅供參考，準確資訊請依據地方稅稽徵機關回覆為準。**
+    """, elem_classes="center-text")
     with gr.Row():
-        query_input = gr.Textbox(lines=2, placeholder="請輸入您的問題（支援同時輸入多個問題，例如：問題1？問題2？）", label="輸入您的問題，系統將基於學習到的知識資料提供相關答案。", elem_id="query-input")
-        api_key_input = gr.Textbox(type="password", placeholder="請輸入您的 API Key", label="API authentication key for large language models", elem_classes="api-key-input")
-    answer_output = gr.Textbox(lines=6, label="答案：", elem_id="answer-box")
+        query_input = gr.Textbox(label="輸入您的問題，系統將基於學習到的知識資料提供相關答案。", placeholder="請輸入您的問題（支援同時輸入多個問題，例如：問題1？問題2？）", scale=3, elem_classes="query-input")
+        api_key_input = gr.Textbox(label="請輸入您的 API Key", type="password", placeholder="API authentication key", scale=1, elem_classes="api-key-input")
+    answer_output = gr.Textbox(label="答案：", max_lines=40, elem_classes="answer-box")
     with gr.Row():
         insight_q1 = gr.Button("洞察問題 1", visible=False, elem_classes=["insight-btn"])
         insight_q2 = gr.Button("洞察問題 2", visible=False, elem_classes=["insight-btn"])
         insight_q3 = gr.Button("洞察問題 3", visible=False, elem_classes=["insight-btn"])
     state = gr.State()
     current_question = gr.Textbox(lines=2, label="當前問題", visible=False)
-    submit_btn = gr.Button("傳送", elem_id="submit-btn")
+    with gr.Row():
+        submit_btn = gr.Button("傳送", variant="primary", scale=3, elem_id="submit-btn")
+        clear_button = gr.Button("清除", variant="secondary", scale=1, elem_classes="clear-button")
     def update_ui(answer, q1, q2, q3, state, current_q):
         return [
             answer,
@@ -285,9 +316,16 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as iface:
             inputs=[btn],
             outputs=[query_input]
         )
+    def clear_outputs():
+        return "", ""
+    clear_button.click(
+        fn=clear_outputs,
+        inputs=[],
+        outputs=[query_input, answer_output]
+    )
 
 if __name__ == "__main__":
     if "SPACE_ID" in os.environ:
         iface.launch()
     else:
-        iface.launch(share=True)
+        iface.launch(share=True, show_api=False)