import gradio as gr
import requests

# Hugging Face API 端點
API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/ukiyo-e-art"
headers = {"Authorization": "Bearer hf_MySpaceToken"}

# 呼叫模型的函數
def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    if response.status_code == 200:
        return response.content  # 圖片回應為二進位格式
    else:
        return f"Error {response.status_code}: {response.text}"

# 定義 Gradio 介面功能
def generate_image(prompt):
    image_data = query({"inputs": prompt})
    if isinstance(image_data, bytes):
        return image_data  # 返回生成的圖片
    else:
        return "無法生成圖片，請檢查輸入或模型狀態。"

# 建立 Gradio App
interface = gr.Interface(
    fn=generate_image,
    inputs="text",
    outputs="image",
    title="浮世繪生成器",
    description="輸入文字提示，生成日本浮世繪風格的藝術圖片。",
)

interface.launch(share=True)  # 設定 share=True 方便測試