import gradio as gr import os import sys from pathlib import Path import random import string text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion") proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0") proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0") def add_random_noise(prompt): noise = "".join(random.choices(string.ascii_letters + string.punctuation, k=random.randint(1, 10))) return prompt + noise def send_it1(inputs,proc1=proc1): prompt_with_noise = add_random_noise(inputs) output1=proc1(prompt_with_noise) return output1 def send_it2(inputs,proc2=proc2): prompt_with_noise = add_random_noise(inputs) output2=proc2(prompt_with_noise) return output2 def send_it3(inputs,proc3=proc3): prompt_with_noise = add_random_noise(inputs) output3=proc3(prompt_with_noise) return output3 def send_it4(inputs,proc4=proc4): prompt_with_noise = add_random_noise(inputs) output4=proc4(prompt_with_noise) return output4 def send_it5(inputs,proc5=proc5): prompt_with_noise = add_random_noise(inputs) output5=proc5(prompt_with_noise) return output5 def send_it6(inputs,proc6=proc6): prompt_with_noise = add_random_noise(inputs) output6=proc6(prompt_with_noise) return output6 def send_it7(inputs,proc7=proc7): prompt_with_noise = add_random_noise(inputs) output7=proc7(prompt_with_noise) return output7 def send_it8(inputs,proc8=proc8): prompt_with_noise = add_random_noise(inputs) output8=proc8(prompt_with_noise) return output8 def get_prompts(prompt_text): return text_gen(prompt_text) with gr.Blocks() as myface: with gr.Row(): input_text=gr.Textbox(label="Short Prompt") see_prompts=gr.Button("Magic Prompt") with gr.Row(): prompt=gr.Textbox(label="Prettified Prompt") run=gr.Button("Launch") with gr.Row(): output1=gr.Image(label="DreamlikeArt PhotoReal 1") output2=gr.Image(label="DreamlikeArt PhotoReal 2") with gr.Row(): output3=gr.Image(label="DreamlikeArt PhotoReal 3") output4=gr.Image(label="DreamlikeArt PhotoReal 4") with gr.Row(): output5=gr.Image(label="DreamlikeArt Diffusion 1") output6=gr.Image(label="DreamlikeArt Diffusion 2") with gr.Row(): output7=gr.Image(label="DreamlikeArt Diffusion 3") output8=gr.Image(label="DreamlikeArt Diffusion 4") run.click(send_it1, inputs=[prompt], outputs=[output1]) run.click(send_it2, inputs=[prompt], outputs=[output2]) run.click(send_it3, inputs=[prompt], outputs=[output3]) run.click(send_it4, inputs=[prompt], outputs=[output4]) run.click(send_it5, inputs=[prompt], outputs=[output5]) run.click(send_it6, inputs=[prompt], outputs=[output6]) run.click(send_it7, inputs=[prompt], outputs=[output7]) run.click(send_it8, inputs=[prompt], outputs=[output8]) see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt]) myface.queue(concurrency_count=2) myface.launch(enable_queue=True, inline=True)