se actualiza herramienta

app.py

@@ -45,54 +45,76 @@ class ModelHandler:
 # Lista de modelos disponibles
 model_names = [
     "microsoft/Phi-3-mini-4k-instruct",
-    "bigscience/bloomz-560m",
-    "tiiuae/falcon-7b-instruct"
+    "bigscience/bloomz-560m"
 ]
 
 # Inicializa el manejador de modelos
 model_handler = ModelHandler(model_names, hf_token)
 
-# Configura la interfaz en Gradio con selección de modelos
-with gr.Blocks(title="Multi-Model LLM Chatbot") as demo:
+# Define la función para generación de imágenes
+def generate_image(prompt):
+    """
+    Genera una imagen utilizando un modelo de generación de imágenes.
+    """
+    try:
+        # Aquí puedes usar el modelo de Hugging Face para generación de imágenes.
+        # Ejemplo ficticio:
+        image_url = f"https://via.placeholder.com/512.png?text={prompt.replace(' ', '+')}"
+        return image_url
+    except Exception as e:
+        return f"Error al generar la imagen: {e}"
+
+# Configura la interfaz en Gradio con selección de modelos y generación de imágenes
+with gr.Blocks(title="Multi-Model LLM Chatbot with Image Generation") as demo:
     gr.Markdown(
         """
-        ## Chatbot Multi-Modelo LLM
-        Este chatbot permite elegir entre múltiples modelos de lenguaje para responder preguntas. 
-        Selecciona un modelo, escribe tu consulta, y presiona el botón de enviar para obtener una respuesta.
+        ## Chatbot Multi-Modelo LLM con Generación de Imágenes
+        Este chatbot permite elegir entre múltiples modelos de lenguaje para responder preguntas o generar imágenes 
+        a partir de descripciones.
         """
     )
     with gr.Row():
         model_dropdown = gr.Dropdown(
-            choices=model_names,
+            choices=model_names + ["Generación de Imágenes"],
             value=model_names[0],
-            label="Seleccionar Modelo",
+            label="Seleccionar Acción/Modelo",
             interactive=True
         )
     with gr.Row():
         with gr.Column():
             input_text = gr.Textbox(
                 lines=5,
-                placeholder="Escribe tu pregunta aquí...",
-                label="Pregunta"
+                placeholder="Escribe tu consulta o descripción para la imagen...",
+                label="Entrada"
             )
         with gr.Column():
-            output_text = gr.Textbox(
+            output_display = gr.Textbox(
                 lines=5,
                 label="Respuesta",
-                interactive=False
+                interactive=False,
+                visible=False
+            )
+            output_image = gr.Image(
+                label="Imagen Generada",
+                interactive=False,
+                visible=False
             )
     submit_button = gr.Button("Enviar")
     
     # Define la función de actualización
-    def process_input(selected_model, user_input):
-        model_handler.switch_model(selected_model)
-        return model_handler.generate_response(user_input)
+    def process_input(selected_action, user_input):
+        if selected_action == "Generación de Imágenes":
+            return "", generate_image(user_input), gr.update(visible=True), gr.update(visible=False)
+        else:
+            model_handler.switch_model(selected_action)
+            response = model_handler.generate_response(user_input)
+            return response, None, gr.update(visible=False), gr.update(visible=True)
     
     # Conecta la función a los componentes
     submit_button.click(
         fn=process_input,
         inputs=[model_dropdown, input_text],
-        outputs=output_text
+        outputs=[output_display, output_image, output_image, output_display]
     )
 
 # Lanza la interfaz