import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0")

clientTranslate = InferenceClient("Helsinki-NLP/opus-mt-vi-en")

def translate(text):
    res = clientTranslate.translation(text)
    return res.translation_text

def greet(prompt):
    text = translate(prompt)
    image = client.text_to_image(text)
    print(image)
    return image

demo = gr.Interface(
    fn=greet,
    inputs=[
        gr.Textbox(label="prompt")
    ],
    outputs=[gr.Image()],
    api_name="text-to-image",
    clear_btn=None
)

if __name__ == "__main__":
    demo.launch()