Update app.py

app.py

@@ -8,7 +8,6 @@ from langchain.vectorstores import Chroma
 from langchain_community.embeddings import OpenAIEmbeddings
 from langchain_community.chat_models import ChatOpenAI
 import shutil  # 用於文件複製
-import logging
 
 # 獲取 OpenAI API 密鑰（初始不使用固定密鑰）
 api_key_env = os.getenv("OPENAI_API_KEY")
@@ -116,7 +115,7 @@ def process_files(files, state):
         try:
             api_key = state.get('api_key', None)
             if not api_key:
-                return "請先在「設定 API 密鑰」標籤中輸入並保存您的 OpenAI API 密鑰。", state
+                return "請先輸入並保存您的 OpenAI API 密鑰。", state
 
             saved_file_paths = []
             for idx, file_data in enumerate(files):
@@ -139,7 +138,7 @@ def chat_interface(user_message, chat_history, state):
     if not vectordb:
         return chat_history, state, "請先上傳 PDF 文件以進行處理。"
     if not api_key:
-        return chat_history, state, "請先在「設定 API 密鑰」標籤中輸入並保存您的 OpenAI API 密鑰。"
+        return chat_history, state, "請先輸入並保存您的 OpenAI API 密鑰。"
 
     updated_history = handle_query(user_message, chat_history, vectordb, api_key)
     return updated_history, state, ""
@@ -150,52 +149,36 @@ with gr.Blocks() as demo:
 
     state = gr.State({"vectordb": None, "api_key": None})
 
-    with gr.Tab("設定 API 密鑰"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                api_key_input = gr.Textbox(
-                    label="輸入您的 OpenAI API 密鑰",
-                    placeholder="sk-...",
-                    type="password",
-                    interactive=True
-                )
-                save_api_key_btn = gr.Button("保存 API 密鑰")
-                api_key_status = gr.Textbox(label="狀態", interactive=False)
-
-    with gr.Tab("上傳勞動法 PDF 文件"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                upload = gr.File(
-                    file_count="multiple",
-                    file_types=[".pdf"],
-                    label="上傳勞動法 PDF 文件",
-                    interactive=True,
-                    type="binary"
-                )
-                upload_btn = gr.Button("上傳並處理")
-                upload_status = gr.Textbox(label="上傳狀態", interactive=False)
-
-    with gr.Tab("智能諮詢"):
-        chatbot = gr.Chatbot()
-
-        with gr.Row():
-            with gr.Column(scale=0.85):
-                txt = gr.Textbox(show_label=False, placeholder="請輸入您的法律問題...")
-            with gr.Column(scale=0.15, min_width=0):
-                submit_btn = gr.Button("提問")
-
-        submit_btn.click(
-            chat_interface,
-            inputs=[txt, chatbot, state],
-            outputs=[chatbot, state, txt]
-        )
+    # API 密鑰輸入框
+    api_key_input = gr.Textbox(
+        label="輸入您的 OpenAI API 密鑰",
+        placeholder="sk-...",
+        type="password",
+        interactive=True
+    )
+    save_api_key_btn = gr.Button("保存 API 密鑰")
+    api_key_status = gr.Textbox(label="狀態", interactive=False)
+
+    # 上傳 PDF 文件
+    gr.Markdown("請上傳勞動法相關文檔，讓我協助解決您的職場問題！🤖")
+    upload = gr.File(
+        file_count="multiple",
+        file_types=[".pdf"],
+        label="上傳勞動法 PDF 文件",
+        interactive=True,
+        type="binary"
+    )
+    upload_btn = gr.Button("上傳並處理")
+    upload_status = gr.Textbox(label="上傳狀態", interactive=False)
 
-        txt.submit(
-            chat_interface,
-            inputs=[txt, chatbot, state],
-            outputs=[chatbot, state, txt]
-        )
+    # 智能諮詢
+    gr.Markdown("### 勞動法小幫手")
+    chatbot = gr.Chatbot()
+
+    txt = gr.Textbox(show_label=False, placeholder="請輸入您的法律問題...")
+    submit_btn = gr.Button("提問")
 
+    # 綁定事件
     save_api_key_btn.click(
         save_api_key,
         inputs=[api_key_input, state],
@@ -208,5 +191,17 @@ with gr.Blocks() as demo:
         outputs=[upload_status, state]
     )
 
+    submit_btn.click(
+        chat_interface,
+        inputs=[txt, chatbot, state],
+        outputs=[chatbot, state, txt]
+    )
+
+    txt.submit(
+        chat_interface,
+        inputs=[txt, chatbot, state],
+        outputs=[chatbot, state, txt]
+    )
+
 # 啟動 Gradio 應用
 demo.launch()