Update main.py

main.py

@@ -194,29 +194,47 @@ def handle_message(event):
         agent_input = {"messages": [{"role": "user", "content": user_text}]} 
         try:
             response = agent_executor.invoke(agent_input)
-            raw_content = response["messages"][-1].content
-            print(f"原始內容: {raw_content}")
-
+            messages = response.get("messages", [])
+            
             out = ""
-            # 1. 判斷並提取內容
-            if isinstance(raw_content, str):
-                out = raw_content
-            elif isinstance(raw_content, list):
-                # 串接所有文字區塊
-                out = "".join([block.get("text", "") for block in raw_content if isinstance(block, dict) and block.get("type") == "text"])
-            elif isinstance(raw_content, dict):
-                # 如果直接回傳字典 (如您的 Log 所示)
-                # 優先抓取內容，如果沒內容就轉成字串避免出錯
-                out = raw_content.get("imageUrl", str(raw_content))
-
-            # 2. 搜尋 URL 邏輯 (修正 Regex 並過濾)
-            # 使用 str(out) 確保處理的是字串
-            urls = re.findall(r'https?://[^\s<>"\)]+|www\.[^\s<>"\)]+', str(out))
-            image_url = next((u for u in urls if any(ext in u.lower() for ext in ['.png', '.jpg', '.jpeg', '.gif'])), None)
-
-            # 3. 回覆邏輯
+            image_url = None
+
+            # 1. 由後往前找，優先找出包含 imageUrl 的訊息
+            for msg in reversed(messages):
+                content = msg.content
+                
+                # 如果內容是字典 (工具回傳的格式)
+                if isinstance(content, dict) and "imageUrl" in content:
+                    image_url = content["imageUrl"]
+                    break
+                
+                # 如果內容是字串，嘗試從中提取網址
+                elif isinstance(content, str) and content.strip():
+                    # 這裡用我們先前的 Regex 找找看
+                    found_urls = re.findall(r'https?://[^\s<>"\)]+|www\.[^\s<>"\)]+', content)
+                    img = next((u for u in found_urls if any(ext in u.lower() for ext in ['.png', '.jpg', '.jpeg', '.gif'])), None)
+                    if img:
+                        image_url = img.split(')')[0]
+                        out = content # 保留文字內容
+                        break
+                    elif not out: # 如果還沒找到圖片，先暫存文字內容
+                        out = content
+
+                # 如果內容是列表 (Gemini 2.5 有時會回傳 block list)
+                elif isinstance(content, list):
+                    for block in content:
+                        if isinstance(block, dict):
+                            # 找圖片網址
+                            text = block.get("text", "")
+                            found_urls = re.findall(r'https?://[^\s<>"\)]+|www\.[^\s<>"\)]+', text)
+                            img = next((u for u in found_urls if any(ext in u.lower() for ext in ['.png', '.jpg', '.jpeg', '.gif'])), None)
+                            if img:
+                                image_url = img.split(')')[0]
+                            if text:
+                                out += text
+
+            # 2. 最終回覆邏輯
             if image_url:
-                image_url = image_url.split(')')[0]
                 line_bot_api.reply_message(
                     event.reply_token,
                     [
@@ -224,15 +242,15 @@ def handle_message(event):
                         ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
                     ]
                 )
+            elif out.strip():
+                line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out.strip()))
             else:
-                # 確保 out 不是空的，否則 LINE 會報 400 錯誤
-                final_text = out.strip() if out else "處理完成，但沒有文字內容。"
-                line_bot_api.reply_message(event.reply_token, TextSendMessage(text=final_text))
+                # 萬一真的什麼都沒有，回報一個預設訊息，避免 LINE 噴 400 錯誤
+                line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片已生成，但我暫時無法取得連結，請稍後再試。"))
 
         except Exception as e:
             print(f"Agent Error: {e}")
-            # 同樣確保錯誤回覆不是空的
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="抱歉，系統處理時發生錯誤。"))
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="抱歉，我現在無法處理這個請求。"))
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)