Update app.py

app.py

@@ -1,15 +1,18 @@
 import openai
 import gradio as gr
 import fitz  # PyMuPDF
+from openai import OpenAI  # ✅ 新增正確 Client
 
 api_key = ""
 selected_model = ""
 summary_text = ""
 
+client = None  # ✅ 全域 client 物件
+
 def set_api_key(user_api_key):
-    global api_key
+    global api_key, client
     api_key = user_api_key
-    openai.api_key = api_key  # ✅ 重點修正：直接全域設置
+    client = OpenAI(api_key=api_key)  # ✅ 正確初始化新版 Client
     return "✅ API Key 已設定"
 
 def set_model(model_name):
@@ -29,25 +32,25 @@ def generate_summary(pdf_file):
     pdf_text = extract_pdf_text(pdf_file)
     if not pdf_text.strip():
         return "⚠️ 無法解析 PDF 文字，可能為純圖片 PDF。"
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model=selected_model,
         messages=[
             {"role": "system", "content": "請將以下 PDF 內容整理為條列式摘要重點。"},
             {"role": "user", "content": pdf_text[:4000]}
         ]
     )
-    summary_text = response['choices'][0]['message']['content']
+    summary_text = response.choices[0].message.content
     return summary_text
 
 def ask_question(user_question):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model=selected_model,
         messages=[
             {"role": "system", "content": f"根據以下 PDF 摘要內容回答問題：\n{summary_text}"},
             {"role": "user", "content": user_question}
         ]
     )
-    return response['choices'][0]['message']['content']
+    return response.choices[0].message.content
 
 with gr.Blocks() as demo:
     gr.Markdown("# 📄 PDF 摘要 & 問答助手 (Hugging Face 版)")