Update main.py

main.py

@@ -5,6 +5,7 @@ from google import genai
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
+from pypdf import PdfMerger
 
 # 設定 Google AI API 金鑰
 client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -15,7 +16,13 @@ generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, te
 
 # Retrieve the PDF
 files = glob.glob('docs/*.pdf')
-files_content = [client.files.upload(file=filepath) for filepath in files]
+merger = PdfMerger()
+for file in files:
+    merger.append(file)
+merger.write("docs/combined.pdf")
+merger.close()
+
+pdf_content = client.files.upload(file="docs/combined.pdf")
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -89,15 +96,14 @@ def handle_message(event):
             # 取得使用者輸入的文字
             prompt = event.message.text
             # Combine the PDF contents and the new prompt into a single list for the 'contents' argument
-            combined_contents = files_content + [prompt]
             completion = client.models.generate_content(
                                 model="gemini-2.5-flash",
-                                contents=combined_contents,
-                                config=generation_config)
+                                contents=[pdf_content, prompt],
+                                config=generation_config).text
             # 檢查生成結果是否為空
             if (completion != None):
                 # 取得生成結果
-                out = completion.text
+                out = completion
             else:
                 # 回覆 "Gemini沒答案!請換個說法！"
                 out = "Gemini沒答案!請換個說法！"