I-am-Justin.py

import gradio as gr
import os 
import sys
from pathlib import Path
       
models = [
    "Yntec/photoMovieRealistic",
    "Yntec/a-ZovyaRPGV3VAE",
    "Yntec/a-ZoviaRPGArtistV2VAE",
    "Yntec/InsaneRealisticCVAE",
    "Yntec/Lunar",
    "Yntec/LunarLuma",
    "Yntec/QToriReloaded",
    "Yntec/Ninja-Diffusers",
    "Yntec/ChildrenStoriesAnime",
    "Yntec/DucHaitenAnime768",
    "Yntec/DucHaitenClassicAnime768",
    "Yntec/WesternAnimation",
    "Yntec/GOLDFish",
    "Yntec/Rainbowsphere",
    "Yntec/DreamAnything",
    "Yntec/yabalMixTrue25D_v2_VAE",
    "dreamlike-art/dreamlike-anime-1.0",
    "Yntec/DucHaiten-Retro-Diffusers",
    "Hemlok/QuinceMix",
    "Yntec/KIDSILLUSTRATIONS",
    "Yntec/COOLKIDSV2",
    "Yntec/Pavo-Mix-Diffusers",
    "Yntec/DucHaitenAIart-beta",
    "Ojimi/anime-kawai-diffusion",
    "LottePeisch/RevAnimated-Diffusers",
    "Linaqruf/animagine-xl",
    "Envvi/Inkpunk-Diffusion",
    "0xJustin/Dungeons-and-Diffusion",
    ######################################################
    #"digiplay/PersonaStyleCheckpoint",
    #"darkstorm2150/Protogen_Infinity_Official_Release",
    #"PeggyWang/openjourney-v2",
    #"darkstorm2150/Protogen_x3.4_Official_Release",
    #"stablediffusionapi/deliberateappfactory", #236
    #"digiplay/CrossoverMix_v2",
    #"stablediffusionapi/spybg",
    #"stablediffusionapi/dreamshaper-v6", #239
    #"stablediffusionapi/the-ally",
    #"darkstorm2150/Protogen_x5.8_Official_Release",
    #coreco/seek.art_MEGA",
    #"digiplay/BlankCanvas_v1", #07.11
    #"digiplay/OnlyAnime_v2.3",
    #"Korakoe/OpenNiji",
    #"digiplay/Photon_v1",
    #"digiplay/Pika_v2",
    #"digiplay/RealCartoon3D_F16full_v3.1", #254
    #"digiplay/realidefmix_3.5VAE",
    #"digiplay/realmixUnrealjourney_v1",
    #"digiplay/SyncMix_v1.5",
    #"digiplay/TWingshadow_v1.2",
    #"digiplay/V3_by_Hans_Asian",
    #"digiplay/whatamix_v1",
    #
    #"digiplay/2K", #216
    #"digiplay/AIGEN_v1.4_diffusers",
    #"digiplay/BrickAndMortarMix_v2.0_diffusers", #224
    #"digiplay/BeautyFool_v1.2VAE_pruned",
    #"digiplay/breakdomainrealistic_R2333",
 #   "digiplay/CCTV2.5d_v1", #219
#    "digiplay/ChikMix_V3", #253
#    "stablediffusionapi/chilledremixsazyou-r", #195
#    "digiplay/CityEdge_StyleMix_v1.44",
#    "stablediffusionapi/dalcefopainting2", #199
#    "digiplay/EdisonNilMix_v1", #07.10
#    "digiplay/DiamondCoalMix_v2_pruned_diffusers",
#    "digiplay/DreamShaper_7", #259
#    "digiplay/elegantEntropy_v1.1", #221
#    "digiplay/EtherRealMix_LUX2",
#    "digiplay/KawaiiRealisticAnimeMix_A0.3",
#    "digiplay/highQualityCGMIX_v1",
#    "digiplay/HIMAWARI_v1",
#    "digiplay/Hodgepodge_v2.1", #217
#    "digiplay/illustro1stEdition_illustroV1", #214
#    "digiplay/Juggernaut_final", #07.11
#    "digiplay/Landscape_PhotoReal_v1",
#    "digiplay/LuckyStrikeMix0.2Realistic", #07.10
#    "digiplay/Matrix_Stellar_VAE_v1",
#    "digiplay/PrefixRealisticMix_v1",
#    "digiplay/RealEpicMajicRevolution_v1", #07.11
   # "digiplay/ShampooMix_4", #252
  #  "digiplay/SoapMix2.5D_v1",
 #   "digiplay/ZemiHR_v2_diffusers",
#        
    #"Redamancy2299/dreambooth",
    #"Lykon/DreamShaper", #240
    #"trysem/DreamShaper-3.3",
    #"HusseinHE/hussein-deliberate-1000steps", #237
    #"stablediffusionapi/majicmixfantasy",
    #"stablediffusionapi/majicmixsombre", #247
    #"wavymulder/modelshoot",
    #"digiplay/ChillyMix_v1", #215
    #"stablediffusionapi/foto-assisted-diffusion", #197
    #"wavymulder/portraitplus",
    #"stablediffusionapi/chilloutmix-4264",
    #"stablediffusionapi/product-design", #194
   # "kandinsky-community/kandinsky-2-1", #251
   # 
    #"digiplay/2.5DSET_diffusers", #227
    #"digiplay/2-KWI", #213
    #"digiplay/alstroemeriaMix_v1",
    #"wavymulder/Analog-Diffusion",
    #"digiplay/AniRealityMix_v1", #257
    #"digiplay/ARRealVX1.1",
    #"digiplay/BadAnime_v1",
    #"digiplay/BasilKorea_v2", #07.11
    #"digiplay/bluePencilRealistic_v01",
    #"digiplay/bra_v40_diffusers",
    #"digiplay/Burger_Mix_semiR2Lite", #222
    #"digiplay/calicomixreal_v2.0_diffusers",
    #"digiplay/CampurSari_Gen1",
    #"digiplay/cocotifacute_v1", #07.10
    #"digiplay/cosfMix_v1", #223
    #"digiplay/CounterMix_v2", #211
    #"digiplay/CuriousMerge2.5D_v5",
    #"digiplay/dosmix",
    #"stablediffusionapi/droodlyrielv15",
    #"digiplay/fantexi_v0.7",
    #"digiplay/fishmix_other_v1",
    #"digiplay/FormCleansingMix_v1", #228
    #"digiplay/FumizukiMix_v1",
    #"digiplay/helloworld_v3",
    #"digiplay/HenmixArt_v1",
    #"digiplay/ISOmix_v3.22",
    #"digiplay/kencanmix_v2.0beta",
    #"wavymulder/lomo-diffusion",
    #"stablediffusionapi/majicmixv5", #192 
    #"digiplay/mecha_musume_vivid_soft",
    #"digiplay/MiracleMixGlitter_v1",
    #"digiplay/MixTape_RocknRoll_v3punk_bake_fp16",
    #"digiplay/NextPhoto_v1",
    #"digiplay/Noosphere_v3",
    #"digiplay/nk15_diffusers", #230
    #"digiplay/PeachMixsRelistic_R0", #262
   # "wavymulder/timeless-diffusion",
  #  "digiplay/WhiteDreamyHillMix_v1", #220
 #   "digiplay/ya3p_VAE", #258
#
    #"DucHaiten/DucHaitenAnime",
    #"DucHaiten/DucHaitenAIart",
    #"Manseo/Colorful-v4.5-Plus", #244
    #"Guizmus/SDArt_ChaosAndOrder",
    #"DucHaiten/DH_ClassicAnime",
    #"stablediffusionapi/disneypixar",
    #"johnslegers/epic-diffusion-v1.1",
    #"emilianJR/epiCRealism",
    #"johnslegers/epic-diffusion",
    #"digiplay/endlessMixRenatus_v1.1", #07.10
    #"digiplay/fantasticAnime_diffusers",
    #"stablediffusionapi/ghostmix",
    #"Duskfallcrew/EpicMix_Realism",
    #"nitrosocke/Nitro-Diffusion",
    #"prompthero/openjourney",
    #"Guizmus/SDArt_something",
   # "DucHaiten/DucHaiten-StyleLikeMe",
  #  "ddPn08/subtly", #250
 #   "22h/vintedois-diffusion-v0-1",
#
    #"circulus/sd-anireal-v2.7",
    #"0xJustin/Dungeons-and-Diffusion",
   # "Guizmus/SDArt_AliceInDiffusionLand",
  #  "stablediffusionapi/realistic-vision-v20-2047",
 #   "redstonehero/RPG-v5-itr17_A10T",
#
   # "stablediffusionapi/camelliamix25d",
  #  "Guizmus/SDArt_cosmichorrors",
 #   "DGSpitzer/DGSpitzer-Art-Diffusion",
#    "stablediffusionapi/emotion-puppeteer-v2",
    #"stablediffusionapi/fengjing",
   # "stablediffusionapi/fuwafuwamix",
  #  "Fred99774/girlnew1",
 #   "stablediffusionapi/majicmixrealistic",
#    "badmonk/nxka",
    #"ItsJayQz/SynthwavePunk-v2",
    #"zhyemmmm/ToonYou",
    #"stablediffusionapi/uber-realistic-merge",
    #"stablediffusionapi/vne732h9dh4",
    #"stablediffusionapi/wand-magic2",
    #"stablediffusionapi/waifu-journey-2",
    #"stablediffusionapi/zovya",
    
    #"Guizmus/SDArt_cosmichorrors768",
    #"stablediffusionapi/counterfeit-v30",
    #"stablediffusionapi/amireal",
    ##"JamesFlare/pastel-mix",    #"andite/pastel-mix",
    #"stablediffusionapi/rev-anim",
    #"aipicasso/picasso-diffusion-1-1",
    #"xiaolxl/Gf_style2",
    #"circulus/sd-semireal-v2.8",
    #"Crosstyan/BPModel", #07.11
    
    #"digiplay/Dusk-1",
    #"ogkalu/Comic-Diffusion",
    #"Guizmus/SDArt_ChaosAndOrder768",
    #"gsdf/Counterfeit-V2.0",
    #"dwancin/memoji", #07.11
    #"nousr/robo-diffusion-2-base",
        
    ###"hakurei/waifu-diffusion",
    #"WarriorMama777/AbyssOrangeMix2",
    #"stablediffusionapi/abyssorangemix2nsfw", #200
    #"cag/anything-v3-1",
    #"iZELX1/Anything-V3-X",
    #"xyn-ai/anything-v4.0", #"andite/anything-v4.0",
    #"D1b4l4p/AsianMix",
    #"Fred99774/chilloutvlara", 
    #"aipicasso/cool-japan-diffusion-2-1-2",
    #"stablediffusionapi/corneos-7th-heaven-m", #196
    #"DGSpitzer/Cyberpunk-Anime-Diffusion",
    #"stablediffusionapi/dark-sushi-mix",
    #"joachimsallstrom/Double-Exposure-Diffusion",
    #"eimiss/EimisAnimeDiffusion_1.0v",
    #"prompthero/funko-diffusion",
    #"nitrosocke/Ghibli-Diffusion",
    ###"iZELX1/Grapefruit",
    #"xiaolxl/GuoFeng3",
    #"stablediffusionapi/tmnd-mix",
    #"coder119/Vectorartz_Diffusion", #203
    
    #"WarriorMama777/AbyssOrangeMix",
    #"AIARTCHAN/7pa",
    #"JosephusCheung/ACertainModel",
    #"JosephusCheung/ACertainThing",
    #"AIARTCHAN/AbyssHellHero",
    #"JosephusCheung/ACertainty",
    #"AIARTCHAN/AbyssHellVer3",
    #"AIARTCHAN/AbyssMapleVer3",
    #"stablediffusionapi/abyssorangemixsfw",
    #"AIARTCHAN/anidosmixV2",
    #"stablediffusionapi/anime-model-v2",
    #"kubanemil/AnyLORA",
    #"stablediffusionapi/hc-anything-v3-vae", #231
    #"mm00/anything-v3.0-light",
    #"stablediffusionapi/anythingelse-v4",
    #"stablediffusionapi/anything-v45-fixed",
    #"stablediffusionapi/anything-v5",
    #"nitrosocke/Arcane-Diffusion",
    #"nitrosocke/archer-diffusion",
    #"stablediffusionapi/architecture-tuned-model",
    #"WarriorMama777/BloodOrangeMix",
    #"wavymulder/collage-diffusion",
    #"stablediffusionapi/camelliamixline",
    #"digiplay/chrysanthemumMix_v1",
    #"digiplay/CiderMix_ciderR", #260
    #"Johnhex/Clam", #243
    #"stablediffusionapi/cosmic-babes",
    #"digiplay/CoffeeDonut_v1",
    #"stablediffusionapi/dark-sushi-25d",
    #"digiplay/Defacta_v1_diffusers", #226
    ### "WarriorMama777/EerieOrangeMix",
    #"digiplay/DuelAnimeMix_v1", #225
    #"Envvi/Inkpunk-Diffusion",
    #"digiplay/kotosmix_diffusers", #229
    #"stablediffusionapi/meinaalter",
    #"Nacholmo/meinamixv7-diffusers",
    #"stablediffusionapi/meinapastel",
    #"AIARTCHAN/MIX-Pro-V4",
    #"Lykon/NeverEnding-Dream",
    #"stablediffusionapi/shirataki-mix", #191
    #"NoCrypt/SomethingV2_2",
    #"NoCrypt/SomethingV2",
    #"badmonk/sxzumi",
    ### "stablediffusionapi/three-delicacy",
    ### "stablediffusionapi/three-delicacy-wonto",
    #"etherealxx/systemy-csrmodel-cutesexyrobutts", #"andite/cutesexyrobutts-diffusion",
    #"sd-dreambooth-library/true-guweiz-style", # "andite/guweiz-diffusion",
    #"stablediffusionapi/vector-art", #198
    #"digiplay/xxMix_4",
    ###"mio/hiten", #"andite/hiten-diffusion",
    ### "andite/mashuu-diffusion",
    ### "andite/mignon-diffusion",
    #### "andite/mikapikazo-diffusion",
    #### "andite/piromizu-diffusion",
    #"digiplay/Zevinemix_v1.0/",

    #"digiplay/AnaMix_v2", #07.11
    #"stablediffusionapi/animetestmodelv3",
    #"yulet1de/anything", #232
    #"hakurei/artstation-diffusion", #07.11
    #"Fictiverse/Stable_Diffusion_BalloonArt_Model",
    #"stablediffusionapi/bg-dream-irl",
    #"stablediffusionapi/bg-dream-model-b", #193
    #"Rardilit/Ciffusion_v0.1",
    #"circulus/sd-anireal-2d-v2",
    #"circulus/sd-photoreal-v2.7",
    #"circulus/sd-photoreal-photo-v2",
    #"circulus/sd-anireal-2.5d-v2",
    #"circulus/sd-anireal-v2.5",
    #"circulus/sd-photoreal-semi-v2",
    #"circulus/sd-photoreal-real-v2",
    #"circulus/sd-photoreal-v2.5",
    #"circulus/sd-anireal-3d-v2",
    #"circulus/sd-anireal-v2.8",
    #"nitrosocke/classic-anim-diffusion", 
    #"Conflictx/Complex-Lineart", #245
    #"sayakpaul/da-vinci-sd-pokemon",
    #"nitrosocke/elden-ring-diffusion",
    #"digiplay/EtherBluMix_1", #07.11
    #"digiplay/fantasticmix_v40_test", #261
    #"theintuitiveye/FantasyMix",
    #"Fictiverse/Stable_Diffusion_FluidArt_Model",
    #"nitrosocke/Future-Diffusion",
    #"ItsJayQz/GTA5_Artwork_Diffusion",  #205
    #"digiplay/hellopure_v2.23",
    #"TheLastBen/hrrzg-style-768px", #246
    #"nevernotsean/IllustratedPaperMini", #242
    #"dallinmackay/JWST-Deep-Space-diffusion",
    #"prompthero/linkedin-diffusion",
    #"mann-e/mann-e_4_rev-0-1", #210
    #"ItsJayQz/Marvel_WhatIf_Diffusion", #206
    #"yuanbit/max-15-1e-6-1500",
    #"MyneFactory/MF-Base", #248
    #"Fictiverse/Stable_Diffusion_Microscopic_model", #249
    #"nitrosocke/mo-di-diffusion",
    #"luongphamit/NeverEnding-Dream2", #241
    #"lambdalabs/sd-naruto-diffusers", #201
    #"Vernon-2/output_test",
    #"Fictiverse/Stable_Diffusion_PaperCut_Model",
    ##"bsuutari/path_to_saved_model",
    #"bsuutari/path_to_saved_model_rafa",
    #"digiplay/PlanetBumix_v1",
    #"lambdalabs/sd-pokemon-diffusers", #202
    #"prompthero/poolsuite-diffusion",
    #"digiplay/RealismEngine_v1",
    #"nitrosocke/redshift-diffusion",
    #"nitrosocke/redshift-diffusion-768",
    #"nousr/robo-diffusion",
    #"digiplay/SDVN1-Real_v1", #255
    #"nitrosocke/spider-verse-diffusion",
    ##"runwayml/stable-diffusion-v1-5",
    #"nicky007/stable-diffusion-logo-fine-tuned",
    #"stablediffusionapi/three-delicacy", #233
    #"stablediffusionapi/three-delicacy-wonto", #234
    #"naclbit/trinart_stable_diffusion_v2",    
    #"dallinmackay/Tron-Legacy-diffusion",
    #"digiplay/unstableDiffusersYamerMIX_v3",
    #dallinmackay/Van-Gogh-diffusion",
    #"Fictiverse/Stable_Diffusion_VoxelArt_Model", #204
    #"wavymulder/wavyfusion",
    #"CompVis/stable-diffusion-v1-3", #207
    #"CompVis/stable-diffusion-v1-2", #208
    #"CompVis/stable-diffusion-v1-1", #209
]
current_model = models[0]

