Update main.py

main.py

@@ -86,7 +86,7 @@ async def webhook(
 #==========================
 # 使用者請求生成圖片
 #==========================
-def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-image", title="Hugging Face Generated Image"):
+def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image", title="gemini Generated Image"):
     # 將 binary 資料轉為 PIL Image
     image = PIL.Image.open(io.BytesIO(image_binary))
 
@@ -99,7 +99,7 @@ def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-i
             'album': album,
             'name': name,
             'title': title,
-            'description': f'Generated by Hugging Face - {datetime.now()}'
+            'description': f'Generated by gemini - {datetime.now()}'
         }
 
         # 使用 client 進行圖片上傳
@@ -108,10 +108,8 @@ def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-i
     # 回傳圖片網址
     return uploaded_image['link']
 
-
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
-
     client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
     response = client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
@@ -138,16 +136,6 @@ def generate_image_hf(prompt):
     else:
         print("圖片生成失敗:", response.text)
         return None
-
-def translate_zh_to_en(text):
-    model_name = "Helsinki-NLP/opus-mt-zh-en"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-    
-    inputs = tokenizer(text, return_tensors="pt")
-    outputs = model.generate(**inputs)
-    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return translated_text
     
 #==========================
 # 使用者上傳圖片
@@ -221,7 +209,6 @@ def handle_image_message(event):
         line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
         client = ImgurClient(client_id, client_secret, access_token, refresh_token)
 
-        #prompt_en = translate_zh_to_en(prompt)
         # 生成圖片
         image_binary = generate_image_with_gemini(prompt)