Update main.py

main.py

@@ -170,7 +170,7 @@ 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
+    user_text = event.message.text if event.message.type == "text" 
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()
 
@@ -187,9 +187,7 @@ def handle_image_message(event):
             )
         else:
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
-    else:
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成失敗，請稍後再試~"))
-    
+
     
     # 從 LINE API 獲取圖片數據
     image_path = get_image_url(event.message.id)