Update app.py

app.py

@@ -49,13 +49,14 @@ def generate_recipe(description):
 
 # Paso 3: Define la función principal para procesar la imagen y generar la receta
 def process_image(image):
-    # Paso 3.1: Generar descripción del plato
+    # Paso 3.1: Generar descripción específica del plato de comida
     description = image_to_text(image)[0]['generated_text']
+    specific_description = f"The dish appears to contain the following elements: {description}. Please describe in detail the ingredients and their arrangement."
     
-    # Paso 3.2: Generar receta a partir de la descripción
-    recipe = generate_recipe(description)
+    # Paso 3.2: Generar receta a partir de la descripción específica
+    recipe = generate_recipe(specific_description)
     
-    return description, recipe
+    return specific_description, recipe
 
 # Paso 5: Función para traducir la receta si el usuario lo desea
 def translate_recipe(recipe):
@@ -69,28 +70,26 @@ iface = gr.Interface(
     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.",
-    allow_flagging="never"
+    description="Upload an image of a dish to get a detailed description, ingredient list, and step-by-step recipe.",
+    flagging_mode="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],
+    outputs=[
+        gr.Textbox(label="Dish"),
+        gr.Textbox(label="Recipe"),
+        gr.Button("Translate Recipe to Spanish").click(fn=translate_interface, inputs="Recipe", outputs=gr.Textbox(label="Translated 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. Optionally, translate the recipe to Spanish.",
-    allow_flagging="never"
+    description="Upload an image of a dish to get a detailed description, ingredient list, and step-by-step recipe. Optionally, translate the recipe to Spanish.",
+    flagging_mode="never"
 )
 
 # Lanzar la aplicación de Gradio