Update app.py

app.py

@@ -2,8 +2,6 @@ import openai
 import gradio as gr
 import fitz  # PyMuPDF
 from openai import OpenAI
-import os
-import tempfile
 import traceback
 
 # 全域變數
@@ -61,7 +59,7 @@ def extract_pdf_text(file_path):
         text = ""
         for page_num, page in enumerate(doc):
             page_text = page.get_text()
-            if page_text.strip():  # 只添加非空白頁面
+            if page_text.strip():
                 text += f"\n--- 第 {page_num + 1} 頁 ---\n"
                 text += page_text
         doc.close()
@@ -80,20 +78,17 @@ def generate_summary(pdf_file):
         return "❌ 請先上傳 PDF 文件"
     
     try:
-        # 從 PDF 提取文字
         pdf_text = extract_pdf_text(pdf_file.name)
         
         if not pdf_text.strip():
             return "⚠️ 無法解析 PDF 文字，可能為純圖片 PDF 或空白文件。"
         
-        # 檢查文字長度，必要時截斷
-        max_chars = 8000  # 為系統提示留出空間
+        max_chars = 8000
         if len(pdf_text) > max_chars:
             pdf_text_truncated = pdf_text[:max_chars] + "\n\n[文本已截斷，僅顯示前 8000 字符]"
         else:
             pdf_text_truncated = pdf_text
         
