Update main.py
Browse files
main.py
CHANGED
|
@@ -5,7 +5,7 @@ from google import genai
|
|
| 5 |
from linebot import LineBotApi, WebhookHandler
|
| 6 |
from linebot.exceptions import InvalidSignatureError
|
| 7 |
from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
|
| 8 |
-
from pypdf import
|
| 9 |
|
| 10 |
# 設定 Google AI API 金鑰
|
| 11 |
client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
|
|
@@ -16,7 +16,7 @@ generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, te
|
|
| 16 |
|
| 17 |
# Retrieve the PDF
|
| 18 |
files = glob.glob('docs/*.pdf')
|
| 19 |
-
merger =
|
| 20 |
for file in files:
|
| 21 |
merger.append(file)
|
| 22 |
merger.write("docs/combined.pdf")
|
|
|
|
| 5 |
from linebot import LineBotApi, WebhookHandler
|
| 6 |
from linebot.exceptions import InvalidSignatureError
|
| 7 |
from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
|
| 8 |
+
from pypdf import PdfWriter
|
| 9 |
|
| 10 |
# 設定 Google AI API 金鑰
|
| 11 |
client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
|
|
|
|
| 16 |
|
| 17 |
# Retrieve the PDF
|
| 18 |
files = glob.glob('docs/*.pdf')
|
| 19 |
+
merger = PdfWriter()
|
| 20 |
for file in files:
|
| 21 |
merger.append(file)
|
| 22 |
merger.write("docs/combined.pdf")
|