Update main.py

main.py

@@ -11,8 +11,8 @@ from collections import defaultdict
 import uvicorn
 
 # 檢查環境變數
-if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY"]):
-    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, and GOOGLE_API_KEY.")
+if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY", "IMGUR_CLIENT_ID"]):
+    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and IMGUR_CLIENT_ID.")
 
 # Line Bot API 設定
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -32,6 +32,7 @@ user_states = defaultdict(lambda: {
 MAX_IMAGES_PER_TYPE = 3
 GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-preview-05-20:generateContent"
 GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
+IMGUR_CLIENT_ID = os.environ["IMGUR_CLIENT_ID"]
 
 app = FastAPI()
 
@@ -65,6 +66,20 @@ def get_base64_image(message_id: str):
     image_bytes = message_content.content
     return base64.b64encode(image_bytes).decode('utf-8')
 
+def upload_to_imgur(image_base64: str):
+    """
+    將 Base64 編碼的圖片上傳到 Imgur。
+    """
+    headers = {"Authorization": f"Client-ID {IMGUR_CLIENT_ID}"}
+    payload = {"image": image_base64}
+    try:
+        response = requests.post("https://api.imgur.com/3/image", headers=headers, data=payload)
+        response.raise_for_status()
+        return response.json()["data"]["link"]
+    except requests.exceptions.RequestException as e:
+        print(f"Error uploading to Imgur: {e}")
+        return None
+
 def get_gemini_response(prompt: str, images: list):
     payload = {
         "contents": [
@@ -180,7 +195,9 @@ def handle_image_message(event):
         try:
             image_id = event.message.id
             base64_img = get_base64_image(image_id)
-            user_states[user_id]["personal_photo"] = base64_img
+            # 將個人照片上傳到 Imgur，並儲存 URL
+            photo_url = upload_to_imgur(base64_img)
+            user_states[user_id]["personal_photo"] = photo_url
             user_states[user_id]["is_ready_for_photo"] = True
             
             line_bot_api.reply_message(
@@ -192,6 +209,7 @@ def handle_image_message(event):
             user_info = user_states[user_id]["user_info"]
             occasion = user_info["occasion"]
             
+            # Gemini 提示詞現在使用圖片 URL，而不是 Base64 編碼
             prompt = (
                 f"我提供了三件上衣圖片和三件下半身圖片，以及使用者的一張個人照片。使用者資訊：身高 {user_info['height']}cm，三圍 {user_info['bust']}-{user_info['waist']}-{user_info['hip']}，場合是「{occasion}」。"
                 "請根據這些衣物，為我推薦一套最適合的穿搭，並詳細說明為何這套搭配適合這個場合。請以繁體中文回答。"
@@ -211,7 +229,7 @@ def handle_image_message(event):
             best_lower_body_image = user_states[user_id]["lower_body_images"][best_lower_index]
 
             # 呼叫虛擬試穿 API，並取得結果圖片 URL
-            virtual_try_on_url = get_virtual_tryon_image(user_states[user_id]["personal_photo"], best_upper_body_image, best_lower_body_image)
+            virtual_tryon_url = get_virtual_tryon_image(user_states[user_id]["personal_photo"], best_upper_body_image, best_lower_body_image)
             
             # 發送 Gemini 的文字建議
             line_bot_api.push_message(
@@ -222,7 +240,7 @@ def handle_image_message(event):
             # 發送虛擬試穿照片
             line_bot_api.push_message(
                 user_id,
-                ImageSendMessage(original_content_url=virtual_try_on_url, preview_image_url=virtual_try_on_url)
+                ImageSendMessage(original_content_url=virtual_tryon_url, preview_image_url=virtual_tryon_url)
             )
 
             # 重置狀態以便下一次使用
@@ -248,10 +266,18 @@ def handle_image_message(event):
     try:
         image_id = event.message.id
         base64_img = get_base64_image(image_id)
-        
+        # 上傳圖片到 Imgur，並儲存 URL
+        image_url = upload_to_imgur(base64_img)
+        if not image_url:
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text="圖片上傳 Imgur 失敗，請稍後再試。")
+            )
+            return
+            
         if mode == "upper":
             if len(user_states[user_id]["upper_body_images"]) < MAX_IMAGES_PER_TYPE:
-                user_states[user_id]["upper_body_images"].append(base64_img)
+                user_states[user_id]["upper_body_images"].append(image_url)
             else:
                 line_bot_api.reply_message(
                     reply_token,
@@ -260,7 +286,7 @@ def handle_image_message(event):
                 return
         else: # mode == "lower"
             if len(user_states[user_id]["lower_body_images"]) < MAX_IMAGES_PER_TYPE:
-                user_states[user_id]["lower_body_images"].append(base64_img)
+                user_states[user_id]["lower_body_images"].append(image_url)
             else:
                 line_bot_api.reply_message(
                     reply_token,