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 |
-
|
| 9 |
|
| 10 |
# 設定 Google AI API 金鑰
|
| 11 |
client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
|
|
@@ -16,13 +16,13 @@ generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, te
|
|
| 16 |
|
| 17 |
# Retrieve the PDF
|
| 18 |
files = glob.glob('docs/*.pdf')
|
| 19 |
-
|
| 20 |
-
for
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
pdf_content
|
| 26 |
|
| 27 |
# 設定 Line Bot 的 API 金鑰和秘密金鑰
|
| 28 |
line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
|
|
|
|
| 5 |
from linebot import LineBotApi, WebhookHandler
|
| 6 |
from linebot.exceptions import InvalidSignatureError
|
| 7 |
from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
|
| 8 |
+
import PyPDF2
|
| 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 |
+
pdf_content = ''
|
| 20 |
+
for filename in files:
|
| 21 |
+
pdf_file = open(filename, 'rb')
|
| 22 |
+
pdf_reader = PyPDF2.PdfReader(pdf_file)
|
| 23 |
+
for i in range(len(pdf_reader.pages)):
|
| 24 |
+
page = pdf_reader.pages[i]
|
| 25 |
+
pdf_content+= page.extract_text()
|
| 26 |
|
| 27 |
# 設定 Line Bot 的 API 金鑰和秘密金鑰
|
| 28 |
line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
|