Merge pull request #2 from gdsc-ncku/flex-msg

feat: handle text for flex message

Author: yenslife

linebot/app/routers/linebot.py

@@ -15,8 +15,8 @@
 @handler.add(MessageEvent, message=TextMessage)
 def handle_text_message(event):
     print("收到文字訊息")
-    reply_message = message_service.handle_text_message(event)
-    message_service.send_message(event.reply_token, [reply_message])
+    reply_messages = message_service.handle_text_message(event)
+    message_service.send_message(event.reply_token, reply_messages)
 
 
 @handler.add(MessageEvent, message=ImageMessage)

linebot/app/services/message_service.py

@@ -1,12 +1,13 @@
 import logging
-import os
+import json
 
 from linebot.models import (
     TextSendMessage,
     QuickReply,
     QuickReplyButton,
     MessageAction,
     SendMessage,
+    FlexSendMessage,
 )
 from ..api.vision import process_image
 from ..config.line_config import line_bot_api
@@ -37,12 +38,43 @@ def handle_text_message(self, event) -> TextSendMessage:
             # 產生回應訊息
             quick_reply = None
             if user_input in COMMANDS:
-                quick_reply = self._create_quick_reply()
                 response_text = COMMANDS[user_input]
-            else:
-                response_text = inference(user_input, user_id)
-
-            return TextSendMessage(text=response_text, quick_reply=quick_reply)
+                quick_reply = self._create_quick_reply()
+                return [TextSendMessage(text=response_text, quick_reply=quick_reply)]
+
+            # 處理一般查詢
+            response_text = inference(user_input, user_id)
+            quick_reply = self._create_quick_reply()
+
+            # 處理可能包含 Flex Message 的回應
+            if "===FLEX_MESSAGE===" in response_text:
+                parts = response_text.split("===FLEX_MESSAGE===")
+                text_content = parts[0].strip()
+                text_message = TextSendMessage(
+                    text=text_content, quick_reply=quick_reply
+                )
+
+                # 檢查是否有 Flex Message 部分
+                if len(parts) > 1:
+                    flex_content = (
+                        parts[1].strip().replace("```", "").replace("json", "")
+                    )
+                    if flex_content and flex_content != "False":
+                        try:
+                            flex_json = json.loads(flex_content)
+                            return [
+                                text_message,
+                                FlexSendMessage(
+                                    alt_text="詳細資訊", contents=flex_json
+                                ),
+                            ]
+                        except Exception as e:
+                            logger.error(f"Flex訊息解析錯誤: {str(e)}", exc_info=True)
+
+                return [text_message]
+
+            # 純文字回應
+            return [TextSendMessage(text=response_text, quick_reply=quick_reply)]
 
         except Exception as e:
             logger.error(f"處理訊息時發生錯誤: {str(e)}", exc_info=True)