app.py

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


api_url = "https://codinglehrer.pythonanywhere.com/generate_image"

def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampler="k_dpmpp_2s_a", steps=50):
    data = {
        "prompt": prompt,
        "seed": seed,
        "negative_prompt": negative_prompt,
        "model": model,
        "sampler": sampler,
        "steps": steps
    }
    response = requests.post(api_url, data=data)
    if response.status_code == 200:
        # Decode the base64 image string
        img_bytes = base64.b64decode(response.json()["url"])
        
        # Open the image using Pillow
        img = Image.open(io.BytesIO(img_bytes))
        
        # Return the image object
        return img
    else:
        return None

inputs = [
    gr.inputs.Textbox(label="Prompt"),
    gr.inputs.Number(label="Seed", default=0),
    gr.inputs.Textbox(label="Negative Prompt", default=""),
    gr.inputs.Dropdown(['3DKX', 'Abyss OrangeMix', 'AbyssOrangeMix-AfterDark', 'ACertainThing',
    'AIO Pixel Art', 'Analog Diffusion', 'Anime Pencil Diffusion', 'Anygen',
    'Anything Diffusion', 'Anything v3', 'anything_v4_inpainting',
    'App Icon Diffusion', 'Arcane Diffusion', 'Archer Diffusion',
    'Asim Simpsons', 'A to Zovya RPG', 'Balloon Art', 'Borderlands', 'BPModel',
    'BubblyDubbly', 'Char', 'CharHelper', 'Cheese Daddys Landscape Mix',
    'ChilloutMix', 'ChromaV5', 'Classic Animation Diffusion', 'Clazy',
    'Colorful', 'Coloring Book', 'Comic-Diffusion', 'Concept Sheet',
    'Counterfeit', 'Cyberpunk Anime Diffusion', 'CyriousMix',
    'Dan Mumford Style', 'Darkest Diffusion', 'Dark Victorian Diffusion',
    'Deliberate', 'DGSpitzer Art Diffusion', 'Disco Elysium', 'DnD Item',
    'Double Exposure Diffusion', 'Dreamlike Diffusion',
    'dreamlike_diffusion_inpainting', 'Dreamlike Photoreal',
    'DreamLikeSamKuvshinov', 'Dreamshaper', 'DucHaiten',
    'DucHaiten Classic Anime', 'Dungeons and Diffusion', 'Dungeons n Waifus',
    'Eimis Anime Diffusion', 'Elden Ring Diffusion', "Elldreth's Lucid Mix",
    'Elldreths Retro Mix', 'Epic Diffusion', 'Eternos', 'Experience',
    'ExpMix Line', 'FaeTastic', 'Fantasy Card Diffusion', 'FKing SciFi',
    'Funko Diffusion', 'Furry Epoch', 'Future Diffusion', 'Ghibli Diffusion',
    'GorynichMix', 'Grapefruit Hentai', 'Graphic-Art',
    'GTA5 Artwork Diffusion', 'GuoFeng', 'Guohua Diffusion', 'HASDX',
    'Hassanblend', "Healy's Anime Blend", 'Hentai Diffusion', 'HRL', 'iCoMix',
    'Illuminati Diffusion', 'Inkpunk Diffusion', 'Jim Eidomode',
    'JWST Deep Space Diffusion', 'Kenshi', 'Knollingcase', 'Korestyle',
    'kurzgesagt', 'Laolei New Berry Protogen Mix', "Lawlas's yiff mix",
    'Liberty', 'Marvel Diffusion', 'Mega Merge Diffusion', 'Microcasing',
    'Microchars', 'Microcritters', 'Microscopic', 'Microworlds',
    'Midjourney Diffusion', 'Midjourney PaintArt', 'Min Illust Background',
    'ModernArt Diffusion', 'mo-di-diffusion', 'Moedel', 'MoistMix',
    'Movie Diffusion', 'NeverEnding Dream', 'Nitro Diffusion', 'Openniji',
    'OrbAI', 'Papercutcraft', 'Papercut Diffusion', 'Pastel Mix',
    'Perfect World', 'PFG', 'PIXHELL', 'Poison', 'Pokemon3D', 'PortraitPlus',
    'PPP', 'Pretty 2.5D', 'PRMJ', 'Project Unreal Engine 5', 'ProtoGen',
    'Protogen Anime', 'Protogen Infinity', 'Pulp Vector Art', 'PVC',
    'Rachel Walker Watercolors', 'Rainbowpatch', 'Ranma Diffusion',
    'RCNZ Dumb Monkey', 'RCNZ Gorilla With A Brick', 'RealBiter',
    'Realism Engine', 'Realistic Vision', 'Redshift Diffusion', 'Rev Animated',
    'Robo-Diffusion', 'Rodent Diffusion', 'RPG', 'Samdoesarts Ultmerge',
    'Sci-Fi Diffusion', 'SD-Silicon', 'Seek.art MEGA', 'Smoke Diffusion',
    'Something', 'Sonic Diffusion', 'Spider-Verse Diffusion',
    'Squishmallow Diffusion', 'stable_diffusion', 'stable_diffusion_2.1',
    'stable_diffusion_2_inpainting', 'Supermarionation', 'Sygil-Dev Diffusion',
    'Synthwave', 'SynthwavePunk', 'TrexMix', 'trinart', 'Trinart Characters',
    'Tron Legacy Diffusion', 'T-Shirt Diffusion', 'T-Shirt Print Designs',
    'Uhmami', 'Ultraskin', 'UMI Olympus', 'Unstable Ink Dream', 'URPM',
    'Valorant Diffusion', 'Van Gogh Diffusion', 'Vector Art', 'vectorartz',
    'Vintedois Diffusion', 'VinteProtogenMix', 'Vivid Watercolors',
    'Voxel Art Diffusion', 'waifu_diffusion', 'Wavyfusion', 'Woop-Woop Photo',
    'Xynthii-Diffusion', 'Yiffy', 'Zack3D', 'Zeipher Female Model',
    'Zelda BOTW'], label="Model", default="Deliberate"),
    gr.inputs.Dropdown(["k_lms", "k_heun", "k_euler", "k_euler_a", "k_dpm_2", "k_dpm_2_a", "DDIM", "k_dpm_fast", "k_dpm_adaptive", "k_dpmpp_2m", "k_dpmpp_2s_a", "k_dpmpp_sde"], label="Sampler", default="k_dpmpp_2s_a"),
    gr.inputs.Number(label="Steps", default=50)
]

outputs = gr.outputs.Image(label="Generated Image", type="filepath")

interface = gr.Interface(generate_image, inputs, outputs, title="<a href='https://huggingface.co/spaces/ImagineAI-Real/ImagineAI-Image-Generator'>ImagineAI-Real/ImagineAI Image Generator</a>", 
                         description="Enter a prompt, click Submit and wait a bit for your image.<br/>If taking too long, duplicate the space.<br>Like this if you like it :)", 
                         examples=[["A cat sitting on a chair"], 
                                   ["A landscape with mountains and a lake"], 
                                   ["A portrait of a woman with red hair"]])

interface.launch()