Update app.py

app.py

@@ -1,11 +1,35 @@
+import gradio as gr
 import requests
+import io
+from PIL import Image
+from dotenv import load_dotenv
+import os
 
+# 載入環境變數
+load_dotenv()
+
+# Hugging Face API 設定
 API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/ukiyo-e-art"
-headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
-
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.content
-image_bytes = query({
-	"inputs": "Astronaut riding a horse",
-})
+headers = {"Authorization": "Bearer hf_MySpaceToken"}
+
+def generate_image(prompt):
+    try:
+        # 呼叫 Hugging Face API
+        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+        # 轉換回應為圖像
+        image = Image.open(io.BytesIO(response.content))
+        return image
+    except Exception as e:
+        return None
+
+# 建立 Gradio 介面
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="請輸入圖像描述"),
+    outputs=gr.Image(label="生成的圖像"),
+    title="浮世繪風格圖像生成器",
+    description="輸入描述文字,生成浮世繪風格的圖像"
+)
+
+# 啟動應用
+iface.launch()