Update main.py

main.py

@@ -169,24 +169,6 @@ def handle_image_message(event):
         chat = model.start_chat(history=[])
         chat_sessions[user_id] = chat
 
-    user_text = event.message.text if event.message.type == "text" else None
-    if user_text and user_text.startswith("請幫我生成圖片"):
-        prompt = user_text.replace("請幫我生成圖片", "").strip()
-
-        image_path_gen = generate_image_with_gemini(prompt)
-        if image_path:
-            # 使用 Access Token 上傳至 Imgur
-            IMGUR_ACCESS_TOKEN = os.environ.get("IMGUR_ACCESS_TOKEN")
-            image_url = upload_image_to_imgur_with_token(image_path, IMGUR_ACCESS_TOKEN)
-
-        if image_url:
-            line_bot_api.reply_message(
-                event.reply_token,
-                ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
-            )
-        else:
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
-
     
     # 從 LINE API 獲取圖片數據
     image_path = get_image_url(event.message.id)
@@ -241,6 +223,20 @@ def handle_image_message(event):
         # 檢查是否正在與使用者交談
         elif working_status:
             try: 
+
+                 if user_text and user_text.startswith("請幫我生成圖片"):
+                    prompt = user_text.replace("請幫我生成圖片", "").strip()
+
+                    image_path_gen = generate_image_with_gemini(prompt)
+                    if image_path:
+                    # 使用 Access Token 上傳至 Imgur
+                    IMGUR_ACCESS_TOKEN = os.environ.get("IMGUR_ACCESS_TOKEN")
+                    image_url = upload_image_to_imgur_with_token(image_path, IMGUR_ACCESS_TOKEN)
+
+                    if image_url:
+                        line_bot_api.reply_message(event.reply_token,ImageSendMessage(original_content_url=image_url, preview_image_url=image_url))
+        else:
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
                 # 取得使用者輸入的文字
                 prompt = event.message.text
                 store_user_message(user_id, "text", prompt)