added check

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from dotenv import load_dotenv
+from transformers import pipeline
 
 # Load environment variables from the .env file de forma local
 load_dotenv()
@@ -14,12 +15,26 @@ import openai
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
 
-def get_image(text: str):
+
+
+es_en_translator = pipeline("translation",model = "Helsinki-NLP/opus-mt-es-en")
+def translate_text(text):
+    text = es_en_translator(text)[0].get("translation_text")  #type: ignore
+    return text
+
+
+def get_image(text: str,translate: bool):
+    print(text)
+    if(translate):
+        text = translate_text(text)
     response = openai.Image.create(prompt=text, n=1, size="512x512") #,response_format="b64_json"
-    print(response['data'][0]['url']) # type: ignore
+    print(text)
     return response['data'][0]['url']# type: ignore
 
 
+
+
+
 with gr.Blocks() as demo:
     gr.Markdown(
         """
@@ -42,7 +57,13 @@ with gr.Blocks() as demo:
     )
     with gr.Row():
         with gr.Column():
-            gr.Markdown("Primero debes ingresar el texto para generar la imagen:")
+           
+            with gr.Row():
+                gr.Markdown("El modelo funciona mejor con entradas en inglés, podés elegir traducir al inglés con este check:")
+                check = gr.Checkbox(label="Traducir al inglés")
+            with gr.Row():
+                gr.Markdown("Primero debes ingresar el texto para generar la imagen:")
+
             with gr.Row():
                 with gr.Column(scale=4):
                     prompt = gr.Textbox(
@@ -62,7 +83,7 @@ with gr.Blocks() as demo:
 
     btn.click(
         fn=get_image,
-        inputs=prompt,
+        inputs=[prompt,check],
         outputs=[output],
     )  # steps,guidance,width,height]