Update main.py

main.py

@@ -10,10 +10,6 @@ import PyPDF2
 # 設定 Google AI API 金鑰
 client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# 設定生成文字的參數 + 角色扮演
-generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, temperature=0.1, top_p=0.2,
-                    system_instruction="您是一位問答助手。請僅限使用以上提供的內容來回答問題。如果您不知道答案，請說您不知道，不要捏造答案。")
-
 # Retrieve the PDF
 files = glob.glob('docs/*.pdf')
 pdf_content = ''
@@ -23,6 +19,11 @@ for filename in files:
     for i in range(len(pdf_reader.pages)):
         page = pdf_reader.pages[i]
         pdf_content+= page.extract_text()
+        
+# 設定生成文字的參數 + 角色扮演
+generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, temperature=0.1, top_p=0.2,
+                    system_instruction=pdf_content + "\n" +
+                    "您是一位問答助手。請僅限使用以上提供的內容來回答問題。如果您不知道答案，請說您不知道，不要捏造答案。")
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -97,7 +98,7 @@ def handle_message(event):
             # Combine the PDF contents and the new prompt into a single list for the 'contents' argument
             completion = client.models.generate_content(
                                 model="gemini-2.0-flash",
-                                contents=[pdf_content, prompt],
+                                contents=prompt,
                                 config=generation_config).text
             # 檢查生成結果是否為空
             if (completion != None):