Update main.py

main.py

@@ -17,6 +17,7 @@ from imgurpython import ImgurClient
 import requests
 import tempfile
 from datetime import datetime
+from transformers import pipeline
 
 # 設定 Google AI API 金鑰
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
@@ -190,13 +191,16 @@ def handle_image_message(event):
 
     user_text = event.message.text if event.message.type == "text" else None
     image_url = None
-    if user_text and user_text.startswith("請幫我生成圖片"):
-        prompt = user_text.replace("請幫我生成圖片", "").strip()
+    if user_text and user_text.startswith("生成圖片"):
+        prompt = user_text.replace("生成圖片", "").strip()
 
          # 先立即回覆避免token過期
         line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
-
         client = ImgurClient(client_id, client_secret, access_token, refresh_token)
+
+        # 使用翻譯任務
+        translator = pipeline("translation", model='Helsinki-NLP/opus-mt-zh-en') 
+        prompt = prompt[0]['translation_text']
         image_binary = generate_image_hf(prompt)
 
         if image_binary: