Update app.py

app.py

@@ -49,49 +49,22 @@ 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 específica del plato de comida
+    # Paso 3.1: Generar descripción del plato
     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 específica
-    recipe = generate_recipe(specific_description)
+    # Paso 3.2: Generar receta a partir de la descripción
+    recipe = generate_recipe(description)
     
-    return specific_description, recipe
+    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 con bloques
-def create_interface():
-    with gr.Blocks() as iface:
-        with gr.Row():
-            image_input = gr.Image(type="pil", label="Upload Dish Image")
-        with gr.Row():
-            description_output = gr.Textbox(label="Dish Description")
-            recipe_output = gr.Textbox(label="Recipe")
-        with gr.Row():
-            translate_button = gr.Button("Translate Recipe to Spanish")
-            translated_output = gr.Textbox(label="Translated Recipe")
-
-        # Procesar imagen y generar receta
-        def process_and_translate(image):
-            description, recipe = process_image(image)
-            return description, recipe, ""
-
-        image_input.change(fn=process_and_translate, inputs=image_input, outputs=[description_output, recipe_output, translated_output])
-
-        # Traducir receta
-        def translate_interface(recipe):
-            translated_recipe = translate_recipe(recipe)
-            return translated_recipe
-
-        translate_button.click(fn=translate_interface, inputs=recipe_output, outputs=translated_output)
-    
-    return iface
+# Paso 4: Crear la interfaz de Gradio
+iface = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="pil"),
+    outputs=["text", "text"],
+    title="Recipe Generator from Dish Image",
+    description="Upload an image of a dish to get a description, ingredient list, and step-by-step recipe."
+)
 
 # Lanzar la aplicación de Gradio
-iface_with_translation = create_interface()
-iface_with_translation.launch()
+iface.launch()