import gradio as gr
import os 
import sys
from pathlib import Path
       
models = [
    "Yntec/OpenLexica",
    "Yntec/MapleSyrup",
]
current_model = models[0]
text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
models2=[
    gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False),
    gr.Interface.load(f"models/{models[1]}",live=True,preprocess=False),
]
   
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("""
     
              
              
              
Top SDXLModels and 401 SD1.5 models for your enjoyment!
              
              
The first time you load a model it takes 200 seconds
              But after it loads each image takes 20 seconds to generate!
             
                                                            """)
    with gr.Row():
        with gr.Column(scale=100):
            #Model selection dropdown    
            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(scale=100):
            magic1=gr.Textbox(label="Your Prompt", lines=4)
  #align-items: center !important;
  #appearance: none !important;
  
  
  #border-style: none !important;
  #box-shadow: rgba(0, 0, 0, .2) 0 3px 5px -1px,rgba(0, 0, 0, .14) 0 6px 10px 0,rgba(0, 0, 0, .12) 0 1px 18px 0 !important;
  #box-sizing: border-box !important;
  
  #cursor: pointer !important;
  #display: inline-flex !important;
  #fill: currentcolor !important;
  #font-family: "Google Sans",Roboto,Arial,sans-serif !important;
  #font-size: 14px !important;
  #font-weight: 500 !important;
  #height: 48px !important;
  #justify-content: center !important;
  #letter-spacing: .25px !important;
  #line-height: normal !important;
  #max-width: 100% !important;
  #overflow: visible !important;
  #padding: 2px 24px !important;
  #position: relative !important;
  #text-align: center !important;
  #text-transform: none !important;
  #transition: box-shadow 280ms cubic-bezier(.4, 0, .2, 1),opacity 15ms linear 30ms,transform 270ms cubic-bezier(0, 0, .2, 1) 0ms !important;
  #user-select: none !important;
  #-webkit-user-select: none !important;
  #touch-action: manipulation !important;
  #width: auto !important;
  #will-change: transform,opacity !important;
  #z-index: 0 !important;
            gr.HTML("""""")
            run=gr.Button("Generate Image")
    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="Use this box to extend an idea automagically, by typing some words and clicking Extend 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)