File size: 4,725 Bytes
edc20ce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8c56a9e
 
edc20ce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import gradio as gr
import os 
import sys
from pathlib import Path

def load_models_from_file(filename):
    with open(filename, 'r') as f:
        return [line.strip() for line in f]

if __name__ == "__main__":
    models = load_models_from_file('models.txt')
    print(models)
    #removed to removed.txt
    
current_model = models[0]

text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
models2 = [gr.Interface.load(f"models/{model}", live=True, preprocess=False) for model in models]


   
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("""<!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset="utf-8" />
    <meta name="twitter:card" content="player"/>
    <meta name="twitter:site" content=""/>
    <meta name="twitter:player" content="https://omnibus-maximum-multiplier-places.hf.space"/>
    <meta name="twitter:player:stream" content="https://omnibus-maximum-multiplier-places.hf.space"/>
    <meta name="twitter:player:width" content="100%"/>
    <meta name="twitter:player:height" content="600"/>    
    <meta property="og:title" content="Embedded Live Viewer"/>
    <meta property="og:description" content="Tweet Genie - A Huggingface Space"/>
    <meta property="og:image" content="https://cdn.glitch.global/80dbe92e-ce75-44af-84d5-74a2e21e9e55/omnicard.png?v=1676772531627"/>
    <!--<meta http-equiv="refresh" content="0; url=https://huggingface.co/spaces/corbt/tweet-genie">-->
  </head>
</html>
""")
    with gr.Row():
        with gr.Tab("Title"):
                gr.HTML("""<title>Prompt to Generate Image</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
                <h1>Enter a Prompt in Textbox then click Generate Image</h1>""")

        with gr.Tab("Tools"):
                    with gr.Tab("View"):
                      with gr.Row():
                        with gr.Column(style="width=50%, height=70%"):
                                gr.Pil(label="Crop")
                        with gr.Column(style="width=50%, height=70%"):
                                gr.Pil(label="Crop")
                            
                    with gr.Tab("Draw"):
                        with gr.Column(style="width=50%, height=70%"):
                                gr.Pil(label="Crop")
                        with gr.Column(style="width=50%, height=70%"):
                                gr.Pil(label="Draw")
                                gr.ImagePaint(label="Draw")
                                    
                    with gr.Tab("Text"):
                        with gr.Row():
                            with gr.Column(scale=50):
                                gr.Textbox(label="", lines=8, interactive=True)            
                            with gr.Column(scale=50):
                                gr.Textbox(label="", lines=8, interactive=True)

                    with gr.Tab("Color Picker"):
                        with gr.Row():
                            with gr.Column(scale=50):
                                gr.ColorPicker(label="Color", interactive=True)            
                            with gr.Column(scale=50):
                                gr.ImagePaint(label="Draw", interactive=True)      
    with gr.Row():
        with gr.Column(scale=100):
            magic1=gr.Textbox(lines=4)
            run=gr.Button("Generate Image")
    with gr.Row():
        with gr.Column(scale=100):
            model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True)
    with gr.Row():
        with gr.Column(style="width=800px"):
            output1=gr.Image(label=(f"{current_model}"))
                
    with gr.Row():
        with gr.Column(scale=50):
            input_text=gr.Textbox(label="Prompt Idea",lines=2)
            use_short=gr.Button("Use Short Prompt")
            see_prompts=gr.Button("Extend Idea")
                
    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)