import gradio as gr
import os 
import sys
from pathlib import Path
import time
import random
from PIL import Image


models =[
    "",
    "CompVis/stable-diffusion-v1-4",
    "runwayml/stable-diffusion-v1-5",
    "prompthero/openjourney",
#4    
    "stabilityai/stable-diffusion-2-1",
    "stabilityai/stable-diffusion-2-1-base",
    "andite/anything-v4.0",

    "Linaqruf/anything-v3.0",
    "eimiss/EimisAnimeDiffusion_1.0v",
    "nitrosocke/Nitro-Diffusion",
#10
    "wavymulder/portraitplus",
    "22h/vintedois-diffusion-v0-1",
    "dreamlike-art/dreamlike-photoreal-2.0",
#11
    "dreamlike-art/dreamlike-diffusion-1.0",
    "wavymulder/Analog-Diffusion",
    "nitrosocke/redshift-diffusion",
    "claudfuen/photorealistic-fuen-v1",
    "prompthero/openjourney-v2",
    "johnslegers/epic-diffusion",
    "nitrosocke/Arcane-Diffusion",
    "darkstorm2150/Protogen_x5.8_Official_Release",

]

model_1=models[1]
proc1=gr.Interface.load(f"models/{model_1}",live=False,preprocess=True, postprocess=False)



#from transformers import pipeline

#pipe = pipeline("translation", model="t5-base")

def translate2(text):

    return text[::-1]
def translate(text):
    print("vvv")
    #img1.update(proc1("girl"))
    #german.update("soijfoijf")
    img0=proc1("girl")
    return img0#pipe(text)[0]["translation_text"]


with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            english = gr.Textbox(label="English text")
            translate_btn = gr.Button(value="btn1")
            translate_btn2 = gr.Button(value="btn2")
        with gr.Column():
            german = gr.Textbox(label="German Text")
    with gr.Row():
        img1=gr.Image()
        img2=gr.Image()
        img3=gr.Image()
        
    translate_btn.click(translate, inputs=english, outputs=img1, api_name="translate-to-german")
    translate_btn2.click(translate2,inputs=english,outputs=german)
    examples = gr.Examples(examples=["I went to the supermarket yesterday.", "Helen is a good swimmer."],
                           inputs=[english])

print("abc01")
demo.launch()