text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")

models2=[
    gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[1]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[2]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[3]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[4]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[5]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[6]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[7]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[8]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[9]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[10]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[11]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[12]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[13]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[14]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[15]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[16]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[17]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[18]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[19]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[20]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[21]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[22]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[23]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[24]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[25]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[26]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[27]}",live=True,preprocess=False),
]

   
def text_it1(inputs,text_gen1=text_gen1):
        go_t1=text_gen1(inputs)
        return(go_t1)

def set_model(current_model):
    current_model = models[current_model]
    return gr.update(label=(f"{current_model}"))


def send_it1(inputs, model_choice):
        proc1=models2[model_choice]
        output1=proc1(inputs)
        return(output1)
css=""""""


with gr.Blocks(css=css) as myface:
    gr.HTML()
    with gr.Row():
        with gr.Tab("说明"):
                gr.HTML("""    <title>多模型加速生成</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
                <h1>输入文字生成图像</h1>
                <br><h4>第一次导入模型可能需要200秒生成</h4>
                <br><h4>之后一般只需20秒左右</h4>""") 
        with gr.Tab("内置模型解释"):
                gr.HTML("""    <title>在内置模型中生成图片可能出现的问题</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
                <h4>1. 使用Envvi/Inkpunk-Diffusion时，需要在提示词中加上“nvinkpunk”</h4>""")
    with gr.Row():
        with gr.Column(scale=100):
            magic1=gr.Textbox(label="输入提示词：", lines=4, placeholder = "英文")
            run=gr.Button("生成图像")
    with gr.Row():
        with gr.Column(scale=100):  
            model_name1 = gr.Dropdown(label="选择模型", choices=[m for m in models], type="index", value=current_model, interactive=True)
    with gr.Row():
        with gr.Column(style="width=400px"):
            output1=gr.Image(label=(f"{current_model}"))
                
            
    with gr.Row():
        with gr.Column(scale=50):
            gr.HTML("""<h1>没有想法？</h1>""")  
            input_text=gr.Textbox(label="输入提示词点子：",lines=2, placeholder = "英文")
            use_short=gr.Button("使用精简提示词（使用框中提示词）")
            see_prompts=gr.Button("扩充框内提示词")
                
     
    def short_prompt(inputs):
        return(inputs)
    
    model_name1.change(set_model,inputs=model_name1,outputs=[output1])
    
    run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
    
    use_short.click(short_prompt,inputs=[input_text],outputs=magic1)
    
    see_prompts.click(text_it1,inputs=[input_text],outputs=magic1)
    
myface.queue(concurrency_count=200)
myface.launch(inline=True, show_api=False, max_threads=400)