Update main.py

main.py

@@ -5,7 +5,6 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
 import google.generativeai as genai
 import base64
-from datetime import datetime
 from collections import defaultdict
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
@@ -15,7 +14,8 @@ import tempfile
 import httpx
 
 from imgurpython import ImgurClient
-
+import requests
+import tempfile
 
 # 設定 Google AI API 金鑰
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
@@ -113,6 +113,17 @@ 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']}"}
+
+def generate_image_hf(prompt):
+    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+    if response.status_code == 200:
+        return response.content  # 圖片binary資料
+    else:
+        print("圖片生成失敗:", response.text)
+        return None
+        
 #==========================
 # 使用者上傳圖片
 #==========================
@@ -185,7 +196,7 @@ 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)
-        image_binary = generate_image_with_gemini(prompt)
+        image_binary = generate_image_hf(prompt)
 
         if image_binary:
             album = "nvsYwgq"