app.py

import gradio as gr
import numpy as np
import random
#import spaces #[uncomment to use ZeroGPU]
from diffusers import DiffusionPipeline
import torch

device = "cuda" if torch.cuda.is_available() else "cpu"
model_repo_id = "stabilityai/sdxl-turbo" #Replace to the model you would like to use

if torch.cuda.is_available():
    torch_dtype = torch.float16
else:
    torch_dtype = torch.float32

pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
pipe = pipe.to(device)

MAX_SEED = np.iinfo(np.int32).max
MAX_IMAGE_SIZE = 1024

# Système de consigne pour Minecraft
system_prompt = """Un homme souriant avec des cheveux blonds ébouriffés, portant des lunettes rondes et une chemise bleue ouverte sur un t-shirt noir.

Un jeune homme confiant avec une allure décontractée, des lunettes rondes, des cheveux blonds en bataille et un large sourire.

Cheveux blonds, lunettes rondes, sourire éclatant : il dégage une énergie chaleureuse et amicale.

Avec ses lunettes et son sourire sincère, cet homme semble chaleureux et accessible.

Ses cheveux blonds et son sourire lumineux reflètent une personnalité décontractée et joyeuse.

Un look bohème moderne avec des cheveux ébouriffés, des lunettes élégantes et une chemise ouverte.

Ce jeune homme porte une tenue simple mais soignée : chemise bleue, lunettes rondes et un t-shirt noir.

Lunettes rondes en métal, cheveux blonds naturels et une expression joviale illuminent son visage.

L’homme arbore un style décontracté avec un sourire radieux et une coupe de cheveux désinvolte.

Avec ses cheveux blonds et son sourire éclatant, il incarne la joie de vivre.

Une allure soignée mais décontractée : cheveux blonds légèrement longs, lunettes rondes, chemise ouverte.

Son style moderne se caractérise par des cheveux blonds ébouriffés et une chemise bleue sur un t-shirt noir.

Ce jeune homme porte des lunettes rondes qui ajoutent une touche de sophistication à son look.

Avec un sourire confiant et des lunettes rondes, il affiche une personnalité accueillante.

Un homme au look moderne, cheveux blonds en bataille et un style minimaliste mais élégant.

Ses lunettes rondes et son sourire éclatant reflètent une énergie positive et accessible.

Ses cheveux ébouriffés et son large sourire donnent une impression de spontanéité.

Avec des lunettes rondes et un sourire sincère, il projette un style simple mais charmant.

Un sourire éclatant, des cheveux blonds désordonnés et des lunettes pour compléter un style décontracté.

Son look bohème chic est accentué par ses lunettes rondes et son sourire chaleureux.

Cheveux blonds, lunettes rondes et chemise ouverte : une allure classique et détendue.

L’homme affiche un sourire radieux, avec une tenue décontractée et des cheveux coiffés naturellement.

Son expression joyeuse et ses cheveux blonds lui donnent une allure pleine de vie.

Avec des lunettes rondes et un large sourire, il semble à l’aise et confiant.

Ce jeune homme arbore un style intemporel avec une chemise bleue sur un t-shirt noir.

Un sourire éclatant illumine son visage, accompagné de lunettes rondes et de cheveux en bataille.

Ses lunettes et sa coiffure décontractée ajoutent du charme à son apparence.

Un look simple mais efficace : cheveux blonds naturels, lunettes rondes et un sourire rayonnant.

Avec ses cheveux ébouriffés et son sourire lumineux, il respire la confiance.

Ses lunettes rondes et sa chemise ouverte créent une tenue moderne et décontractée.

Ce jeune homme semble détendu, avec un sourire sincère et des cheveux coiffés de façon désordonnée.

Lunettes élégantes, cheveux blonds naturels et un sourire chaleureux composent son apparence.

Son style casual est accentué par des lunettes rondes et une coupe de cheveux naturelle.

Avec son large sourire et ses cheveux ébouriffés, il semble prêt à engager la conversation.

Ses lunettes rondes apportent une touche sophistiquée à son look décontracté.

Avec un sourire éclatant, des cheveux en bataille et des lunettes rondes, il est l’incarnation de la simplicité chic.

Ce jeune homme porte une chemise bleue qui souligne son style minimaliste et moderne.

Ses cheveux blonds coiffés au naturel ajoutent du caractère à son visage souriant.

Un sourire radieux illumine son visage, encadré par des lunettes rondes et des cheveux blonds.

Avec une chemise ouverte et un t-shirt noir, il affiche un style élégant mais simple avec un cou de taille moyenne et les cheuveu entre la couleur blonc et marron."""

#@spaces.GPU #[uncomment to use ZeroGPU]
def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):

    if randomize_seed:
        seed = random.randint(0, MAX_SEED)
        
    generator = torch.Generator().manual_seed(seed)
    
    # Ajout du system_prompt
    full_prompt = f"{system_prompt} {prompt}"
    
    image = pipe(
        prompt = full_prompt, 
        negative_prompt = negative_prompt,
        guidance_scale = guidance_scale, 
        num_inference_steps = num_inference_steps, 
        width = width, 
        height = height,
        generator = generator
    ).images[0] 
    
    return image, seed

examples = [
    "A Minecraft-style astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
    "An astronaut riding a green horse in Minecraft style",
    "A delicious ceviche cheesecake slice in Minecraft style",
]

css="""
#col-container {
    margin: 0 auto;
    max-width: 640px;
}
"""

with gr.Blocks(css=css) as demo:
    
    with gr.Column(elem_id="col-container"):
        gr.Markdown(f"""
        # Text-to-Image Gradio Template
        """)
        
        with gr.Row():
            
            prompt = gr.Text(
                label="Prompt",
                show_label=False,
                max_lines=1,
                placeholder="Enter your prompt",
                container=False,
            )
            
            run_button = gr.Button("Run", scale=0)
        
        result = gr.Image(label="Result", show_label=False)

        with gr.Accordion("Advanced Settings", open=False):
            
            negative_prompt = gr.Text(
                label="Negative prompt",
                max_lines=1,
                placeholder="Enter a negative prompt",
                visible=False,
            )
            
            seed = gr.Slider(
                label="Seed",
                minimum=0,
                maximum=MAX_SEED,
                step=1,
                value=0,
            )
            
            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
            
            with gr.Row():
                
                width = gr.Slider(
                    label="Width",
                    minimum=256,
                    maximum=MAX_IMAGE_SIZE,
                    step=32,
                    value=1024, #Replace with defaults that work for your model
                )
                
                height = gr.Slider(
                    label="Height",
                    minimum=256,
                    maximum=MAX_IMAGE_SIZE,
                    step=32,
                    value=1024, #Replace with defaults that work for your model
                )
            
            with gr.Row():
                
                guidance_scale = gr.Slider(
                    label="Guidance scale",
                    minimum=0.0,
                    maximum=10.0,
                    step=0.1,
                    value=0.0, #Replace with defaults that work for your model
                )
                
                num_inference_steps = gr.Slider(
                    label="Number of inference steps",
                    minimum=1,
                    maximum=50,
                    step=1,
                    value=2, #Replace with defaults that work for your model
                )
        
        gr.Examples(
            examples = examples,
            inputs = [prompt]
        )
    gr.on(
        triggers=[run_button.click, prompt.submit],
        fn = infer,
        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
        outputs = [result, seed]
    )

demo.queue().launch()