Update main.py

main.py

@@ -78,9 +78,9 @@ async def webhook(
 #==========================
 # 使用者請求生成圖片
 #==========================
-def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image", title="Gemini Generated Image"):
+def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-image", title="Hugging Face Generated Image"):
     # 將 binary 資料轉為 PIL Image
-    image = Image.open(io.BytesIO(image_binary))
+    image = PIL.Image.open(io.BytesIO(image_binary))
 
     # 建立暫存檔案來上傳 (因為 ImgurClient 需要檔案路徑)
     with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmp:
@@ -91,7 +91,7 @@ def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image",
             'album': album,
             'name': name,
             'title': title,
-            'description': f'Generated by Gemini - {datetime.now()}'
+            'description': f'Generated by Hugging Face - {datetime.now()}'
         }
 
         # 使用 client 進行圖片上傳
@@ -113,10 +113,10 @@ def generate_image_with_gemini(prompt):
 
     return None
 
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
-
+# 使用 Hugging Face 生成圖片
 def generate_image_hf(prompt):
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
+    headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     if response.status_code == 200:
         return response.content  # 圖片binary資料