Update main.py

main.py

@@ -138,11 +138,11 @@ def handle_image_message(event):
 
         # 儲存當前文字訊息
         user_text = event.message.text
-        store_user_message(user_id, "text", user_text)
+        user = 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,14 +164,16 @@ def handle_image_message(event):
             try: 
                 # 取得使用者輸入的文字
                 prompt = event.message.text
-                store_user_message(user_id, "text", prompt)
+                user = store_user_message(user_id, "text", prompt)
+                GeneAI = store_user_message("AI", "text", out)
                 # 使用 Gemini 模型生成文字
-                chat = model.start_chat(history = [])
+                chat = model.start_chat(history = [{"role": "user","parts": [user]},{"role": "model", "parts":[GeneAI]}])
                 completion = chat.send_message(prompt, generation_config=generation_config)
                 # 檢查生成結果是否為空
                 if (completion.parts[0].text != None):
                     # 取得生成結果
                     out = completion.parts[0].text
+                    GeneAI = store_user_message("AI", "text", out)
                 else:
                     # 回覆 "Gemini沒答案!請換個說法！"
                     out = "我不太懂什麼意思也~"