Spaces:
Running
Running
File size: 7,090 Bytes
f387fa4 450d2f3 f387fa4 |
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 124 125 126 127 128 129 130 131 132 |
import gradio as gr
import random
import requests
import time
import os
import argparse
from dotenv import load_dotenv
load_dotenv("config.txt")
from engine.generateColab import gpugen
from engine.upscaler import upscale_image
from engine.promptGenerator import prompting
css = """
#container{
margin: 0 auto;
max-width: 40rem;
}
#intro{
max-width: 100%;
text-align: center;
margin: 0 auto;
}
#generate_button {
color: white;
border-color: #007bff;
background: #007bff;
width: 200px;
height: 50px;
}
footer {
visibility: hidden
}
"""
with gr.Blocks(title="Rensor", css=css, theme="ehristoforu/RE_Theme") as webui:
with gr.Row():
with gr.Row(visible=False, variant="panel") as prompter:
with gr.Column(scale=1):
chatbot = gr.Textbox(show_label=False, interactive=False, max_lines=16, lines=14)
with gr.Row():
chat_text = gr.Textbox(show_label=False, placeholder="Enter short prompt", max_lines=2, lines=1, interactive=True, scale=20)
chat_submit = gr.Button(value="Prompt", scale=1)
chat_submit.click(fn=lambda x: gr.update(value="Prompting...", interactive=False), inputs=chat_submit, outputs=chat_submit).then(prompting, inputs=chat_text, outputs=chatbot).then(fn=lambda x: gr.update(value="Prompt", interactive=True), inputs=chat_submit, outputs=chat_submit)
with gr.Column(scale=3):
with gr.Row():
gallery = gr.Gallery(show_label=False, rows=2, columns=6, preview=True, value=["assets/favicon.png"])
work_time = gr.Markdown(visible=False)
with gr.Row():
prompt = gr.Textbox(show_label=False, placeholder="Your amazing prompt...", max_lines=3, lines=3, interactive=True, scale=18)
button = gr.Button(value="Generate", variant="primary", scale=1)
with gr.Row():
advenced = gr.Checkbox(label="Advanced inputs/settings", value=False, interactive=True)
prompter_change = gr.Checkbox(label="Prompter", value=False, interactive=True)
with gr.Row(visible=False, variant="panel") as settings_tab:
with gr.Column(scale=1):
with gr.Tab("Settings"):
with gr.Row(scale=10):
mode = gr.Radio(label="Mode", choices=["High Quality", "Fast", "Super-fast"], value="Fast", info="Relationship between generation speed and quality.", interactive=True, visible=True)
with gr.Row(scale=10):
width = gr.Slider(label="Width", maximum=2048, minimum=256, value=512, step=8, interactive=True, visible=True)
height = gr.Slider(label="Height", maximum=2048, minimum=256, value=512, step=8, interactive=True, visible=True)
with gr.Row(scale=10):
guidance = gr.Slider(label="Guidance Scale", maximum=20.0, minimum=0.0, value=8.0, step=0.1, interactive=True, visible=True)
with gr.Row(scale=10):
num_images = gr.Slider(label="Number of images", maximum=12, minimum=1, value=1, step=1, interactive=True, visible=True)
with gr.Row(scale=1):
upscale_button = gr.Image(label="🚀 Upload image to 2x upscale", sources="upload", type="numpy", show_download_button=False, interactive=True)
with gr.Tab("Init image"):
with gr.Row():
with gr.Column():
img2img_change = gr.Checkbox(label="Init Image", value=False, visible=True, interactive=True, scale=10)
i2i_strength = gr.Slider(label="Init Strength", minimum=0.01, maximum=2, step=0.01, value=0.70, interactive=False, visible=True)
init_image = gr.Image(label="Init image", type="pil", interactive=False, visible=True, scale=1)
with gr.Tab("Inpaint"):
with gr.Row():
with gr.Column():
inpaint_change = gr.Checkbox(label="Inpaint", value=False, visible=True, interactive=True, scale=4)
inpaint_strength = gr.Slider(label="Inpaint Strength", minimum=0.01, maximum=2, step=0.01, value=0.70, interactive=False, visible=True)
inpaint_image = gr.Image(label="Inpaint image", type="pil", interactive=False, visible=True, tool="sketch", scale=1)
button.click(fn=lambda x: gr.update(visible=False), inputs=work_time, outputs=work_time).then(fn=lambda x: gr.update(value="Generating...", variant="secondary", interactive=False), inputs=button, outputs=button).then(gpugen, inputs=[prompt, mode, guidance, width, height, num_images, i2i_strength, inpaint_strength, img2img_change, inpaint_change, init_image, inpaint_image], outputs=[gallery, work_time]).then(fn=lambda x: gr.update(visible=True), inputs=work_time, outputs=work_time).then(fn=lambda x: gr.update(value="Generate", variant="primary", interactive=True), inputs=button, outputs=button)
upscale_button.upload(fn=lambda x: gr.update(visible=False), inputs=work_time, outputs=work_time).then(fn=lambda x: gr.update(label="🖼️ Image uploaded to 2x upscale", interactive=False), inputs=upscale_button, outputs=upscale_button).then(fn=lambda x: gr.update(value="Upscaling...", variant="secondary", interactive=False), inputs=button, outputs=button).then(upscale_image, inputs=upscale_button, outputs=[gallery, work_time]).then(fn=lambda x: gr.update(label="🚀 Upload image to 2x upscale", interactive=True), inputs=upscale_button, outputs=upscale_button).then(fn=lambda x: gr.update(value="Generate", variant="primary", interactive=True), inputs=button, outputs=button).then(fn=lambda x: gr.update(visible=True), inputs=work_time, outputs=work_time)
advenced.change(fn=lambda x: gr.update(visible=x), inputs=advenced, outputs=settings_tab, queue=False, api_name=False)
prompter_change.change(fn=lambda x: gr.update(visible=x), inputs=prompter_change, outputs=prompter, queue=False, api_name=False)
img2img_change.change(
fn=lambda x: gr.update(interactive=x),
inputs=img2img_change,
outputs=init_image,
queue=False,
api_name=False,
).then(
fn=lambda x: gr.update(interactive=x),
inputs=img2img_change,
outputs=i2i_strength,
queue=False,
api_name=False,
)
inpaint_change.change(
fn=lambda x: gr.update(interactive=x),
inputs=inpaint_change,
outputs=inpaint_image,
queue=False,
api_name=False,
).then(
fn=lambda x: gr.update(interactive=x),
inputs=inpaint_change,
outputs=inpaint_strength,
queue=False,
api_name=False,
)
webui.queue(max_size=20).launch(debug=False, share=True, server_port=5555, quiet=True, show_api=False, favicon_path="assets/favicon.png", inbrowser=True) |