Chat_Bot_Sitio_do_Picapau_Amarelo.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyPHbA6dWqsd2YQGijW+m7vp",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/ClaudioMendonca-Eng/alura-imersao-ia/blob/main/Chat_Bot_Sitio_do_Picapau_Amarelo.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "####« Mais detalhes da configuração\n",
        "<center><a href=\"https://www.alura.com.br/\" target=\"_blank\"><img style=\"margin: 10px\" height=\"150\" width=\"500\" src=\"https://github.com/ClaudioMendonca-Eng/alura-imersao-ia/raw/main/img/logo_imersao.png\" alt=\"Alura logo\"/></a></center>\n",
        "<center><img src='https://upload.wikimedia.org/wikipedia/commons/thumb/4/45/Gemini_language_model_logo.png/200px-Gemini_language_model_logo.png' height=\"120\" alt=\"Gemini-logo\"/></center>\n",
        "\n",
        "<center><h3>Configuração do API Gemini</h3></center><br>\n",
        "<h1>⏬ DEPOIS DAS CONFIGURAÇÕES EXECUTE AQUI!!</h1>\n"
      ],
      "metadata": {
        "id": "AF8DvvT_QNs2"
      }
    },
    {
      "cell_type": "code",
      "execution_count": 80,
      "metadata": {
        "id": "YeM9DgpEmbcd",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title Instalando o SDK do Google Gemini\n",
        "# @markdown #### ⬅️ EXECUTE AQUI!! - Instalando o SDK do Google Gemini\n",
        "!pip install -q -U google-generativeai"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Configuração do API Gemini\n",
        "# @markdown #### ⬅️ EXECUTE AQUI!!\n",
        "#@markdown <center><h1>Configuração do API Gemini</h1></center><br>\n",
        "\n",
        "#@markdown Configurações:<br>&emsp;[Por favor, insira a sua API Key:](https://aistudio.google.com/app/apikey/?utm_source=website&utm_medium=referral&utm_campaign=Alura&utm_content=)\n",
        "API_KEY = \"Por favor, insira a sua API Key\" # @param {type:\"string\"}\n",
        "\n",
        "import google.generativeai as genai\n",
        "\n",
        "GOOGLE_API_KEY = API_KEY\n",
        "\n",
        "def mostrar_parametros_seguranca(parametros):\n",
        "    seguranca = {\n",
        "        \"Bloquear nenhum\": \"BLOCK_NONE\",\n",
        "        \"Bloquear poucos\": \"BLOCK_ONLY_HIGH\",\n",
        "        \"Bloquear alguns\": \"BLOCK_MEDIUM_AND_ABOVE\",\n",
        "        \"Bloquear muitos\": \"BLOCK_LOW_AND_ABOVE\"\n",
        "    }\n",
        "    if parametros in seguranca:\n",
        "        return seguranca[parametros]\n",
        "    else:\n",
        "        return \"Personagem não encontrado\"\n",
        "ASSEDIO = \"Bloquear nenhum\" # @param [\"Bloquear nenhum\", \"Bloquear poucos\", \"Bloquear alguns\", \"Bloquear muitos\"]\n",
        "HARASSMENT = mostrar_parametros_seguranca(ASSEDIO)\n",
        "ODIO = \"Bloquear nenhum\" # @param [\"Bloquear nenhum\", \"Bloquear poucos\", \"Bloquear alguns\", \"Bloquear muitos\"]\n",
        "HATE = mostrar_parametros_seguranca(ODIO)\n",
        "SEXUAL = \"Bloquear nenhum\" # @param [\"Bloquear nenhum\", \"Bloquear poucos\", \"Bloquear alguns\", \"Bloquear muitos\"]\n",
        "SEXY = mostrar_parametros_seguranca(SEXUAL)\n",
        "PERIGOSO = \"Bloquear nenhum\" # @param [\"Bloquear nenhum\", \"Bloquear poucos\", \"Bloquear alguns\", \"Bloquear muitos\"]\n",
        "DANGEROUS = mostrar_parametros_seguranca(PERIGOSO)\n",
        "\n",
        "safety_settings = {\n",
        "    \"HARASSMENT\": HARASSMENT,\n",
        "    \"HATE\": HATE,\n",
        "    \"SEXUAL\": SEXY,\n",
        "    \"DANGEROUS\": DANGEROUS,\n",
        "}\n",
        "\n",
        "def validar_api_key(api_key):\n",
        "  \"\"\"Verifica se a API Key é válida tentando listar os modelos disponíveis.\"\"\"\n",
        "  genai.configure(api_key=api_key)\n",
        "  try:\n",
        "    model = genai.GenerativeModel('gemini-pro')\n",
        "    response = model.generate_content(\"So para testa a API Key! resposta simples A API Key é valida!\")\n",
        "    if response.text == \"API Key válida!\":\n",
        "      print(response.text)\n",
        "      return True\n",
        "    else:\n",
        "      print(\"Nenhum modelo encontrado. Verifique sua chave de API.\")\n",
        "      return False\n",
        "  except Exception as e:\n",
        "    print(f\"Erro ao validar a API Key: {e}\")\n",
        "    return False\n",
        "\n",
        "\n",
        "# Testa a API Key\n",
        "if validar_api_key(GOOGLE_API_KEY):\n",
        "  print(\"Pronto para usar a API!\")\n",
        "else:\n",
        "  print(\"Verifique a API Key e tente novamente.\")"
      ],
      "metadata": {
        "id": "xDQXBO79w5Vp",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "# CHAT BOT - SITIO DO PICAPAU AMARELO\n",
        "<br><center><img src='https://i2.wp.com/sitio.pmvs.pt/wp-content/uploads/2014/06/cropped-11704119222.jpg' height=\"300\" alt=\"Gemini-logo\"/></center>\n",
        "\n",
        "> DIREITOS AUTORAIS - DOMÍNIO PÚBLICO\n",
        ">\n",
        "> Explore as infinitas possibilidades e divirta-se com as aventuras do Sítio do Picapau Amarelo!\n",
        "> Compartilhe essa informação com seus amigos e familiares para que todos possam aproveitar essa oportunidade incrível!\n",
        "<details>\n",
        "  <summary>Mais detalhes</summary>\n",
        "  O Sítio do Picapau Amarelo em Domínio Público: Uma Aventura Aberta à Todos!\n",
        "  \n",
        "  As obras de Monteiro Lobato, incluindo o Sítio do Picapau Amarelo, entraram em domínio público no dia 1º de janeiro de 2019! Isso significa que você é livre para:\n",
        "  \n",
        "  - Ler, reler e compartilhar as histórias com quem quiser.\n",
        "  - Adaptar as obras para outros formatos, como peças de teatro, filmes, músicas e até mesmo jogos.\n",
        "  - Publicar as obras em qualquer plataforma, seja online ou impressa.\n",
        "  - Traduzir as histórias para outros idiomas e levar a magia do Sítio para o mundo todo.\n",
        "  - Criar novas obras usando os personagens e elementos do Sítio, dando vida a novas aventuras.\n",
        "\n",
        "  Mas atenção:\n",
        "\n",
        "  - As ilustrações originais dos livros ainda são protegidas por direitos autorais. Para usá-las, você precisará obter a permissão dos detentores dos direitos.\n",
        "  - É sempre importante citar a fonte quando utilizar qualquer material do Sítio do Picapau Amarelo, mesmo que a obra esteja em domínio público.\n",
        "</details>"
      ],
      "metadata": {
        "id": "r3qYrg7UFnG2"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "# ============================= FORM ============================= #\n",
        "# @markdown #### ⬅️ EXECUTE O CHAT AQUI!!\n",
        "#@title Configuração do CHATBOT\n",
        "#@markdown <center><h3>CHAT COM OS PERSONAGEM </h3></center><br>\n",
        "\n",
        "\n",
        "import textwrap\n",
        "from IPython.display import display\n",
        "from IPython.display import Markdown\n",
        "\n",
        "# Melhorando a visualização\n",
        "def to_markdown(text):\n",
        "    text = text.replace('.', '\\n')\n",
        "    return Markdown(textwrap.indent(text,\n",
        "                                     prefix='## ',\n",
        "                                     predicate=lambda x: True))\n",
        "\n",
        "def mostrar_personalidade(personagem):\n",
        "    personalidades = {\n",
        "        \"Cuca\": \"Seja o personagem Cuca do Sítio do Picapau Amarelo: Aterrorizante e ardilosa, a Cuca é a personificação do medo no Sítio do Picapau Amarelo. Com sua aparência assustadora e seu jeito malvado, ela espalha terror por onde passa, sempre tramando planos nefastos para conseguir o que deseja.\",\n",
        "        \"Dona Benta\": \"Seja o personagem Dona Benta do Sítio do Picapau Amarelo: Sábia e experiente: Dona Benta é a matriarca do Sítio e uma fonte de sabedoria para todos os personagens. Ela tem uma vasta experiência de vida e sempre sabe o que fazer nas situações mais difíceis. Carinhosa e acolhedora: Dona Benta tem um coração bondoso e acolhedor. Ela trata todos com carinho e respeito, e sempre está disposta a oferecer um ombro amigo para quem precisa. Justa e honesta: Dona Benta acredita na justiça e na honestidade, e sempre ensina seus netos a serem pessoas boas e honestas. Paciente e compreensiva: Dona Benta é uma pessoa paciente e compreensiva, que sabe lidar com as travessuras de seus netos e amigos. Líder inspiradora: Dona Benta é uma líder inspiradora que guia e aconselha os personagens do Sítio. Ela é um exemplo de força, sabedoria e bondade para todos.\",\n",
        "        \"Emília\": \"Seja o personagem Emília do Sítio do Picapau Amarelo: Rebelde e questionadora: A boneca de pano que desafia regras e questiona tudo, sempre com um tom divertido e sarcástico. Criativa e inventiva: Emília vive criando soluções inusitadas para os problemas do Sítio, usando sua imaginação fértil e seu conhecimento vasto. Defensora da justiça: Sempre pronta para defender os mais fracos e lutar contra a injustiça, mesmo que isso signifique ir contra a corrente. Teimosa e persistente: Quando Emília coloca algo na cabeça, não desiste fácil. Ela persegue seus objetivos com teimosia e persistência, sempre buscando o melhor para todos. Leal e amiga: Apesar de suas diferenças, Emília é uma amiga fiel e leal para Narizinho e os outros personagens do Sítio. Ela sempre está lá para apoiar e aconselhar, mesmo quando discorda deles.\",\n",
        "        \"Narizinho\": \"Seja o personagem Narizinho do Sítio do Picapau Amarelo: Aventureira e destemida, Narizinho é a protagonista corajosa do Sítio do Picapau Amarelo. Com sua sede de aventura e sua determinação inabalável, ela está sempre pronta para enfrentar desafios e explorar novos horizontes, mesmo que isso signifique se meter em confusão.\",\n",
        "        \"Pedrinho\": \"Seja o personagemPedrinho do Sítio do Picapau Amarelo: Aventuroso e corajoso: Pedrinho adora explorar o mundo e viver novas aventuras, mesmo que isso signifique enfrentar perigos. Ele é corajoso e enfrenta seus medos de cabeça erguida. Curioso e questionador: Pedrinho tem uma mente curiosa e está sempre buscando aprender coisas novas. Ele faz perguntas e desafia as convenções, buscando entender o mundo ao seu redor. Bondoso e generoso: Pedrinho tem um bom coração e está sempre disposto a ajudar os outros. Ele se preocupa com seus amigos e familiares e faz tudo o que pode para ajudá-los. Impulsivo e precipitado: Às vezes, Pedrinho age antes de pensar, o que pode levá-lo a se meter em problemas. Ele precisa aprender a ser mais paciente e ponderado antes de tomar decisões. Leal e protetor: Pedrinho é um amigo leal e protetor, sempre pronto para defender seus amigos e familiares. Ele é um verdadeiro herói do Sítio do Picapau Amarelo.\",\n",
        "        \"Saci Pererê\": \"Seja o personagem Saci Pererê do Sítio do Picapau Amarelo: Brincalhão e travesso, o Saci Pererê é uma figura lendária do folclore brasileiro que habita o Sítio do Picapau Amarelo. Com sua habilidade de se locomover em um redemoinho e suas travessuras constantes, ele traz uma energia divertida e imprevisível para o ambiente, sempre pronto para uma boa brincadeira\",\n",
        "        \"Tia Nastácia\": \"Seja o personagem Tia Nastácia do Sítio do Picapau Amarelo: Amorosa e atenciosa, Tia Nastácia é a figura materna do Sítio do Picapau Amarelo. Com seu coração generoso e seu talento culinário incomparável, ela cuida de todos com carinho e dedicação, garantindo que não falte amor e comida na vida de quem vive no Sítio.\",\n",
        "        \"Visconde de Sabugosa\": \"Seja o personagem Visconde de Sabugosa do Sítio do Picapau Amarelo: Inteligente e estudioso: O Visconde é o intelectual do Sítio, sempre buscando aprender coisas novas e expandir seus conhecimentos. Ele é um estudioso dedicado que lê livros e faz pesquisas sobre os mais diversos assuntos. Racional e lógico: O Visconde sempre busca analisar as situações de forma racional e lógica, buscando soluções baseadas em fatos e dados. Organizado e metódico: O Visconde é uma pessoa organizada e metódica, que gosta de ter tudo em ordem e sob controle. Um pouco tímido e introvertido: O Visconde pode ser um pouco tímido e introvertido, preferindo a companhia de livros e pesquisas à companhia de outras pessoas. Amigo fiel e leal: Apesar de sua timidez, o Visconde é um amigo fiel e leal, que sempre está disposto a ajudar seus amigos.\",\n",
        "        \"Zé Carneiro\": \"Seja o personagem Zé Carneiro do Sítio do Picapau Amarelo: Guloso e preguiçoso, Zé Carneiro é o porco do Sítio do Picapau Amarelo. Com sua paixão por comida e sua tendência a procrastinar, ele representa o lado indulgente e preguiçoso de muitos de nós, sempre em busca de conforto e prazer imediato.\"\n",
        "    }\n",
        "\n",
        "    if personagem in personalidades:\n",
        "        return personalidades[personagem]\n",
        "    else:\n",
        "        return \"Personagem não encontrado\"\n",
        "\n",
        "#@markdown Configurações:<br>&emsp;Por favor, digite seu nome para que eu possa conhecê-lo(a) melhor:\n",
        "Nome = \"Por favor, digite seu nome\" # @param {type:\"string\"}\n",
        "#@markdown <br>&emsp;Escolha o Personagem do Sítio para Conversar:\n",
        "Personagem = \"Emília\" # @param [\"Cuca\", \"Dona Benta\", \"Emília\", \"Narizinho\", \"Pedrinho\", \"Saci Pererê\", \"Tia Nastácia\", \"Visconde de Sabugosa\", \"Zé Carneiro\"]\n",
        "Personalidade = mostrar_personalidade(Personagem)\n",
        "#@markdown Observação:<br>&emsp;Quando quiser encerrar o chat, basta digitar **\"fim\"**.\n",
        "\n",
        "\n",
        "generation_config = {\n",
        "  \"temperature\": 1,\n",
        "  \"top_p\": 0.95,\n",
        "  \"top_k\": 0,\n",
        "  \"max_output_tokens\": 8192,\n",
        "}\n",
        "system_instruction = f\"{Personalidade} Sabendo que eu sou {Nome} uma pessoa que vc acabou de conhecer!\"\n",
        "\n",
        "model = genai.GenerativeModel(model_name=\"gemini-1.5-pro-latest\",\n",
        "                              generation_config=generation_config,\n",
        "                              system_instruction=system_instruction,\n",
        "                              safety_settings=safety_settings)\n",
        "\n",
        "print('=============================================================================================================================================================================')\n",
        "print('███████╗██╗████████╗██╗ ██████╗     ██████╗  ██████╗     ██████╗ ██╗ ██████╗ █████╗ ██████╗  █████╗ ██╗   ██╗     █████╗ ███╗   ███╗ █████╗ ██████╗ ███████╗██╗      ██████╗ ')\n",
        "print('██╔════╝██║╚══██╔══╝██║██╔═══██╗    ██╔══██╗██╔═══██╗    ██╔══██╗██║██╔════╝██╔══██╗██╔══██╗██╔══██╗██║   ██║    ██╔══██╗████╗ ████║██╔══██╗██╔══██╗██╔════╝██║     ██╔═══██╗')\n",
        "print('███████╗██║   ██║   ██║██║   ██║    ██║  ██║██║   ██║    ██████╔╝██║██║     ███████║██████╔╝███████║██║   ██║    ███████║██╔████╔██║███████║██████╔╝█████╗  ██║     ██║   ██║')\n",
        "print('╚════██║██║   ██║   ██║██║   ██║    ██║  ██║██║   ██║    ██╔═══╝ ██║██║     ██╔══██║██╔═══╝ ██╔══██║██║   ██║    ██╔══██║██║╚██╔╝██║██╔══██║██╔══██╗██╔══╝  ██║     ██║   ██║')\n",
        "print('███████║██║   ██║   ██║╚██████╔╝    ██████╔╝╚██████╔╝    ██║     ██║╚██████╗██║  ██║██║     ██║  ██║╚██████╔╝    ██║  ██║██║ ╚═╝ ██║██║  ██║██║  ██║███████╗███████╗╚██████╔╝')\n",
        "print('╚══════╝╚═╝   ╚═╝   ╚═╝ ╚═════╝     ╚═════╝  ╚═════╝     ╚═╝     ╚═╝ ╚═════╝╚═╝  ╚═╝╚═╝     ╚═╝  ╚═╝ ╚═════╝     ╚═╝  ╚═╝╚═╝     ╚═╝╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝ ╚═════╝ ')\n",
        "print('=============================================================================================================================================================================')\n",
        "print(f'Por favor, digite uma pergunta para {Personagem}: ')\n",
        "print('---------------------------------------------')\n",
        "\n",
        "chat = model.start_chat(history=[])\n",
        "prompt = input(f\"{Nome}: \")\n",
        "\n",
        "while prompt != \"fim\":\n",
        "  response = chat.send_message(prompt)\n",
        "  # Melhorando a visualização\n",
        "  # Imprimindo o final histórico\n",
        "  if chat.history:\n",
        "    last_message = chat.history[-1]\n",
        "    display(to_markdown(f'**{Personagem}**: {last_message.parts[0].text}'))\n",
        "    print('---------------------------------------------')\n",
        "\n",
        "  prompt = input(f\"{Nome}: \")\n",
        "\n",
        "# Melhorando a visualização\n",
        "# Imprimindo Resumo do histórico\n",
        "for message in chat.history:\n",
        "    if message.role == \"user\":\n",
        "      display(to_markdown(f'**{Nome}**: {message.parts[0].text}'))\n",
        "      print('------------------')\n",
        "    else:\n",
        "        display(to_markdown(f'**{Personagem}**: {message.parts[0].text}'))\n",
        "        print('------------------')\n"
      ],
      "metadata": {
        "cellView": "form",
        "id": "njpuDyj0eTyd"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# @markdown #### ⬅️ EXECUTE AQUI!! - Histórico da última conversa\n",
        "#@title Configuração do API Gemini\n",
        "import textwrap\n",
        "from IPython.display import display\n",
        "from IPython.display import Markdown\n",
        "\n",
        "# Melhorando a visualização\n",
        "def to_markdown(text):\n",
        "    text = text.replace('.', '\\n')\n",
        "    return Markdown(textwrap.indent(text,\n",
        "                                     prefix='## ',\n",
        "                                     predicate=lambda x: True))\n",
        "if chat.history != []:\n",
        "  print('=============================================================================================================================================================================')\n",
        "  print('███████╗██╗████████╗██╗ ██████╗     ██████╗  ██████╗     ██████╗ ██╗ ██████╗ █████╗ ██████╗  █████╗ ██╗   ██╗     █████╗ ███╗   ███╗ █████╗ ██████╗ ███████╗██╗      ██████╗ ')\n",
        "  print('██╔════╝██║╚══██╔══╝██║██╔═══██╗    ██╔══██╗██╔═══██╗    ██╔══██╗██║██╔════╝██╔══██╗██╔══██╗██╔══██╗██║   ██║    ██╔══██╗████╗ ████║██╔══██╗██╔══██╗██╔════╝██║     ██╔═══██╗')\n",
        "  print('███████╗██║   ██║   ██║██║   ██║    ██║  ██║██║   ██║    ██████╔╝██║██║     ███████║██████╔╝███████║██║   ██║    ███████║██╔████╔██║███████║██████╔╝█████╗  ██║     ██║   ██║')\n",
        "  print('╚════██║██║   ██║   ██║██║   ██║    ██║  ██║██║   ██║    ██╔═══╝ ██║██║     ██╔══██║██╔═══╝ ██╔══██║██║   ██║    ██╔══██║██║╚██╔╝██║██╔══██║██╔══██╗██╔══╝  ██║     ██║   ██║')\n",
        "  print('███████║██║   ██║   ██║╚██████╔╝    ██████╔╝╚██████╔╝    ██║     ██║╚██████╗██║  ██║██║     ██║  ██║╚██████╔╝    ██║  ██║██║ ╚═╝ ██║██║  ██║██║  ██║███████╗███████╗╚██████╔╝')\n",
        "  print('╚══════╝╚═╝   ╚═╝   ╚═╝ ╚═════╝     ╚═════╝  ╚═════╝     ╚═╝     ╚═╝ ╚═════╝╚═╝  ╚═╝╚═╝     ╚═╝  ╚═╝ ╚═════╝     ╚═╝  ╚═╝╚═╝     ╚═╝╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝ ╚═════╝ ')\n",
        "  print('=============================================================================================================================================================================')\n",
        "  print(f'Histórico da última conversa com {Personagem}: ')\n",
        "  print('---------------------------------------------')\n",
        "else:\n",
        "  print('---------------------------------------------')\n",
        "  print('---------------  SEM HISTORICO  -------------')\n",
        "  print('---------------------------------------------')\n",
        "\n",
        "# Imprimindo o histórico\n",
        "for message in chat.history:\n",
        "    if message.role == \"user\":\n",
        "      display(to_markdown(f'**{Nome}**: {message.parts[0].text}'))\n",
        "      print('------------------------------------------------------------------------------------------------------------')\n",
        "    else:\n",
        "        display(to_markdown(f'**{Personagem}**: {message.parts[0].text}'))\n",
        "        print('------------------------------------------------------------------------------------------------------------')"
      ],
      "metadata": {
        "cellView": "form",
        "id": "Jzaw2E36lR30"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "#Licença\n",
        "\n",
        "Copyright © 2024 <a href=\"https://www.claudiomendonca.eng.br\" target=\"_blank\">ClaudioMendonca.eng.br</a>. Projeto <a href=\"https://github.com/ClaudioMendonca-Eng/alura-imersao-ia\" target=\"_blank\">GitHub</a>"
      ],
      "metadata": {
        "id": "Q22uoLcOptrD"
      }
    }
  ]
}