Update main.py

main.py

@@ -141,11 +141,12 @@ def get_previous_message(user_id):
 @line_handler.add(MessageEvent, message=(ImageMessage,TextMessage))
 def handle_image_message(event):
     # 從 LINE API 獲取圖片數據
+    user_id = event.source.user_id
     image_path = get_image_url(event.message.id)
     if image_path:
         # 處理圖片或進行進一步分析
         print(f"圖片已保存至: {image_path}")
-        
+        store_user_message(user_id, "image", image_path)
         # 回應用戶
         line_bot_api.reply_message(
             event.reply_token,
@@ -159,14 +160,12 @@ def handle_image_message(event):
         )
 
     
-    user_id = event.source.user_id
-    store_user_message(user_id, "text", user_text)
     previous_message = get_previous_message(user_id)
-    
     # 獲取傳送圖片後的用戶問題（文字）
     if previous_message and previous_message["type"] == "image":
         # 獲取用戶問題（文字）
         user_text = event.message.text  # 自行實現，根據需求追蹤用戶文字
+        store_user_message(user_id, "text", user_text)
         # 將圖片與文字結合發送到 OpenAI
         out = analyze_with_openai(image_path, user_text)
 
@@ -194,6 +193,8 @@ def handle_image_message(event):
             try: 
                 # 取得使用者輸入的文字
                 prompt = event.message.text
+                user_text = event.message.text 
+                store_user_message(user_id, "text", user_text)
                 # 使用 Gemini 模型生成文字
                 completion = model.generate_content(prompt, generation_config=generation_config)
                 # 檢查生成結果是否為空