Update main.py

main.py

@@ -6,7 +6,7 @@ import google.generativeai as genai
 import openai  # 導入 OpenAI 函式庫
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
-from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
+from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage, ImageMessage
 
 #設定open AI API 金鑰
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")