# Crear la interfaz de Gradio
demo = gr.Interface(
    fn=handle_message,  # Función de manejo de mensaje y archivos
    inputs=[
        gr.Textbox(placeholder="Escribe un mensaje..."),
        gr.File(label="Subir imágenes", file_count="multiple", file_types=["image"])
    ],
    outputs=[
        gr.Chatbot(type="messages"),  # Formato actualizado para mensajes con "role" y "content"
        gr.Textbox(label="Conteo de imágenes")
    ],
    examples=[  # Ejemplos iniciales de mensajes
        ["No files", []],  # Ejemplo con solo texto y sin archivos
        ["Aquí tienes una imagen", ["path/to/example1.jpg"]],  # Ejemplo con texto y una imagen
    ],
    cache_examples=True
)

# Iniciar la interfaz
demo.launch()