import gradio as gr
import os 
import sys
from pathlib import Path

text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")

proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)

def send_it1(inputs, proc=proc1):
        output1=proc(inputs)
        return(output1)
def send_it2(inputs, proc=proc2):
        output2=proc(inputs)
        return(output2)
def send_it3(inputs, proc=proc3):
        output3=proc(inputs)
        return(output3)
def send_it4(inputs, proc=proc4):
        output4=proc(inputs)
        return(output4)
def send_it5(inputs, proc=proc5):
        output5=proc(inputs)
        return(output5)
def send_it6(inputs, proc=proc6):
        output6=proc(inputs)
        return(output6)
def send_it7(inputs, proc=proc7):
        output7=proc(inputs)
        return(output7)
def send_it8(inputs, proc=proc8):
        output8=proc(inputs)
        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")
        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")
        output7=gr.Image(label="DreamlikeArt Diffusion 3")
        output8=gr.Image(label="DreamlikeArt Diffusion 4")
    
    see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
    
    run.click(send_it1, inputs=[prompt], outputs=[output1], extra_args=[proc1])
    run.click(send_it2, inputs=[prompt], outputs=[output2], extra_args=[proc2])
    run.click(send_it3, inputs=[prompt], outputs=[output3], extra_args=[proc3])
    run.click(send_it4, inputs=[prompt], outputs=[output4], extra_args=[proc4])
    run.click(send_it5, inputs=[prompt], outputs=[output5], extra_args=[proc5])
    run.click(send_it6, inputs=[prompt], outputs=[output6], extra_args=[proc6])
    run.click(send_it7, inputs=[prompt], outputs=[output7], extra_args=[proc7])
    run.click(send_it8, inputs=[prompt], outputs=[output8], extra_args=[proc8])

myface.queue(concurrency_count=8)
myface.launch(enable_queue=True, inline=True)