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=8)
myface.launch(enable_queue=True, inline=True)