Update main.py

main.py

@@ -14,7 +14,7 @@ import PIL.Image
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
 # 設定生成文字的參數
-generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
+generation_config = genai.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
 
 # 使用 Gemini-1.5-flash 模型
 model = genai.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
@@ -61,8 +61,51 @@ async def webhook(
         # 處理無效的簽章錯誤
         raise HTTPException(status_code=400, detail="Invalid signature")
     return "ok"
+    
+#==========================
+# 使用者請求生成圖片
+#==========================
+
+def upload_image_to_imgur_with_token(image_path, access_token):
+    """
+    使用 Imgur OAuth2 Access Token 上傳圖片
+    """
+    try:
+        with open(image_path, "rb") as img_file:
+            image_binary = img_file.read()
 
+        headers = {
+            "Authorization": f"Bearer {access_token}"
+        }
 
+        data = {
+            "image": base64.b64encode(image_binary).decode("utf-8"),
+            "type": "base64",
+        }
+
+        response = httpx.post("https://api.imgur.com/3/image", headers=headers, data=data)
+        if response.status_code == 200:
+            return response.json()["data"]["link"]
+        else:
+            print("Imgur 上傳失敗：", response.text)
+            return None
+    except Exception as e:
+        print("圖片上傳例外：", e)
+        return None
+        
+def generate_image_with_gemini(prompt):
+    model_name = "gemini-2.0-flash-exp-image-generation"
+    image_model = genai.GenerativeModel(model_name)
+    response = image_model.generate_content(prompt)
+    if response.parts and hasattr(response.parts[0], "inline_data"):
+        image_data = response.parts[0].inline_data.data
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir="/tmp") as tmp_file:
+            tmp_file.write(image_data)
+            return tmp_file.name
+    return None
+#==========================
+# 使用者上傳圖片
+#==========================
 def get_image_url(message_id):
     try:
         message_content = line_bot_api.get_message_content(message_id)
@@ -112,7 +155,10 @@ def get_previous_message(user_id):
         # 返回最後一則訊息
         return user_message_history[user_id][-1]
     return None
-    
+
+#==========================
+# 主程式(圖片與文字)
+#==========================
 # 建立 chat_sessions 字典
 chat_sessions = {}
 @line_handler.add(MessageEvent, message=(ImageMessage,TextMessage))
@@ -123,6 +169,27 @@ def handle_image_message(event):
     else:
         chat = model.start_chat(history=[])
         chat_sessions[user_id] = chat
+
+    user_text = event.message.text if event.message.type == "text" else None
+    if user_text and user_text.startswith("請幫我生成圖片"):
+        prompt = user_text.replace("請幫我生成圖片", "").strip()
+
+        image_path_gen = generate_image_with_gemini(prompt)
+        if image_path:
+            # 使用 Access Token 上傳至 Imgur
+            IMGUR_ACCESS_TOKEN = os.environ.get("IMGUR_ACCESS_TOKEN")
+            image_url = upload_image_to_imgur_with_token(image_path, IMGUR_ACCESS_TOKEN)
+
+        if image_url:
+            line_bot_api.reply_message(
+                event.reply_token,
+                ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+            )
+        else:
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
+    else:
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成失敗，請稍後再試~"))
+    
     
     # 從 LINE API 獲取圖片數據
     image_path = get_image_url(event.message.id)