Create app.py

app.py

@@ -0,0 +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": f"Bearer {os.getenv('HF_API_TOKEN')}"}
+
+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()