app.py

import gradio as gr
import requests
from PIL import Image
import io
import os


API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
HF_TOKEN = os.getenv( "HF_TOKEN" )

def generate_image(prompt):
    payload = {"inputs": prompt}
    response = requests.post(API_URL, json=payload)
    image_bytes = response.content
    image = Image.open(io.BytesIO(image_bytes))
    return image

    
interface = gr.Interface(
    fn=generate_image, 
    inputs=gr.Textbox(lines=4, placeholder="Digite seu prompt..."), 
    outputs="image", 
   # title="crie ilustrações para o seu livro",
    description=" <center><img src='https://editoraitacaiunas.com.br/wp-content/uploads/2023/04/ilustrador-ia-assistente.png'></center><br/> Confira nossos exemplos de prompts. Clique em um deles aguarde carregar. Acesse nossa biblioteca de prompts para ilustração acessando <a href='https://editoraitacaiunas.com.br/repositorio-de-prompts-para-ilustracao-de-livros/'>Repositório de Prompts</a>.",

                
    examples=[["A small cabin on top of a snowy mountain in the style of Disney, artstation"],
              ["Positive prompt: Elsa, d & d, fantasy, intricate, elegant, highly detailed, digital painting, artstation, concept art, matte, sharp focus, illustration, hearthstone, art by artgerm and greg rutkowski and alphonse mucha, 8k. Negative prompt: illustration, 3d, painting, cartoons, sketch, (worst quality:2), (low quality:2), (normal quality:2), lowres, bad anatomy, bad hands, normal quality, ((monochrome)), ((grayscale:1.2)),newhalf, collapsed eyeshadow, multiple eyebrows, pink hair, analog, analogphoto, multiple bodies, multiple hands, multiple heads"],
              ["A high tech solarpunk utopia in the Amazon rainforest"], 
              ["Closeup face portrait of a black girl wearing crown of flowers, smooth soft skin, big dreamy eyes, beautiful intricate colored hair, symmetrical, anime wide eyes, soft lighting, detailed face, by makoto shinkai, stanley artgerm lau, wlop, rossdraws, concept art, digital painting, looking into camera"],
              ["Pixar style little girl, 4k, 8k, unreal engine, octane render photorealistic by cosmicwonder, hdr, photography by cosmicwonder, high definition, symmetrical eyes, volumetric lighting, dusty haze, photo, octane render, 24mm, 4k, 24mm, DSLR, high quality, 60 fps, ultra realistic"], 
              ["a highly detailed matte painting of a man on a hill watching a rocket launch in the distance by studio ghibli, makoto shinkai, by artgerm, by wlop, by greg rutkowski, volumetric lighting, octane render, 4 k resolution, trending on artstation, masterpiece "]]
)

interface.launch()