Update app.py

app.py

@@ -2,6 +2,8 @@ import openai
 import gradio as gr
 import fitz  # PyMuPDF
 from openai import OpenAI
+import os
+import tempfile
 import traceback
 
 # 全域變數
@@ -19,38 +21,24 @@ def set_api_key(user_api_key):
         if not api_key:
             return "❌ API Key 不能為空"
         
-        if not api_key.startswith('sk-'):
-            return "❌ API Key 格式錯誤，應該以 'sk-' 開頭"
-        
         client = OpenAI(api_key=api_key)
         
         # 測試 API Key 是否有效
-        try:
-            test_response = client.chat.completions.create(
-                model="gpt-3.5-turbo",
-                messages=[{"role": "user", "content": "Hello"}],
-                max_tokens=5
-            )
-            return "✅ API Key 已設定並驗證成功！可以開始使用了。"
-        except Exception as e:
-            if "incorrect_api_key" in str(e).lower():
-                return "❌ API Key 無效，請檢查是否正確"
-            elif "quota" in str(e).lower():
-                return "⚠️ API Key 有效，但配額不足"
-            else:
-                return f"⚠️ API Key 設定成功，但測試時出現問題：{str(e)[:100]}"
-                
+        test_response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[{"role": "user", "content": "你好"}],
+            max_tokens=5
+        )
+        
+        return "✅ API Key 已設定並驗證成功"
     except Exception as e:
-        return f"❌ 設定失敗：{str(e)}"
+        return f"❌ API Key 設定失敗: {str(e)}"
 
 def set_model(model_name):
     """設定選擇的模型"""
     global selected_model
-    if model_name:
-        selected_model = model_name
-        return f"✅ 模型已更新為：{model_name}"
-    else:
-        return "❌ 請選擇一個模型"
+    selected_model = model_name
+    return f"✅ 模型已選擇：{model_name}"
 
 def extract_pdf_text(file_path):
     """從 PDF 文件中提取文字"""
