Update app.py

app.py

@@ -93,21 +93,23 @@ 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":
-        # 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
-
+    def process_input(selected_action, user_input):
+        try:
+            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
+        except Exception as e:
+            return f"Error: {e}", None
     
     # Conecta la función a los componentes
     submit_button.click(