Update main.py

main.py

@@ -90,25 +90,7 @@ def upload_image_to_imgur_with_token(image_path, access_token):
     except Exception as e:
         print("圖片上傳例外：", e)
         return None
-        
-def generate_image_with_gemini(prompt, size=(512, 512)):
-    model_name = "gemini-2.0-flash-exp-image-generation"
-    image_model = genai.GenerativeModel(model_name)
-    prompt += f"，尺寸：{size[0]}x{size[1]}"
-    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:
-            # 清理先前生成的暫存圖檔（選擇性，可加強）
-            for f in os.listdir("/tmp"):
-                if f.endswith(".png") and os.path.getmtime(f"/tmp/{f}") < os.path.getmtime(tmp_file.name):
-                    try:
-                        os.remove(f"/tmp/{f}")
-                    except:
-                        pass
-            tmp_file.write(image_data)
-            return tmp_file.name
-    return None
+
     
 #==========================
 # 使用者上傳圖片
@@ -179,9 +161,15 @@ def handle_image_message(event):
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()
 
-        image_path = generate_image_with_gemini(prompt, size=(512, 512))
+        image_model = genai.GenerativeModel("gemini-2.0-flash-exp-image-generation")
+        response = image_model.generate_content(prompt)
+
+        if response.parts and hasattr(response.parts[0], "inline_data"):
+            image_data = response.parts[0].inline_data.data
+            access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
+            image_url = upload_image_to_imgur_with_token(image_data,access_token)
+
         if image_path:
-            image_url = f"https://your-domain.com/tmp/{os.path.basename(image_path)}"
             line_bot_api.reply_message(
                 event.reply_token,
                 [