File size: 5,989 Bytes
883428d
 
 
 
 
 
 
 
 
 
be17911
883428d
 
 
9b54e29
fe56be9
05dab56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
883428d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
bf862b4
883428d
deb4898
883428d
4296384
883428d
 
301b0e2
883428d
 
 
 
 
 
 
 
888c0bf
 
883428d
 
 
b65c666
 
 
883428d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
cb81c80
 
883428d
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import gradio as gr
import os
import sys
from pathlib import Path
import numpy as np
from PIL import Image



models = [
    {"name": "❤ STABLE DIFFUSION MODELS ==========", "url": "stabilityai/stable-diffusion-2-1"},
     {"name": "SD ComVis 1.2","url": "CompVis/stable-diffusion-v1-2"},
    {"name": "SD Comvis 1.4","url": "CompVis/stable-diffusion-v1-4"},
    {"name": "SD runawayml 1.5","url": "runwayml/stable-diffusion-v1-5"},
    {"name": "SD stable-diffusion xl base 1.0","url": "stabilityai/stable-diffusion-xl-base-1.0"},
     {"name": "SD Dreamshaper-Anime","url": "Lykon/DreamShaper"},
     {"name": "Dreamlike Anime","url": "dreamlike-art/dreamlike-photoreal-2.0"},
    {"name": "❤ REALISTIC PHOTO MODELS ==========", "url": "dreamlike-art/dreamlike-photoreal-2.0"},
    {"name": "AmiIReal", "url": "stablediffusionapi/amireal"},
    {"name": "Analog Diffusion", "url": "wavymulder/Analog-Diffusion"},
    {"name": "Circulus 2.8", "url": "circulus/sd-photoreal-v2.8"},
    {"name": "UltraSkin", "url": "VegaKH/Ultraskin"},
    {"name": "Wavyfusion", "url": "wavymulder/wavyfusion"},
    {"name": "❤ SEMI-REALISTIC MODELS ==========", "url": "stablediffusionapi/all-526"},    
    {"name": "All 526", "url": "stablediffusionapi/all-526"},
    {"name": "All 526 animated", "url": "stablediffusionapi/all-526-animated"},
    {"name": "Circulus Semi Real 2", "url": "circulus/sd-photoreal-semi-v2"},
    {"name": "Semi Real Mix", "url": "robotjung/SemiRealMix"},
    {"name": "SpyBG", "url": "stablediffusionapi/spybg"},
    {"name": "❤ STABLE DIFFUSION MODELS ==========", "url": "stabilityai/stable-diffusion-2-1"},
    {"name": "Stable Diffusion 1.4","url": "CompVis/stable-diffusion-v1-4"},
    {"name": "Stable Diffusion 1.5","url": "runwayml/stable-diffusion-v1-5"},
    {"name": "Stable Diffusion 2.1","url": "stabilityai/stable-diffusion-2-1"},
    {"name": "Stable Diffusion 2.1 Base","url": "stabilityai/stable-diffusion-2-1-base"},
    {"name": "Stable Diffusion 2.1 Unclip","url": "stabilityai/stable-diffusion-2-1-unclip"},
    {"name": "❤ SCI FI MODELS ==========", "url": "nitrosocke/Future-Diffusion"},
    {"name": "Future Diffusion", "url": "nitrosocke/Future-Diffusion"},
    {"name": "JWST Deep Space Diffusion", "url": "dallinmackay/JWST-Deep-Space-diffusion"},
    {"name": "Robo Diffusion 3 Base", "url": "nousr/robo-diffusion-2-base"},
    {"name": "Robo Diffusion", "url": "nousr/robo-diffusion"},
    {"name": "Tron Legacy Diffusion", "url": "dallinmackay/Tron-Legacy-diffusion"},
    {"name": "❤ 3D ART MODELS ==========", "url": "DucHaiten/DucHaitenAIart"},
    {"name": "DucHaiten Art", "url": "DucHaiten/DucHaitenAIart"},
    {"name": "DucHaiten ClassicAnime", "url": "DucHaiten/DH_ClassicAnime"},
    {"name": "DucHaiten DreamWorld", "url": "DucHaiten/DucHaitenDreamWorld"},
    {"name": "DucHaiten Journey", "url": "DucHaiten/DucHaitenJourney"},
    {"name": "DucHaiten StyleLikeMe", "url": "DucHaiten/DucHaiten-StyleLikeMe"},
    {"name": "DucHaiten SuperCute", "url": "DucHaiten/DucHaitenSuperCute"},
    {"name": "Redshift Diffusion 768", "url": "nitrosocke/redshift-diffusion-768"},
    {"name": "Redshift Diffusion", "url": "nitrosocke/redshift-diffusion"},
    
    ]

current_model = models[0]

text_gen = gr.Interface.load("spaces/daspartho/prompt-extend") 

models2 = []
for model in models:
    model_url = f"models/{model['url']}"
    loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
    models2.append(loaded_model)


def text_it(inputs, text_gen=text_gen):
    return text_gen(inputs)


def set_model(current_model_index):
    global current_model
    current_model = models[current_model_index]
    return gr.update(value=f"{current_model['name']}")


def send_it(inputs, model_choice):
    proc = models2[model_choice]
    return proc(inputs)


with gr.Blocks (css ='main.css')  as myface:
    
    gr.HTML(" <div style='font-size: 20px; font-family:verdana; background-color: #9579d7; color:#FFFFFFF; border:1px solid #FFFFFF; border-radius: 10px; width:40%; width:40%; height: 30px; float: left; text-align:center;'> Your Promt Here</div> <div style='font-size: 20px; font-family:verdana; background-color:#9579d7; color:#FFFFFFF; border:1px solid #FFFFFF; border-radius: 10px; width:40%; height: 30px; float:right;text-align: center;'> Choose model here </div> " )
    with gr.Row():
        input_text = gr.Textbox(label=" ",placeholder="1.PROMPT IDEA HERE ! ",lines=4)
        # Model selection dropdown
        model_name1 = gr.Dropdown(
        label="2. Choose Model",
        choices=[m["name"] for m in models],
        type="index",
        value=current_model["name"],
        interactive=True,
    
        
       )
    with gr.Row():
        see_prompts = gr.Button(" 3. GENERATE YOUR PROMT IDEA HERE!")                                                      
        run = gr.Button("4. GENERATE THE IMAGE HERE!", varant="primery")
        
 #
    with gr.Row():
        output1 = gr.Image(label="Generated Image")
        output2 = gr.Image(label="Generated Image")
        output3 = gr.Image(label="Generated Image")
    with gr.Row():          
        magic1 = gr.Textbox(label="Generated Prompt", lines=2)
        magic2 = gr.Textbox(label="Generated Prompt", lines=2)
        magic3 = gr.Textbox(label="Generated Prompt", lines=2)
   
    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3,])

    run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
    run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
    run.click(send_it, inputs=[magic3, model_name1], outputs=[output3])
   

    see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
    see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
    see_prompts.click(text_it, inputs=[input_text], outputs=[magic3])
   
title="Daylight (SD) ",
article="",
myface.queue(concurrency_count=200)
myface.launch(inline=True, max_threads=400)