Java_bot.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "gpuType": "T4",
      "authorship_tag": "ABX9TyPfiWRe8sDDWpZ2hUrQPbx0",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/mancher07/EmiReproductor/blob/main/Java_bot.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# @title Give me a name {display-mode: \"form\"}\n",
        "\n",
        "# This code will be hidden when the notebook is loaded.\n",
        "\n",
        "\n",
        "from transformers import AutoModelForCausalLM, AutoTokenizer\n",
        "import torch\n",
        "import ipywidgets as widgets\n",
        "from IPython.display import display, HTML\n",
        "\n",
        "# Cargar el modelo y el tokenizador\n",
        "checkpoint = \"Salesforce/codegen-350M-mono\"\n",
        "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
        "model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)\n",
        "tokenizer = AutoTokenizer.from_pretrained(checkpoint)\n",
        "\n",
        "# Función para generar respuestas del modelo con manejo de errores\n",
        "def generate_response(prompt):\n",
        "    try:\n",
        "        inputs = tokenizer(prompt, return_tensors=\"pt\").to(device)\n",
        "        if inputs['input_ids'].numel() == 0:\n",
        "            return \"Error: No input provided.\"\n",
        "        max_length = 1500  # Aumentar la longitud máxima para generación\n",
        "        completion = model.generate(\n",
        "            input_ids=inputs['input_ids'],\n",
        "            attention_mask=inputs.get('attention_mask'),\n",
        "            max_length=max_length,\n",
        "            temperature=0.7,\n",
        "            do_sample=True,\n",
        "            pad_token_id=tokenizer.eos_token_id\n",
        "        )\n",
        "        return tokenizer.decode(completion[0], skip_special_tokens=True)\n",
        "    except Exception as e:\n",
        "        return f\"Error: {str(e)}\"\n",
        "\n",
        "# Crear campos de entrada y botones\n",
        "input_box = widgets.Textarea(\n",
        "    description='',\n",
        "    placeholder='Type your message here...',\n",
        "    layout=widgets.Layout(width='100%', height='100px', margin='0px 0px 10px 0px')\n",
        ")\n",
        "send_button = widgets.Button(description=\"Send\")\n",
        "\n",
        "# Crear un área de salida desplazable y con mayor capacidad\n",
        "output_box = widgets.HTML(\n",
        "    value='',\n",
        "    placeholder='Bot replies will appear here...',\n",
        "    layout=widgets.Layout(width='100%', height='400px', overflow_y='auto', border='solid 1px gray')\n",
        ")\n",
        "\n",
        "# Contenedor para toda la aplicación\n",
        "container = widgets.VBox([output_box, input_box, send_button])\n",
        "\n",
        "# CSS para personalizar el estilo de los mensajes, los avatares y el ícono de verificado\n",
        "css = \"\"\"\n",
        "<style>\n",
        "    .message-container {\n",
        "        display: flex;\n",
        "        align-items: flex-start;\n",
        "        margin: 10px 0;\n",
        "        max-width: 100%;  /* Ajustar al 100% del contenedor padre */\n",
        "        word-wrap: break-word;\n",
        "    }\n",
        "    .message-avatar {\n",
        "        width: 40px;\n",
        "        height: 40px;\n",
        "        border-radius: 50%;\n",
        "        margin: 0 10px;  /* Asegura un espacio entre la imagen y la burbuja */\n",
        "    }\n",
        "    .message-box {\n",
        "        padding: 10px;\n",
        "        border-radius: 15px;  /* Borde redondeado */\n",
        "        max-width: 80%;  /* Limita el ancho de la burbuja para que no ocupe todo el espacio */\n",
        "        display: flex;\n",
        "        align-items: center;\n",
        "        flex-grow: 1;  /* Permite que la burbuja crezca y se ajuste al contenido */\n",
        "        box-sizing: border-box;  /* Incluye el padding en el ancho y alto */\n",
        "    }\n",
        "    .user-message {\n",
        "        background-color: #e7e7e7;\n",
        "        flex-direction: row-reverse;\n",
        "        text-align: left;\n",
        "    }\n",
        "    .bot-message {\n",
        "        background-color: #d1ffd1;\n",
        "        flex-direction: row;\n",
        "        text-align: left;\n",
        "    }\n",
        "    .message-name {\n",
        "        font-weight: bold;\n",
        "        margin-bottom: 5px;\n",
        "        display: flex;\n",
        "        align-items: center;\n",
        "    }\n",
        "    .message-name img {\n",
        "        margin-left: 5px;\n",
        "    }\n",
        "    .verified-icon {\n",
        "        width: 16px;\n",
        "        height: 16px;\n",
        "    }\n",
        "</style>\n",
        "\"\"\"\n",
        "\n",
        "def on_send_button_click(b):\n",
        "    user_message = input_box.value\n",
        "    if user_message.strip():  # Verificar que el mensaje no esté vacío\n",
        "        bot_reply = generate_response(user_message)\n",
        "        # Mostrar el mensaje del usuario y la respuesta del bot en el área de salida\n",
        "        output_box.value += f\"\"\"\n",
        "        <div class=\"message-container user-message\">\n",
        "            <img src=\"https://static.wikia.nocookie.net/yugiohenespanol/images/8/8b/Foto_invitaci%C3%B3n_avatar_doble.jpg/revision/latest?cb=20200827063040&path-prefix=es\" alt=\"User Avatar\" class=\"message-avatar\">\n",
        "            <div>\n",
        "                <div class=\"message-name\">You <img src=\"https://iconape.com/wp-content/png_logo_vector/google-verified.png\" alt=\"Verified\" class=\"verified-icon\"></div>\n",
        "                <div class=\"message-box\">\n",
        "                    {user_message}\n",
        "                </div>\n",
        "            </div>\n",
        "        </div>\n",
        "        <div class=\"message-container bot-message\">\n",
        "            <img src=\"https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR0m46PamPDvagmS__4zoTLlAAhUq7vL0fTbQ&s\" alt=\"Bot Avatar\" class=\"message-avatar\">\n",
        "            <div>\n",
        "                <div class=\"message-name\">Bot</div>\n",
        "                <div class=\"message-box\">\n",
        "                    {bot_reply}\n",
        "                </div>\n",
        "            </div>\n",
        "        </div>\n",
        "        \"\"\"\n",
        "        # Limpiar el campo de entrada después del envío\n",
        "        input_box.value = ''\n",
        "    else:\n",
        "        output_box.value += \"\"\"\n",
        "        <div class=\"message-container user-message\">\n",
        "            <img src=\"https://static.wikia.nocookie.net/yugiohenespanol/images/8/8b/Foto_invitaci%C3%B3n_avatar_doble.jpg/revision/latest?cb=20200827063040&path-prefix=es\" alt=\"User Avatar\" class=\"message-avatar\">\n",
        "            <div>\n",
        "                <div class=\"message-name\">You <img src=\"https://iconape.com/wp-content/png_logo_vector/google-verified.png\" alt=\"Verified\" class=\"verified-icon\"></div>\n",
        "                <div class=\"message-box\">\n",
        "                    (empty message)\n",
        "                </div>\n",
        "            </div>\n",
        "        </div>\n",
        "        <div class=\"message-container bot-message\">\n",
        "            <img src=\"https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR0m46PamPDvagmS__4zoTLlAAhUq7vL0fTbQ&s\" alt=\"Bot Avatar\" class=\"message-avatar\">\n",
        "            <div>\n",
        "                <div class=\"message-name\">Bot</div>\n",
        "                <div class=\"message-box\">\n",
        "                    Please enter a message.\n",
        "                </div>\n",
        "            </div>\n",
        "        </div>\n",
        "        \"\"\"\n",
        "\n",
        "send_button.on_click(on_send_button_click)\n",
        "\n",
        "# Mostrar los widgets en la interfaz de usuario\n",
        "display(HTML(css))  # Insertar el CSS para el estilo\n",
        "display(container)"
      ],
      "metadata": {
        "id": "jZpHBR7EmcX7"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}