import gradio as gr

def generate_waifu(seed, temperature, truncation_psi, num_samples, interpolation):
    model = gr.load("models/hakurei/waifu-diffusion")
    generated_images = model.generate(
        seed=seed, 
        temperature=temperature, 
        truncation_psi=truncation_psi, 
        num_samples=num_samples, 
        interpolation=interpolation
    )
    return generated_images

iface = gr.Interface(fn=generate_waifu, 
                     inputs=["text", "number", "number", "number", ["linear", "cubic"]],
                     outputs="image",
                     inputs_labels=["Seed", "Temperature", "Truncation Psi", "Num Samples", "Interpolation"],
                     inputs_ranges=[(0, 1000000), (0.1, 2.0), (0.1, 2.0), (1, 10), ["linear", "cubic"]],
                     title="Waifu Generator",
                     description="Generate anime-style waifus with customizable settings.")

iface.launch()