Update main.py

main.py

@@ -96,6 +96,42 @@ def handle_image_message(event):
         store_user_message(user_id, "text", user_text)
         return
         
+    # ========
+    # 生成圖片
+    # ========
+    elif user_text and user_text.startswith("生成圖片"):
+        prompt = user_text.replace("生成圖片", "").strip()
+
+         # 先立即回覆避免token過期
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
+        image_generator = Image_text_Generator(user_id)
+        # 生成圖片
+        image_binary = image_generator.generate_image_with_gemini(prompt)
+
+        if image_binary:
+            image_url = 'https://alanchen1115-linebot.hf.space/'+image_generator.upload_image_to_tmp(image_binary)
+
+            if image_url:
+                # 使用 push message 發送圖片，避免 reply token 超時
+                line_bot_api.push_message(
+                    event.source.user_id,
+                    [
+                        TextSendMessage(text="✨ 這是我為你生成的圖片喔～"),
+                        ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+                    ]
+                )
+            else:
+                line_bot_api.push_message(
+                    event.source.user_id,
+                    TextSendMessage(text="⚠️ 圖片上傳失敗，請稍後再試~")
+                )
+        else:
+            line_bot_api.push_message(
+                event.source.user_id,
+                TextSendMessage(text="⚠️ 圖片生成失敗，請稍後再試~")
+            )
+        return
+        
     # ========
     # 純文字
     # ========
@@ -134,42 +170,6 @@ def handle_image_message(event):
         else:
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="沒有接收到圖片~"))
         return
-        
-    # ========
-    # 生成圖片
-    # ========
-    elif user_text and user_text.startswith("生成圖片"):
-        prompt = user_text.replace("生成圖片", "").strip()
-
-         # 先立即回覆避免token過期
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
-        image_generator = Image_text_Generator(user_id)
-        # 生成圖片
-        image_binary = image_generator.generate_image_with_gemini(prompt)
-
-        if image_binary:
-            image_url = 'https://alanchen1115-linebot.hf.space/'+image_generator.upload_image_to_tmp(image_binary)
-
-            if image_url:
-                # 使用 push message 發送圖片，避免 reply token 超時
-                line_bot_api.push_message(
-                    event.source.user_id,
-                    [
-                        TextSendMessage(text="✨ 這是我為你生成的圖片喔～"),
-                        ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
-                    ]
-                )
-            else:
-                line_bot_api.push_message(
-                    event.source.user_id,
-                    TextSendMessage(text="⚠️ 圖片上傳失敗，請稍後再試~")
-                )
-        else:
-            line_bot_api.push_message(
-                event.source.user_id,
-                TextSendMessage(text="⚠️ 圖片生成失敗，請稍後再試~")
-            )
-        return
 
     line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))