import gradio as gr
import numpy as np
from PIL import Image, ImageDraw, ImageFont

# Função para gerar a imagem
def generate_image():
    # Criar uma imagem em branco
    img = Image.new("RGB", (512, 512), color=(255, 255, 255))
    draw = ImageDraw.Draw(img)

    # Definir gradiente de amarelo para rosa
    gradient = np.zeros((512, 512, 3), dtype=np.uint8)
    for i in range(512):
        for j in range(512):
            r = int((i / 512) * 255)  # Gradiente de vermelho
            g = int((i / 512) * 255)  # Gradiente de verde
            b = int(255 - (i / 512) * 255)  # Gradiente de azul
            gradient[i, j] = [r, g, b]

    # Converter o gradiente para uma imagem PIL
    gradient_img = Image.fromarray(gradient)
    img.paste(gradient_img, (0, 0))

    # Adicionar o título na imagem
    font = ImageFont.load_default()
    draw.text((20, 20), "FLUX.1 [dev] 🖥️", font=font, fill=(0, 0, 0))

    return img

# Definir a interface Gradio
with gr.Blocks() as demo:
    with gr.Row():
        gr.Markdown("### FLUX.1 [dev] 🖥️")
        gr.Markdown(
            "This is a Gradio app using FLUX.1. The app generates an image with a yellow to pink gradient."
        )
    result = gr.Image(label="Generated Image")

    gr.Button("Generate Image").click(generate_image, outputs=[result])

demo.launch()