@@ -59,7 +47,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()
@@ -78,17 +66,20 @@ 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=[
@@ -128,6 +119,7 @@ def ask_question(user_question):
         return "❌ 請輸入問題"
     
     try:
+        # 使用摘要和原始文本來提供更好的上下文
         context = f"PDF 摘要:\n{summary_text}\n\n原始內容（部分）:\n{pdf_text[:2000]}"
         
         response = client.chat.completions.create(
@@ -166,127 +158,301 @@ def clear_all():
     pdf_text = ""
     return "", "", ""
 
-# 創建最簡單的 Gradio 介面
+# 創建 Gradio 介面 - 全螢幕版本
 with gr.Blocks(
+    theme=gr.themes.Soft(), 
     title="PDF 摘要助手",
     css="""
-    /* 隱藏 Gradio logo 和 footer */
-    footer { display: none !important; }
-    .gradio-container footer { display: none !important; }
-    div[class*="footer"] { display: none !important; }
-    div[class*="Footer"] { display: none !important; }
-    .gr-footer { display: none !important; }
+    /* 全螢幕樣式 */
+    .gradio-container {
+        max-width: none !important;
+        width: 100vw !important;
+        height: 100vh !important;
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        margin: 0 !important;
+        padding: 0 !important;
+    }
+    
+    /* 主要內容區域 - 全螢幕 */
+    .main-content {
+        background: rgba(255, 255, 255, 0.95) !important;
+        border-radius: 20px !important;
+        margin: 15px !important;
+        padding: 30px !important;
+        box-shadow: 0 20px 40px rgba(0, 0, 0, 0.1) !important;
+        backdrop-filter: blur(10px) !important;
+        width: calc(100vw - 30px) !important;
+        min-height: calc(100vh - 30px) !important;
+        box-sizing: border-box !important;
+    }
+    
+    /* 標題樣式 */
+    .main-header {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        -webkit-background-clip: text !important;
+        -webkit-text-fill-color: transparent !important;
+        text-align: center !important;
+        font-size: 2.8em !important;
+        font-weight: bold !important;
+        margin-bottom: 25px !important;
+    }
+    
+    /* 分頁導航 */
+    .gradio-tabs {
+        border-radius: 15px !important;
+        overflow: hidden !important;
+        width: 100% !important;
+    }
+    
+    .gradio-tabitem {
+        padding: 30px !important;
+        background: white !important;
+        border-radius: 0 0 15px 15px !important;
+        width: 100% !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 {
-        pointer-events: auto !important;
-        z-index: 9999 !important;
-        position: relative !important;
+        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;
+        min-height: 50px !important;
+        margin: 8px 4px !important;
     }
-    """
-) as demo:
-    gr.Markdown("# 📄 PDF 摘要 & 問答助手")
     
-    with gr.Tab("🔧 設定"):
-        gr.Markdown("## API Key 設定")
-        
-        api_key_input = gr.Textbox(
-            label="輸入 OpenAI API Key", 
-            type="password",
-            placeholder="請輸入您的 OpenAI API Key (sk-...)"
-        )
-        
-        with gr.Row():
-            api_key_btn = gr.Button("確認設定 API Key", variant="primary")
-        
-        api_key_status = gr.Textbox(
-            label="API 狀態", 
-            interactive=False,
-            value="等待設定 API Key..."
-        )
-        
-        gr.Markdown("## 模型選擇")
-        
-        model_choice = gr.Radio(
-            choices=["gpt-4", "gpt-4.1", "gpt-4.5"], 
-            label="選擇 AI 模型",
-            value="gpt-4"
-        )
-        
-        with gr.Row():
-            model_btn = gr.Button("確認選擇模型", variant="secondary")
-        
-        model_status = gr.Textbox(
-            label="當前模型", 
-            interactive=False,
-            value="已選擇：gpt-4"
-        )
+    button:hover {
+        transform: translateY(-2px) !important;
+        box-shadow: 0 8px 25px rgba(102, 126, 234, 0.6) !important;
+    }
     
-    with gr.Tab("📄 PDF 處理"):
-        gr.Markdown("## 上傳 PDF 文件")
-        
-        pdf_upload = gr.File(
-            label="選擇 PDF 文件",
-            file_types=[".pdf"]
-        )
-        
-        with gr.Row():
-            summary_btn = gr.Button("生成摘要", variant="primary")
-            clear_btn = gr.Button("清除資料", variant="secondary")
-        
-        summary_output = gr.Textbox(
-            label="PDF 摘要",
-            lines=15,
-            placeholder="上傳 PDF 文件並點擊「生成摘要」按鈕"
-        )
+    /* 次要按鈕 */
+    button[variant="secondary"] {
+        background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%) !important;
+    }
+    
+    button[variant="secondary"]:hover {
+        background: linear-gradient(135deg, #e081e9 0%, #e3455a 100%) !important;
+    }
+    
+    /* 文件上傳區域 */
+    .file-upload {
+        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;
+        width: 100% !important;
+    }
+    
+    .file-upload:hover {
+        background: rgba(102, 126, 234, 0.1) !important;
+        border-color: #764ba2 !important;
+    }
+    
+    /* 行和列的布局 */
+    .gr-row {
+        display: flex !important;
+        gap: 20px !important;
+        width: 100% !important;
+        margin: 15px 0 !important;
+    }
+    
+    .gr-column {
+        flex: 1 !important;
+        min-width: 0 !important;
+    }
     
-    with gr.Tab("❓ 問答"):
-        gr.Markdown("## AI 問答")
+    /* 文字輸出區域 */
+    .gr-textbox {
+        width: 100% !important;
+    }
+    
+    /* 隱藏所有 footer 和 logo */
+    footer,
+    .gradio-container footer,
+    div[class*="footer"],
+    div[class*="Footer"],
+    .gr-footer,
+    .gradio-footer {
+        display: none !important;
+    }
+    
+    /* 響應式設計 */
+    @media (max-width: 768px) {
+        .main-content {
+            margin: 5px !important;
+            padding: 15px !important;
+            width: calc(100vw - 10px) !important;
+        }
         
-        question_input = gr.Textbox(
-            label="請輸入您的問題",
-            placeholder="例如：這份文件的主要結論是什麼？",
-            lines=3
-        )
+        .main-header {
+            font-size: 2em !important;
+        }
         
-        with gr.Row():
-            question_btn = gr.Button("發送問題", variant="primary")
+        .gr-row {
+            flex-direction: column !important;
+        }
         
-        answer_output = gr.Textbox(
-            label="AI 回答",
-            lines=10,
-            placeholder="AI 回答將顯示在這裡"
-        )
-    
-    # 事件綁定 - 使用最明確的方式
-    def handle_api_key():
-        return set_api_key(api_key_input.value if hasattr(api_key_input, 'value') else "")
-    
-    def handle_model():
-        return set_model(model_choice.value if hasattr(model_choice, 'value') else "gpt-4")
+        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>
+        
+        <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; flex: 1; max-width: 300px;">
+                    <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; flex: 1; max-width: 300px;">
+                    <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; flex: 1; max-width: 300px;">
+                    <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>
+        """)
     
-    def handle_summary():
-        return generate_summary(pdf_upload.value if hasattr(pdf_upload, 'value') else None)
+        with gr.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-...)"
+                    )
+                    api_key_status = gr.Textbox(
+                        label="📊 API 連接狀態", 
+                        interactive=False,
+                        value="🔄 等待設定 API Key..."
+                    )
+                
+                with gr.Column():
+                    model_choice = gr.Radio(
+                        ["gpt-4", "gpt-4.1", "gpt-4.5"], 
+                        label="🤖 選擇 AI 模型",
+                        value="gpt-4"
+                    )
+                    model_status = gr.Textbox(
+                        label="🎯 當前模型", 
+                        interactive=False,
+                        value="✅ 已選擇：gpt-4"
+                    )
     
-    def handle_question():
-        return ask_question(question_input.value if hasattr(question_input, 'value') else "")
+        with gr.Tab("📄 PDF 處理"):
+            gr.Markdown("### 📁 文件上傳與摘要生成")
+            with gr.Row():
+                with gr.Column():
+                    pdf_upload = gr.File(
+                        label="📎 選擇 PDF 文件",
+                        file_types=[".pdf"]
+                    )
+                    gr.Markdown("**支援格式**：PDF 文件 (建議 < 10MB)")
+                    
+                    with gr.Row():
+                        summary_btn = gr.Button(
+                            "🔄 開始生成摘要", 
+                            variant="primary"
+                        )
+                        clear_btn = gr.Button(
+                            "🗑️ 清除所有資料", 
+                            variant="secondary"
+                        )
+                
+                with gr.Column():
+                    summary_output = gr.Textbox(
+                        label="📋 AI 生成的文檔摘要",
+                        lines=20,
+                        placeholder="📄 上傳 PDF 文件並點擊「開始生成摘要」按鈕，AI 將為您分析文檔內容..."
+                    )
     
-    # 綁定點擊事件
-    api_key_btn.click(set_api_key, inputs=[api_key_input], outputs=[api_key_status])
-    model_btn.click(set_model, inputs=[model_choice], outputs=[model_status])
-    summary_btn.click(generate_summary, inputs=[pdf_upload], outputs=[summary_output])
-    question_btn.click(ask_question, inputs=[question_input], outputs=[answer_output])
-    clear_btn.click(clear_all, inputs=[], outputs=[summary_output, question_input, answer_output])
+        with gr.Tab("❓ 智能問答"):
+            gr.Markdown("### 💬 基於文檔內容的 AI 問答")
+            with gr.Row():
+                with gr.Column():
+                    question_input = gr.Textbox(
+                        label="💭 請輸入您的問題",
+                        placeholder="例如：這份文件的主要結論是什麼？文中提到的關鍵數據有哪些？",
+                        lines=4
+                    )
+                    question_btn = gr.Button(
+                        "📤 發送問題", 
+                        variant="primary"
+                    )
+                    
+                    gr.Markdown("""
+                    **💡 問題範例：**
+                    - 這份文件討論的主要議題是什麼？
+                    - 文中有哪些重要的統計數據？
+                    - 作者的主要觀點和結論是什麼？
+                    - 文件中提到的建議有哪些？
+                    """)
+                
+                with gr.Column():
+                    answer_output = gr.Textbox(
+                        label="🤖 AI 智能回答",
+                        lines=18,
+                        placeholder="🤖 請先上傳並生成 PDF 摘要，然後在左側輸入問題，AI 將基於文檔內容為您提供詳細回答..."
+                    )
     
-    # Enter 鍵支援
-    api_key_input.submit(set_api_key, inputs=[api_key_input], outputs=[api_key_status])
-    question_input.submit(ask_question, inputs=[question_input], outputs=[answer_output])
+    # 事件處理器
+    api_key_input.submit(set_api_key, inputs=api_key_input, outputs=api_key_status)
+    model_choice.change(set_model, inputs=model_choice, outputs=model_status)
+    summary_btn.click(generate_summary, inputs=pdf_upload, outputs=summary_output)
+    question_btn.click(ask_question, inputs=question_input, outputs=answer_output)
+    question_input.submit(ask_question, inputs=question_input, outputs=answer_output)
+    clear_btn.click(clear_all, outputs=[summary_output, question_input, answer_output])
 
 if __name__ == "__main__":
     demo.launch(
         share=False,
         show_api=False,
-        show_error=True,
-        favicon_path=None
+        show_error=True
     )