-        # 生成摘要
         response = client.chat.completions.create(
             model=selected_model,
             messages=[
@@ -133,7 +128,6 @@ def ask_question(user_question):
         return "❌ 請輸入問題"
     
     try:
-        # 使用摘要和原始文本來提供更好的上下文
         context = f"PDF 摘要:\n{summary_text}\n\n原始內容（部分）:\n{pdf_text[:2000]}"
         
         response = client.chat.completions.create(
@@ -172,633 +166,127 @@ def clear_all():
     pdf_text = ""
     return "", "", ""
 
-# 創建 Gradio 介面
-with gr.Blocks(
-    theme=gr.themes.Soft(), 
-    title="PDF 摘要助手",
-    css="""
-    /* 修復頁面大小問題 */
-    .gradio-container {
-        max-width: none !important;
-        width: 100% !important;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-        min-height: 100vh !important;
-        padding: 0 !important;
-        margin: 0 !important;
-    }
+# 創建最簡單的 Gradio 介面
+with gr.Blocks(title="PDF 摘要助手") as demo:
+    gr.Markdown("# 📄 PDF 摘要 & 問答助手")
     
-    /* 主要內容區域 - 修復大小問題 */
-    .main-content {
-        background: rgba(255, 255, 255, 0.95) !important;
-        border-radius: 20px !important;
-        margin: 20px auto !important;
-        padding: 30px !important;
-        box-shadow: 0 20px 40px rgba(0, 0, 0, 0.1) !important;
-        backdrop-filter: blur(10px) !important;
-        max-width: 1400px !important;
-        width: calc(100% - 40px) !important;
-    }
-    
-    /* 標題樣式 */
-    .main-header {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-        -webkit-background-clip: text !important;
-        -webkit-text-fill-color: transparent !important;
-        background-clip: text !important;
-        text-align: center !important;
-        font-size: 2.5em !important;
-        font-weight: bold !important;
-        margin-bottom: 20px !important;
-    }
-    
-    /* 分頁導航 - 修復點擊問題 */
-    .gradio-tabs {
-        border-radius: 15px !important;
-        overflow: hidden !important;
-    }
-    
-    .gradio-tabitem {
-        padding: 25px !important;
-        background: white !important;
-        border-radius: 0 0 15px 15px !important;
-    }
-    
-    /* 輸入框樣式 - 修復交互問題 */
-    input[type="text"], 
-    input[type="password"], 
-    textarea {
-        border: 2px solid #e0e6ff !important;
-        border-radius: 12px !important;
-        padding: 15px !important;
-        font-size: 16px !important;
-        transition: all 0.3s ease !important;
-        width: 100% !important;
-        box-sizing: border-box !important;
-    }
-    
-    input[type="text"]:focus, 
-    input[type="password"]:focus, 
-    textarea:focus {
-        border-color: #667eea !important;
-        box-shadow: 0 0 20px rgba(102, 126, 234, 0.3) !important;
-        outline: none !important;
-    }
-    
-    /* 按鈕樣式 - 強力修復所有按鈕 */
-    button, 
-    .gr-button,
-    input[type="submit"],
-    input[type="button"],
-    #summary-button,
-    #clear-button,
-    #question-button,
-    #api-key-button,
-    #model-button,
-    .action-button,
-    [data-testid*="button"] {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-        border: none !important;
-        border-radius: 12px !important;
-        color: white !important;
-        font-weight: 600 !important;
-        padding: 15px 30px !important;
-        font-size: 16px !important;
-        transition: all 0.3s ease !important;
-        box-shadow: 0 5px 15px rgba(102, 126, 234, 0.4) !important;
-        cursor: pointer !important;
-        pointer-events: auto !important;
-        z-index: 9999 !important;
-        position: relative !important;
-        display: inline-block !important;
-        min-height: 44px !important;
-        line-height: normal !important;
-        width: auto !important;
-        min-width: 120px !important;
-        text-align: center !important;
-        user-select: none !important;
-        margin: 10px 5px !important;
-    }
-    
-    button:hover, 
-    .gr-button:hover,
-    input[type="submit"]:hover,
-    input[type="button"]:hover,
-    #summary-button:hover,
-    #clear-button:hover,
-    #question-button:hover,
-    #api-key-button:hover,
-    #model-button:hover,
-    .action-button:hover,
-    [data-testid*="button"]:hover {
-        transform: translateY(-2px) !important;
-        box-shadow: 0 8px 25px rgba(102, 126, 234, 0.6) !important;
-        background: linear-gradient(135deg, #5a6fd8 0%, #6a4190 100%) !important;
-    }
-    
-    button:active, 
-    .gr-button:active,
-    input[type="submit"]:active,
-    input[type="button"]:active,
-    #summary-button:active,
-    #clear-button:active,
-    #question-button:active,
-    #api-key-button:active,
-    #model-button:active,
-    .action-button:active,
-    [data-testid*="button"]:active {
-        transform: translateY(0px) !important;
-        box-shadow: 0 3px 10px rgba(102, 126, 234, 0.4) !important;
-    }
-    
-    /* 次要按鈕樣式 */
-    button[data-testid*="secondary"],
-    .gr-button.secondary,
-    button.secondary,
-    #clear-button,
-    #model-button,
-    .secondary-btn,
-    button[variant="secondary"] {
-        background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%) !important;
-    }
-    
-    button[data-testid*="secondary"]:hover,
-    .gr-button.secondary:hover,
-    button.secondary:hover,
-    #clear-button:hover,
-    #model-button:hover,
-    .secondary-btn:hover,
-    button[variant="secondary"]:hover {
-        background: linear-gradient(135deg, #e081e9 0%, #e3455a 100%) !important;
-    }
-    
-    /* 修復 Gradio 特定的按鈕容器 */
-    .gr-form,
-    .gr-form > div,
-    .gr-button-group,
-    div[data-testid="button"],
-    div[data-testid*="button"] {
-        pointer-events: auto !important;
-        z-index: 9998 !important;
-    }
-    
-    /* 確保按鈕內的文字和圖標不會阻擋點擊 */
-    button span,
-    button svg,
-    button i,
-    .gr-button span,
-    .gr-button svg,
-    .gr-button i,
-    .action-button span,
-    .action-button svg,
-    .action-button i {
-        pointer-events: none !important;
-        user-select: none !important;
-    }
-    
-    /* 文件上傳區域 */
-    .file-upload-area {
-        border: 3px dashed #667eea !important;
-        border-radius: 15px !important;
-        background: rgba(102, 126, 234, 0.05) !important;
-        padding: 40px !important;
-        text-align: center !important;
-        transition: all 0.3s ease !important;
-        min-height: 120px !important;
-    }
-    
-    .file-upload-area:hover {
-        background: rgba(102, 126, 234, 0.1) !important;
-        border-color: #764ba2 !important;
-    }
-    
-    /* 單選按鈕容器 */
-    .radio-group {
-        background: rgba(102, 126, 234, 0.05) !important;
-        border-radius: 12px !important;
-        padding: 20px !important;
-        margin: 10px 0 !important;
-    }
-    
-    /* 輸出文本區域 */
-    .output-text {
-        background: #f8f9ff !important;
-        border: 1px solid #e0e6ff !important;
-        border-radius: 12px !important;
-        padding: 20px !important;
-        min-height: 200px !important;
-    }
-    
-    /* 隱藏 Gradio logo 和 footer */
-    footer,
-    .gradio-container footer,
-    div[class*="footer"],
-    div[class*="Footer"],
-    .gr-footer {
-        display: none !important;
-    }
-    
-    /* 修復響應式問題 */
-    .gr-row {
-        display: flex !important;
-        gap: 20px !important;
-        width: 100% !important;
-    }
-    
-    .gr-column {
-        flex: 1 !important;
-        min-width: 0 !important;
-    }
-    
-    /* 確保所有交互元素正常工作 */
-    * {
-        pointer-events: auto !important;
-    }
-    
-    /* 特殊修復：覆蓋可能的 Gradio 樣式衝突 */
-    .gradio-container * {
-        pointer-events: auto !important;
-    }
-    
-    /* 修復單選按鈕 */
-    input[type="radio"] {
-        pointer-events: auto !important;
-        cursor: pointer !important;
-        z-index: 1000 !important;
-        position: relative !important;
-    }
-    
-    /* 修復文件上傳 */
-    input[type="file"] {
-        pointer-events: auto !important;
-        cursor: pointer !important;
-        z-index: 1000 !important;
-    }
-    
-    /* 添加 JavaScript 來確保按鈕響應 */
-    </style>
-    <script>
-    document.addEventListener('DOMContentLoaded', function() {
-        console.log('開始修復按鈕...');
+    with gr.Tab("🔧 設定"):
+        gr.Markdown("## API Key 設定")
         
-        // 強力修復所有按鈕
-        function fixAllButtons() {
-            // 所有可能的按鈕選擇器
-            const selectors = [
-                'button',
-                'input[type="submit"]', 
-                'input[type="button"]',
-                '.gr-button',
-                '[data-testid*="button"]',
-                '#summary-button',
-                '#clear-button', 
-                '#question-button',
-                '#api-key-button',
-                '#model-button',
-                '.action-button',
-                '.primary-btn',
-                '.secondary-btn'
-            ];
-            
-            let buttonCount = 0;
-            selectors.forEach(selector => {
-                const elements = document.querySelectorAll(selector);
-                elements.forEach(btn => {
-                    if (btn) {
-                        // 強制設定樣式
-                        btn.style.pointerEvents = 'auto';
-                        btn.style.cursor = 'pointer';
-                        btn.style.zIndex = '9999';
-                        btn.style.position = 'relative';
-                        btn.style.display = 'inline-block';
-                        btn.style.minHeight = '44px';
-                        btn.style.minWidth = '120px';
-                        btn.style.padding = '15px 30px';
-                        btn.style.borderRadius = '12px';
-                        btn.style.border = 'none';
-                        btn.style.fontWeight = '600';
-                        btn.style.fontSize = '16px';
-                        btn.style.color = 'white';
-                        btn.style.userSelect = 'none';
-                        btn.style.margin = '10px 5px';
-                        btn.style.textAlign = 'center';
-                        
-                        // 設定背景色
-                        if (btn.id === 'clear-button' || 
-                            btn.id === 'model-button' || 
-                            btn.classList.contains('secondary-btn')) {
-                            btn.style.background = 'linear-gradient(135deg, #f093fb 0%, #f5576c 100%)';
-                        } else {
-                            btn.style.background = 'linear-gradient(135deg, #667eea 0%, #764ba2 100%)';
-                        }
-                        
-                        // 移除可能阻止點擊的屬性
-                        btn.removeAttribute('disabled');
-                        
-                        // 確保子元素不會阻止點擊
-                        const children = btn.querySelectorAll('*');
-                        children.forEach(child => {
-                            child.style.pointerEvents = 'none';
-                            child.style.userSelect = 'none';
-                        });
-                        
-                        buttonCount++;
-                    }
-                });
-            });
-            
-            console.log(`修復了 ${buttonCount} 個按鈕`);
-        }
+        api_key_input = gr.Textbox(
+            label="輸入 OpenAI API Key", 
+            type="password",
+            placeholder="請輸入您的 OpenAI API Key (sk-...)"
+        )
         
-        // 立即修復
-        fixAllButtons();
+        with gr.Row():
+            api_key_btn = gr.Button("確認設定 API Key", variant="primary")
         
-        // 每秒檢查一次
-        setInterval(fixAllButtons, 1000);
+        api_key_status = gr.Textbox(
+            label="API 狀態", 
+            interactive=False,
+            value="等待設定 API Key..."
+        )
         
-        // 監聽 DOM 變化
-        const observer = new MutationObserver(function(mutations) {
-            let shouldFix = false;
-            mutations.forEach(function(mutation) {
-                if (mutation.type === 'childList') {
-                    mutation.addedNodes.forEach(function(node) {
-                        if (node.nodeType === 1) { // 元素節點
-                            if (node.tagName === 'BUTTON' || 
-                                node.querySelector && node.querySelector('button')) {
-                                shouldFix = true;
-                            }
-                        }
-                    });
-                }
-            });
-            
-            if (shouldFix) {
-                setTimeout(fixAllButtons, 100);
-            }
-        });
+        gr.Markdown("## 模型選擇")
         
-        observer.observe(document.body, { 
-            childList: true, 
-            subtree: true,
-            attributes: true,
-            attributeFilter: ['style', 'class']
-        });
+        model_choice = gr.Radio(
+            choices=["gpt-4", "gpt-4.1", "gpt-4.5"], 
+            label="選擇 AI 模型",
+            value="gpt-4"
+        )
         
-        // 添加全域點擊事件監聽
-        document.addEventListener('click', function(e) {
-            console.log('點擊事件:', e.target);
-        }, true);
-    });
-    </script>
-    <style>
-    
-    /* 響應式設計 */
-    @media (max-width: 768px) {
-        .main-content {
-            margin: 10px !important;
-            padding: 20px !important;
-            width: calc(100% - 20px) !important;
-        }
+        with gr.Row():
+            model_btn = gr.Button("確認選擇模型", variant="secondary")
         
-        .main-header {
-            font-size: 1.8em !important;
-        }
+        model_status = gr.Textbox(
+            label="當前模型", 
+            interactive=False,
+            value="已選擇：gpt-4"
+        )
+    
+    with gr.Tab("📄 PDF 處理"):
+        gr.Markdown("## 上傳 PDF 文件")
         
-        .gr-row {
-            flex-direction: column !important;
-        }
+        pdf_upload = gr.File(
+            label="選擇 PDF 文件",
+            file_types=[".pdf"]
+        )
         
-        button {
-            width: 100% !important;
-            margin: 5px 0 !important;
-        }
-    }
-    """
-) as demo:
-    with gr.Column(elem_classes="main-content"):
-        gr.HTML("""
-        <div class="main-header">📄 PDF 摘要 & 問答助手</div>
+        with gr.Row():
+            summary_btn = gr.Button("生成摘要", variant="primary")
+            clear_btn = gr.Button("清除資料", variant="secondary")
         
-        <div style="text-align: center; margin-bottom: 30px; padding: 25px; background: linear-gradient(135deg, rgba(102, 126, 234, 0.1) 0%, rgba(118, 75, 162, 0.1) 100%); border-radius: 15px; border-left: 5px solid #667eea;">
-            <h3 style="color: #667eea; margin-bottom: 15px;">🚀 歡迎使用 PDF 智能分析工具！</h3>
-            <div style="display: flex; justify-content: space-around; flex-wrap: wrap; margin: 20px 0;">
-                <div style="margin: 10px; padding: 15px; background: white; border-radius: 10px; box-shadow: 0 3px 10px rgba(0,0,0,0.1); min-width: 200px;">
-                    <div style="font-size: 24px; margin-bottom: 10px;">📋</div>
-                    <strong>智能摘要生成</strong><br>
-                    <span style="color: #666;">自動分析 PDF 內容並生成結構化摘要</span>
-                </div>
-                <div style="margin: 10px; padding: 15px; background: white; border-radius: 10px; box-shadow: 0 3px 10px rgba(0,0,0,0.1); min-width: 200px;">
-                    <div style="font-size: 24px; margin-bottom: 10px;">🤖</div>
-                    <strong>AI 問答系統</strong><br>
-                    <span style="color: #666;">基於文檔內容回答您的問題</span>
-                </div>
-                <div style="margin: 10px; padding: 15px; background: white; border-radius: 10px; box-shadow: 0 3px 10px rgba(0,0,0,0.1); min-width: 200px;">
-                    <div style="font-size: 24px; margin-bottom: 10px;">💡</div>
-                    <strong>快速理解</strong><br>
-                    <span style="color: #666;">快速掌握長篇文檔的核心內容</span>
-                </div>
-            </div>
-            <div style="background: rgba(255, 193, 7, 0.1); padding: 15px; border-radius: 10px; border-left: 4px solid #ffc107; margin-top: 20px;">
-                <strong style="color: #e65100;">⚠️ 重要提醒：</strong> 使用前請先在「🔧 設定」頁面輸入您的 OpenAI API Key
-            </div>
-        </div>
-        """)
-    
-        with gr.Tab("🔧 設定", elem_classes="settings-tab"):
-            gr.Markdown("### ⚙️ 基本設定")
-            with gr.Row():
-                with gr.Column():
-                    api_key_input = gr.Textbox(
-                        label="🔑 輸入 OpenAI API Key", 
-                        type="password",
-                        placeholder="請輸入您的 OpenAI API Key (sk-...)",
-                        elem_classes="api-input"
-                    )
-                    # 添加確認按鈕
-                    api_key_btn = gr.Button(
-                        "✅ 確認設定 API Key",
-                        variant="primary",
-                        elem_id="api-key-button",
-                        elem_classes="action-button primary-btn"
-                    )
-                    api_key_status = gr.Textbox(
-                        label="📊 API 連接狀態", 
-                        interactive=False,
-                        value="🔄 等待設定 API Key...",
-                        elem_classes="status-display"
-                    )
-                
-                with gr.Column():
-                    model_choice = gr.Radio(
-                        choices=["gpt-4", "gpt-4.1", "gpt-4.5"], 
-                        label="🤖 選擇 AI 模型",
-                        value="gpt-4",
-                        interactive=True,
-                        elem_classes="model-selector"
-                    )
-                    # 添加模型確認按鈕
-                    model_btn = gr.Button(
-                        "🎯 確認選擇模型",
-                        variant="secondary", 
-                        elem_id="model-button",
-                        elem_classes="action-button secondary-btn"
-                    )
-                    model_status = gr.Textbox(
-                        label="🎯 當前模型", 
-                        interactive=False,
-                        value="✅ 已選擇：gpt-4",
-                        elem_classes="status-display"
-                    )
-            
-            # 添加使用說明和測試區域
-            gr.HTML("""
-            <div style='margin: 20px 0; padding: 15px; background: rgba(102, 126, 234, 0.1); border-radius: 10px;'>
-                <strong>📋 使用步驟：</strong>
-                <ol style='margin: 10px 0; padding-left: 20px;'>
-                    <li>輸入您的 OpenAI API Key 並點擊「確認設定」</li>
-                    <li>選擇 AI 模型並點擊「確認選擇」</li>
-                    <li>前往「PDF 處理」頁面上傳文件開始使用！</li>
-                </ol>
-                <div style='margin-top: 15px; padding: 10px; background: rgba(255, 193, 7, 0.1); border-radius: 8px;'>
-                    <strong>🔧 調試信息：</strong>
-                    <br>• 如果按鈕沒反應，請檢查瀏覽器控制台
-                    <br>• API Key 應該以 'sk-' 開頭
-                    <br>• 模型選擇為單選，一次只能選一個
-                </div>
-            </div>
-            """)
-            
-            # 添加測試按鈕
-            with gr.Row():
-                test_btn = gr.Button(
-                    "🧪 測試按鈕功能", 
-                    variant="secondary",
-                    elem_id="test-button"
-                )
-                test_output = gr.Textbox(
-                    label="測試結果", 
-                    interactive=False,
-                    placeholder="點擊測試按鈕查看是否正常工作"
-                )
-            
-            def test_function():
-                return "✅ 按鈕功能正常！如果您看到這個訊息，表示按鈕點擊事件正常工作。"
-            
-            test_btn.click(test_function, outputs=test_output)
-    
-        with gr.Tab("📄 PDF 處理", elem_classes="pdf-tab"):
-            gr.Markdown("### 📁 文件上傳與摘要生成")
-            with gr.Row():
-                with gr.Column():
-                    pdf_upload = gr.File(
-                        label="📎 選擇 PDF 文件",
-                        file_types=[".pdf"],
-                        elem_classes="file-upload"
-                    )
-                    gr.Markdown("**支援格式**：PDF 文件 (建議 < 10MB)")
-                    
-                    with gr.Row():
-                        summary_btn = gr.Button(
-                            "🔄 開始生成摘要", 
-                            variant="primary",
-                            elem_id="summary-button",
-                            elem_classes="action-button primary-btn"
-                        )
-                        clear_btn = gr.Button(
-                            "🗑️ 清除所有資料", 
-                            variant="secondary",
-                            elem_id="clear-button", 
-                            elem_classes="action-button secondary-btn"
-                        )
-                
-                with gr.Column():
-                    summary_output = gr.Textbox(
-                        label="📋 AI 生成的文檔摘要",
-                        lines=15,
-                        placeholder="📄 上傳 PDF 文件並點擊「開始生成摘要」按鈕，AI 將為您分析文檔內容...",
-                        elem_classes="summary-output"
-                    )
-    
-        with gr.Tab("❓ 智能問答", elem_classes="qa-tab"):
-            gr.Markdown("### 💬 基於文檔內容的 AI 問答")
-            with gr.Row():
-                with gr.Column():
-                    question_input = gr.Textbox(
-                        label="💭 請輸入您的問題",
-                        placeholder="例如：這份文件的主要結論是什麼？文中提到的關鍵數據有哪些？",
-                        lines=3,
-                        elem_classes="question-input"
-                    )
-                    question_btn = gr.Button(
-                        "📤 發送問題", 
-                        variant="primary",
-                        elem_id="question-button",
-                        elem_classes="action-button primary-btn"
-                    )
-                    
-                    gr.Markdown("""
-                    **💡 問題範例：**
-                    - 這份文件討論的主要議題是什麼？
-                    - 文中有哪些重要的統計數據？
-                    - 作者的主要觀點和結論是什麼？
-                    - 文件中提到的建議有哪些？
-                    """)
-                
-                with gr.Column():
-                    answer_output = gr.Textbox(
-                        label="🤖 AI 智能回答",
-                        lines=12,
-                        placeholder="🤖 請先上傳並生成 PDF 摘要��然後在左側輸入問題，AI 將基於文檔內容為您提供詳細回答...",
-                        elem_classes="answer-output"
-                    )
-    
-    # 事件處理器 - 確保按鈕點擊有反應
-    # API Key 設定事件（雙重綁定確保有效）
-    def handle_api_key_click(api_key_value):
-        print(f"處理 API Key: {api_key_value[:10]}..." if api_key_value else "空值")
-        return set_api_key(api_key_value)
+        summary_output = gr.Textbox(
+            label="PDF 摘要",
+            lines=15,
+            placeholder="上傳 PDF 文件並點擊「生成摘要」按鈕"
+        )
     
-    def handle_model_click(model_value):
-        print(f"處理模型選擇: {model_value}")
-        return set_model(model_value)
+    with gr.Tab("❓ 問答"):
+        gr.Markdown("## AI 問答")
+        
+        question_input = gr.Textbox(
+            label="請輸入您的問題",
+            placeholder="例如：這份文件的主要結論是什麼？",
+            lines=3
+        )
+        
+        with gr.Row():
+            question_btn = gr.Button("發送問題", variant="primary")
+        
+        answer_output = gr.Textbox(
+            label="AI 回答",
+            lines=10,
+            placeholder="AI 回答將顯示在這裡"
+        )
     
-    # 綁定事件
+    # 事件綁定 - 使用最簡單的方式
     api_key_btn.click(
-        fn=handle_api_key_click, 
-        inputs=[api_key_input], 
-        outputs=[api_key_status]
-    )
-    api_key_input.submit(
-        fn=handle_api_key_click, 
-        inputs=[api_key_input], 
-        outputs=[api_key_status]
+        fn=set_api_key, 
+        inputs=api_key_input, 
+        outputs=api_key_status
     )
     
-    # 模型選擇事件
     model_btn.click(
-        fn=handle_model_click, 
-        inputs=[model_choice], 
-        outputs=[model_status]
+        fn=set_model, 
+        inputs=model_choice, 
+        outputs=model_status
     )
-    model_choice.change(
-        fn=handle_model_click, 
-        inputs=[model_choice], 
-        outputs=[model_status]
+    
+    summary_btn.click(
+        fn=generate_summary, 
+        inputs=pdf_upload, 
+        outputs=summary_output
     )
     
-    # PDF 處理事件
-    summary_btn.click(generate_summary, inputs=pdf_upload, outputs=summary_output)
+    question_btn.click(
+        fn=ask_question, 
+        inputs=question_input, 
+        outputs=answer_output
+    )
+    
+    clear_btn.click(
+        fn=clear_all, 
+        outputs=[summary_output, question_input, answer_output]
+    )
     
-    # 問答事件
-    question_btn.click(ask_question, inputs=question_input, outputs=answer_output)
-    question_input.submit(ask_question, inputs=question_input, outputs=answer_output)
+    # 也支援 Enter 鍵
+    api_key_input.submit(
+        fn=set_api_key, 
+        inputs=api_key_input, 
+        outputs=api_key_status
+    )
     
-    # 清除功能
-    clear_btn.click(clear_all, outputs=[summary_output, question_input, answer_output])
+    question_input.submit(
+        fn=ask_question, 
+        inputs=question_input, 
+        outputs=answer_output
+    )
 
 if __name__ == "__main__":
-    # Hugging Face Spaces 部署設定
     demo.launch(
-        share=False,  # Hugging Face Spaces 不需要 share
+        share=False,
         show_api=False,
         show_error=True
     )