File size: 4,725 Bytes
8ae0ff2
57a9c16
 
8ae0ff2
ab9256d
 
 
 
 
628051d
 
 
57a9c16
 
628051d
57a9c16
 
9f870a3
8ae0ff2
57a9c16
 
 
 
 
8ae0ff2
628051d
 
 
8ae0ff2
57a9c16
8ae0ff2
57a9c16
 
 
 
e4f2a60
628051d
 
57a9c16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
628051d
57a9c16
 
 
628051d
57a9c16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
628051d
57a9c16
 
 
 
 
 
e4f2a60
57a9c16
 
 
628051d
57a9c16
 
628051d
57a9c16
 
 
628051d
57a9c16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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)