Update app.py

app.py

@@ -93,13 +93,21 @@ with gr.Blocks(title="Multi-Model LLM Chatbot with Image Generation") as demo:
     submit_button = gr.Button("Enviar")
     
     # Define la función de actualización
-    def process_input(selected_action, user_input):
-        if selected_action == "Generación de Imágenes":
-            return generate_image_with_progress(user_input)
-        else:
-            model_handler.switch_model(selected_action)
-            response = model_handler.generate_response(user_input)
-            return response, None
+  def process_input(selected_action, user_input):
+    if selected_action == "Generación de Imágenes":
+        # Manejamos el generador de progreso
+        progress_generator = generate_image_with_progress(user_input)
+        last_status = None
+        last_image = None
+        for status, image in progress_generator:
+            last_status = status
+            last_image = image
+        return last_status, last_image
+    else:
+        model_handler.switch_model(selected_action)
+        response = model_handler.generate_response(user_input)
+        return response, None
+
     
     # Conecta la función a los componentes
     submit_button.click(