Update app.py

app.py

@@ -57,14 +57,41 @@ def process_image(image):
     
     return description, recipe
 
+# Paso 5: Función para traducir la receta si el usuario lo desea
+def translate_recipe(recipe):
+    translator = pipeline("translation_en_to_es", model="Helsinki-NLP/opus-mt-en-es")
+    translated_recipe = translator(recipe)[0]['translation_text']
+    return translated_recipe
+
 # Paso 4: Crear la interfaz de Gradio
 iface = gr.Interface(
     fn=process_image,
     inputs=gr.Image(type="pil"),
     outputs=[gr.Textbox(label="Dish"), gr.Textbox(label="Recipe")],
     title="Recipe Generator from Dish Image",
-    description="Upload an image of a dish to get a description, ingredient list, and step-by-step recipe."
+    description="Upload an image of a dish to get a description, ingredient list, and step-by-step recipe.",
+    allow_flagging="never"
+)
+
+# Añadir botón para traducir la receta
+translate_button = gr.Button("Translate Recipe to Spanish")
+translate_output = gr.Textbox(label="Translated Recipe")
+
+def translate_interface(recipe):
+    translated_recipe = translate_recipe(recipe)
+    return translated_recipe
+
+translate_button.click(fn=translate_interface, inputs=iface.outputs[1], outputs=translate_output)
+
+# Crear una interfaz con el botón de traducción
+iface_with_translation = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="pil"),
+    outputs=[gr.Textbox(label="Dish"), gr.Textbox(label="Recipe"), translate_output],
+    title="Recipe Generator from Dish Image",
+    description="Upload an image of a dish to get a description, ingredient list, and step-by-step recipe. Optionally, translate the recipe to Spanish.",
+    allow_flagging="never"
 )
 
 # Lanzar la aplicación de Gradio
-iface.launch()
+iface_with_translation.launch()