Update main.py

main.py

@@ -118,6 +118,12 @@ def get_previous_message(user_id):
 @line_handler.add(MessageEvent, message=(ImageMessage,TextMessage))
 def handle_image_message(event):
     user_id = event.source.user_id
+    if user_id in chat_sessions:
+        chat_session = chat_sessions[user_id]
+    else:
+        chat_session = model.start_chat(history=[])
+    chat[user_id] = chat_session
+    
     # 從 LINE API 獲取圖片數據
     image_path = get_image_url(event.message.id)
     if image_path: