Update main.py

main.py

@@ -17,7 +17,7 @@ genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
 
 # 使用 Gemini-1.5-flash 模型
-model = genai.GenerativeModel('gemini-1.5-flash', system_instruction="請用繁體中文回答。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題") # 或是使用 "你是博通古今的萬應機器人！"
+model = genai.GenerativeModel('gemini-1.5-flash', system_instruction="請用繁體中文回答。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -96,7 +96,6 @@ def analyze_with_gemini(image_path, user_text):
             raise FileNotFoundError(f"圖片路徑無效：{image_path}")
         
         organ = PIL.Image.open(image_path)
-        chat = model.start_chat()
         response = chat.send_message([user_text, organ])
         
         # 提取回應內容
@@ -138,11 +137,11 @@ def handle_image_message(event):
 
         # 儲存當前文字訊息
         user_text = event.message.text
-        user = store_user_message(user_id, "text", user_text)
+        store_user_message(user_id, "text", user_text)
 
         # 結合圖片與文字分析
         out = analyze_with_gemini(image_path, user_text)
-        GeneAI = store_user_message("AI", "text", out)
+
     else:
         global working_status
         # 檢查事件類型和訊息類型
@@ -164,16 +163,19 @@ def handle_image_message(event):
             try: 
                 # 取得使用者輸入的文字
                 prompt = event.message.text
-                user = store_user_message(user_id, "text", prompt)
-                GeneAI = store_user_message("AI", "text", out)
+                store_user_message(user_id, "text", prompt) 
+                
+                history = [{"role": "user", "parts": [store_user_message]}]
+                    if GeneAI:
+                        history.append({"role": "model", "parts": [out]})
+   
                 # 使用 Gemini 模型生成文字
-                chat = model.start_chat(history = [{"role": "user","parts": [user]},{"role": "model", "parts":[GeneAI]}])
-                completion = chat.send_message(prompt, generation_config=generation_config)
+                chat = model.start_chat(history=history)
+                completion = chat.send_message(prompt)
                 # 檢查生成結果是否為空
                 if (completion.parts[0].text != None):
                     # 取得生成結果
                     out = completion.parts[0].text
-                    GeneAI = store_user_message("AI", "text", out)
                 else:
                     # 回覆 "Gemini沒答案!請換個說法！"
                     out = "我不太懂什麼意思也~"