Update main.py

main.py

@@ -177,7 +177,6 @@ llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.5-flash"
 prompt_template = ChatPromptTemplate.from_messages([
     ("system", "你是一個強大的助理，可以根據用戶的請求使用提供的工具。"),
     ("user", "{input}"),
-    # 修正此處的語法，使用字串 "{agent_scratchpad}"
     ("placeholder", "{agent_scratchpad}"),
 ])
 
@@ -248,11 +247,23 @@ def handle_message(event):
             response = agent_executor.invoke(agent_input)
             out = response["output"]
             
+            if 'htts' in out:
+                img_tmp = out.split('https')[1]
+                image_url = img_tmp.split('png')[0]
+
+            if image_url:
+                # 使用 push message 發送圖片，避免 rep.ly token 超時
+                line_bot_api.push_message(
+                    event.source.user_id,
+                    [
+                        TextSendMessage(text="✨ 這是我為你生成的圖片喔～"),
+                        ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+                    ]
+                )
         except Exception as e:
             print(f"代理人執行出錯: {e}")
             out = f"代理人執行出錯!請換個說法或稍後再試！錯誤訊息：{e}"
-            
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
 
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)