Upload app.py

app.py

@@ -79,8 +79,6 @@ def parse_response(response):
 
 def generate(message, history, model, system_prompt, thinking_budget, api_key):
     """Generates the chatbot response."""
-    gr.Info("開始思考分析問題，請稍待片刻......")
-    start_time = time.time()
     # 獲取AI回應和思考時間
     response, thinking_time = respond(message, history, model, system_prompt, thinking_budget, api_key)
     # 如果回應是錯誤資訊，直接返回
@@ -103,8 +101,6 @@ def generate(message, history, model, system_prompt, thinking_budget, api_key):
     })
     # 新增AI的最終答案
     messages.append({"role": "assistant", "content": f"<b>{answer}</b>"})
-    end_time = time.time()
-    gr.Info(f"已成功推理回覆，總執行時間： {(end_time - start_time):.2f} 秒。")
     # 返回更新後的歷史記錄和空字串（用於清空輸入框）
     return history + messages, ""
 
@@ -150,10 +146,10 @@ custom_css = """
     text-align: center !important;
     color: #ff4081;
     text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
-    margin-bottom: -20px !important;
+    margin-bottom: -10px !important;
 }
 .input-background {
-    background-color: #e3f2fd !important;
+    background-color: #FFF4B5 !important;
     padding: 15px !important;
     border-radius: 10px !important;
     margin: 0 !important;
@@ -167,11 +163,6 @@ custom_css = """
 .custom-button {
     border-radius: 10px !important;
 }
-.natural-bg {
-    background-color: #e8f5e9 !important;
-    padding: 15px !important;
-    border-radius: 10px !important;
-}
 .pink-bg {
     background-color: #ff4081 !important;
     border-radius: 10px !important;
@@ -199,7 +190,7 @@ custom_css = """
 with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
     gr.Markdown("# Multi-Chain Reasoning using Llama-3.1-Instruct. Deployed by 江信宗", elem_classes="center-aligned")
 
-    with gr.Row(elem_classes="natural-bg"):
+    with gr.Row():
         model = gr.Dropdown(choices=MODELS, label="選擇模型", value=MODELS[0], elem_classes="input-background")
         thinking_budget = gr.Slider(minimum=1, maximum=100, value=20, step=1, label="思維規劃", info="模型所能進行的最大思考次數", elem_classes="input-background")
         api_key = gr.Textbox(label="API Key", type="password", placeholder="API authentication key for large language models", elem_classes="api-background")
@@ -218,7 +209,11 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=custom_css) as demo:
     )
     msg = gr.Textbox(label="請輸入您的問題：", placeholder="輸入完成後直接按 Enter 開始執行......", elem_classes="pink-bg", max_lines=10)
     clear_button = gr.Button("清除聊天記錄", elem_classes="custom-button")
-    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, msg])
+    clear_button.click(
+        lambda: ([], "", gr.Info("已成功清除聊天記錄，請繼續提問......")),
+        inputs=None,
+        outputs=[chatbot, msg]
+    )
     system_prompt = gr.Textbox(label="System Prompt", value=DEFAULT_SYSTEM_PROMPT, visible=False)
     msg.submit(generate, inputs=[msg, chatbot, model, system_prompt, thinking_budget, api_key], outputs=[chatbot, msg])