import gradio as gr
import os 
import sys
from pathlib import Path
       
models = [
    "Yntec/Lyriel",
    "Yntec/3DKX",
    "Yntec/Dreamshaper8",

]
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),
    gr.Interface.load(f"models/{models[2]}",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=""""""
title = r"""
<p style='text-align: center;font-size: large'>Xmodel + Magic Prompt </p>
"""

article = r"""
<br><br><br><br><br><br><br><br><br><br><br><br>
"""
with gr.Blocks(css=css) as myface:
    # Добавление title
    gr.HTML(title)
    with gr.Row():
        with gr.Tab("Рисовать"):
                gr.HTML(""" """)

        with gr.Tab("Как использовать"):
            gr.HTML(f"""<title>Как использовать</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
                <h4>1. </h4>
                <h4>2. </h4>
                <h4>3. </h4>
                <h4>4. </h4>
                <h4>5. </h4></div>
                """)


    with gr.Row():
        with gr.Column(scale=50):
            input_text=gr.Textbox(label="Идея изображения:",lines=2)
            use_short=gr.Button("Активировать идею")
            see_prompts=gr.Button("Генерировать Magic промт")
    
    with gr.Row():
        with gr.Column(scale=100):
            #Model selection dropdown    
            model_name1 = gr.Dropdown(label="Модель", 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="Magic промт：", lines=4, placeholder = "На английском")
            gr.HTML("""<style>           .gr-button {
            color: white !important;
            border-color: #000000 !important;
            background: #499afc !important;
            }</style>""")
            run=gr.Button("Создать изображение")





            
    with gr.Row():
        with gr.Column(style="width=800px"):
            output1=gr.Image(label=(f"{current_model}"))
                
            

                
    # Добавление article
    gr.HTML(article)    
    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)