aula_4.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "a60f416c",
   "metadata": {
    "id": "1whNGuuGHS8j"
   },
   "source": [
    "# Aula 1 - Utilizando modelos de LLM"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "68b52afd",
   "metadata": {
    "id": "O_fo8tAJTzKR"
   },
   "source": [
    "## Vídeo 1.2 - Obtendo uma resposta da LLM"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "20738c57",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "MckJUd_VHFAe",
    "outputId": "5eb1fb20-3de3-4d20-c005-7eb171136614"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting litellm\n",
      "  Downloading litellm-1.50.1-py3-none-any.whl.metadata (32 kB)\n",
      "Requirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from litellm) (3.10.10)\n",
      "Requirement already satisfied: click in /usr/local/lib/python3.10/dist-packages (from litellm) (8.1.7)\n",
      "Requirement already satisfied: importlib-metadata>=6.8.0 in /usr/local/lib/python3.10/dist-packages (from litellm) (8.5.0)\n",
      "Requirement already satisfied: jinja2<4.0.0,>=3.1.2 in /usr/local/lib/python3.10/dist-packages (from litellm) (3.1.4)\n",
      "Requirement already satisfied: jsonschema<5.0.0,>=4.22.0 in /usr/local/lib/python3.10/dist-packages (from litellm) (4.23.0)\n",
      "Collecting openai>=1.52.0 (from litellm)\n",
      "  Downloading openai-1.52.0-py3-none-any.whl.metadata (24 kB)\n",
      "Requirement already satisfied: pydantic<3.0.0,>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from litellm) (2.9.2)\n",
      "Collecting python-dotenv>=0.2.0 (from litellm)\n",
      "  Downloading python_dotenv-1.0.1-py3-none-any.whl.metadata (23 kB)\n",
      "Requirement already satisfied: requests<3.0.0,>=2.31.0 in /usr/local/lib/python3.10/dist-packages (from litellm) (2.32.3)\n",
      "Collecting tiktoken>=0.7.0 (from litellm)\n",
      "  Downloading tiktoken-0.8.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (6.6 kB)\n",
      "Requirement already satisfied: tokenizers in /usr/local/lib/python3.10/dist-packages (from litellm) (0.19.1)\n",
      "Requirement already satisfied: zipp>=3.20 in /usr/local/lib/python3.10/dist-packages (from importlib-metadata>=6.8.0->litellm) (3.20.2)\n",
      "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2<4.0.0,>=3.1.2->litellm) (3.0.1)\n",
      "Requirement already satisfied: attrs>=22.2.0 in /usr/local/lib/python3.10/dist-packages (from jsonschema<5.0.0,>=4.22.0->litellm) (24.2.0)\n",
      "Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /usr/local/lib/python3.10/dist-packages (from jsonschema<5.0.0,>=4.22.0->litellm) (2024.10.1)\n",
      "Requirement already satisfied: referencing>=0.28.4 in /usr/local/lib/python3.10/dist-packages (from jsonschema<5.0.0,>=4.22.0->litellm) (0.35.1)\n",
      "Requirement already satisfied: rpds-py>=0.7.1 in /usr/local/lib/python3.10/dist-packages (from jsonschema<5.0.0,>=4.22.0->litellm) (0.20.0)\n",
      "Requirement already satisfied: anyio<5,>=3.5.0 in /usr/local/lib/python3.10/dist-packages (from openai>=1.52.0->litellm) (3.7.1)\n",
      "Requirement already satisfied: distro<2,>=1.7.0 in /usr/lib/python3/dist-packages (from openai>=1.52.0->litellm) (1.7.0)\n",
      "Collecting httpx<1,>=0.23.0 (from openai>=1.52.0->litellm)\n",
      "  Downloading httpx-0.27.2-py3-none-any.whl.metadata (7.1 kB)\n",
      "Collecting jiter<1,>=0.4.0 (from openai>=1.52.0->litellm)\n",
      "  Downloading jiter-0.6.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (5.2 kB)\n",
      "Requirement already satisfied: sniffio in /usr/local/lib/python3.10/dist-packages (from openai>=1.52.0->litellm) (1.3.1)\n",
      "Requirement already satisfied: tqdm>4 in /usr/local/lib/python3.10/dist-packages (from openai>=1.52.0->litellm) (4.66.5)\n",
      "Requirement already satisfied: typing-extensions<5,>=4.11 in /usr/local/lib/python3.10/dist-packages (from openai>=1.52.0->litellm) (4.12.2)\n",
      "Requirement already satisfied: annotated-types>=0.6.0 in /usr/local/lib/python3.10/dist-packages (from pydantic<3.0.0,>=2.0.0->litellm) (0.7.0)\n",
      "Requirement already satisfied: pydantic-core==2.23.4 in /usr/local/lib/python3.10/dist-packages (from pydantic<3.0.0,>=2.0.0->litellm) (2.23.4)\n",
      "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0,>=2.31.0->litellm) (3.4.0)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0,>=2.31.0->litellm) (3.10)\n",
      "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0,>=2.31.0->litellm) (2.2.3)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests<3.0.0,>=2.31.0->litellm) (2024.8.30)\n",
      "Requirement already satisfied: regex>=2022.1.18 in /usr/local/lib/python3.10/dist-packages (from tiktoken>=0.7.0->litellm) (2024.9.11)\n",
      "Requirement already satisfied: aiohappyeyeballs>=2.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (2.4.3)\n",
      "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (1.3.1)\n",
      "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (1.4.1)\n",
      "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (6.1.0)\n",
      "Requirement already satisfied: yarl<2.0,>=1.12.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (1.15.2)\n",
      "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->litellm) (4.0.3)\n",
      "Requirement already satisfied: huggingface-hub<1.0,>=0.16.4 in /usr/local/lib/python3.10/dist-packages (from tokenizers->litellm) (0.24.7)\n",
      "Requirement already satisfied: exceptiongroup in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.5.0->openai>=1.52.0->litellm) (1.2.2)\n",
      "Collecting httpcore==1.* (from httpx<1,>=0.23.0->openai>=1.52.0->litellm)\n",
      "  Downloading httpcore-1.0.6-py3-none-any.whl.metadata (21 kB)\n",
      "Collecting h11<0.15,>=0.13 (from httpcore==1.*->httpx<1,>=0.23.0->openai>=1.52.0->litellm)\n",
      "  Downloading h11-0.14.0-py3-none-any.whl.metadata (8.2 kB)\n",
      "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.16.4->tokenizers->litellm) (3.16.1)\n",
      "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.16.4->tokenizers->litellm) (2024.6.1)\n",
      "Requirement already satisfied: packaging>=20.9 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.16.4->tokenizers->litellm) (24.1)\n",
      "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.16.4->tokenizers->litellm) (6.0.2)\n",
      "Requirement already satisfied: propcache>=0.2.0 in /usr/local/lib/python3.10/dist-packages (from yarl<2.0,>=1.12.0->aiohttp->litellm) (0.2.0)\n",
      "Downloading litellm-1.50.1-py3-none-any.whl (6.3 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m6.3/6.3 MB\u001b[0m \u001b[31m25.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading openai-1.52.0-py3-none-any.whl (386 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m386.9/386.9 kB\u001b[0m \u001b[31m7.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading python_dotenv-1.0.1-py3-none-any.whl (19 kB)\n",
      "Downloading tiktoken-0.8.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.2/1.2 MB\u001b[0m \u001b[31m23.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading httpx-0.27.2-py3-none-any.whl (76 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m76.4/76.4 kB\u001b[0m \u001b[31m3.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading httpcore-1.0.6-py3-none-any.whl (78 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m78.0/78.0 kB\u001b[0m \u001b[31m3.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading jiter-0.6.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (325 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m325.2/325.2 kB\u001b[0m \u001b[31m18.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading h11-0.14.0-py3-none-any.whl (58 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m58.3/58.3 kB\u001b[0m \u001b[31m2.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hInstalling collected packages: python-dotenv, jiter, h11, tiktoken, httpcore, httpx, openai, litellm\n",
      "Successfully installed h11-0.14.0 httpcore-1.0.6 httpx-0.27.2 jiter-0.6.1 litellm-1.50.1 openai-1.52.0 python-dotenv-1.0.1 tiktoken-0.8.0\n"
     ]
    }
   ],
   "source": [
    "!pip install litellm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9374ee33",
   "metadata": {
    "id": "MPeusxnfUIfl"
   },
   "outputs": [],
   "source": [
    "from google.colab import userdata\n",
    "GROQ_API_KEY = userdata.get('GROQ_API_KEY')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6a55b5db",
   "metadata": {
    "id": "08-ArVtEUQYw"
   },
   "outputs": [],
   "source": [
    "from litellm import completion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5883eb7e",
   "metadata": {
    "id": "M6YPk7gmUVx7"
   },
   "outputs": [],
   "source": [
    "messages = [\n",
    "    {\"role\": \"system\", \"content\": \"Você é o Chat da Terra e do Universo e responde em português brasileiro perguntas sobre a previsão do tempo na Terra e do espaço próximo à Terra, além de informações sobre terremotos. \"},\n",
    "    {\"role\": \"user\", \"content\": \"Qual a frequência dos máximos solares?\"}\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "dc01e664",
   "metadata": {
    "id": "cf8mdnGJUdvq"
   },
   "outputs": [],
   "source": [
    "response = completion(\n",
    "    model=\"groq/gemma2-9b-it\",\n",
    "    messages=messages,\n",
    "    api_key=GROQ_API_KEY,\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b396687e",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "_lfHk7zYUl86",
    "outputId": "7b854ff4-fa6c-4e91-9789-d719e4762bd6"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Olá! 🪐\n",
      "\n",
      "Como Chat da Terra e do Universo, posso te ajudar com isso. \n",
      "\n",
      "Os máximos solares acontecem em ciclos que variam entre 9 e 11 anos. A frequência exata pode ser um pouco irregular, mas **em média, um máximo solar ocorre a cada 11 anos**.\n",
      "\n",
      "Durante esses períodos, a atividade solar aumenta, com mais manchas solares, flares e ejeções de massa coronal.  \n",
      "\n",
      "Preciso ressaltar que estes ciclos são complexos e ainda não compreendidos completamente!  😅\n",
      "\n",
      "Qual outra pergunta você tem sobre o universo? 😊\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(response.choices[0].message.content)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cb32f3f7",
   "metadata": {
    "id": "dzMWPK1wUpW8"
   },
   "outputs": [],
   "source": [
    "response = completion(\n",
    "    model=\"groq/llama3-groq-70b-8192-tool-use-preview\",\n",
    "    messages=messages,\n",
    "    api_key=GROQ_API_KEY,\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f2aa752f",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "UnhZ2c1AUvH8",
    "outputId": "316863e9-e4db-4094-85ba-84b4a2fbbbd2"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "A frequência dos máximos solares, também conhecida como ciclo solar, é de aproximadamente 11 anos. Durante este período, a atividade solar, como manchas solares e erupções solares, aumenta e diminui em um padrão cíclico.\n"
     ]
    }
   ],
   "source": [
    "print(response.choices[0].message.content)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "aa4b1ea0",
   "metadata": {
    "id": "-UUaQWHUVON-"
   },
   "source": [
    "## Vídeo 1.3 - Construindo um chatbot"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "690e7cdb",
   "metadata": {
    "id": "doCTl6pmUw6G"
   },
   "outputs": [],
   "source": [
    "# Função para chamar a API com o histórico de mensagens\n",
    "def call_groq_api(messages, model=\"groq/llama3-groq-70b-8192-tool-use-preview\"):\n",
    "    response = completion(\n",
    "        model=model,\n",
    "        messages=messages,\n",
    "        api_key=GROQ_API_KEY,\n",
    "    )\n",
    "    return response.choices[0].message.content"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "966c8f8c",
   "metadata": {
    "id": "ZCM9JM_SVi6x"
   },
   "outputs": [],
   "source": [
    "# Função para iniciar o chat, mantendo o histórico\n",
    "def chat():\n",
    "    print(\"Iniciando chat com o modelo. Digite 'sair' para encerrar.\")\n",
    "\n",
    "    # Histórico de mensagens\n",
    "    messages = [{\"role\": \"system\", \"content\": \"\"\"\n",
    "    Você é o Chat da Terra e do Universo e responde em português brasileiro\n",
    "    perguntas sobre a previsão do tempo na Terra e do espaço próximo à Terra, além de informações sobre terremotos.\n",
    "    \"\"\"}]\n",
    "\n",
    "    while True:\n",
    "        user_message = input(\"Você: \")\n",
    "        if user_message.lower() == \"sair\":\n",
    "            print(\"Encerrando chat. Até a próxima!\")\n",
    "            break\n",
    "\n",
    "        # Adicionar a mensagem do usuário ao histórico\n",
    "        messages.append({\"role\": \"user\", \"content\": user_message})\n",
    "\n",
    "        # Chamar a API com o histórico completo\n",
    "        model_response = call_groq_api(messages)\n",
    "\n",
    "        # Adicionar a resposta do modelo ao histórico\n",
    "        messages.append({\"role\": \"assistant\", \"content\": model_response})\n",
    "\n",
    "        # Exibir a resposta do assistente\n",
    "        print(f\"Assistente: {model_response}\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3f064a5e",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "QBYg5fX5VziJ",
    "outputId": "09ea6c2f-e62a-449e-cf28-c155559ce7d2"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Iniciando chat com o modelo. Digite 'sair' para encerrar.\n",
      "Você: Qual a previsão do tempo para São Paulo?\n",
      "Assistente: A previsão do tempo para São Paulo é de céu nublado com possibilidade de chuvas isoladas ao longo do dia. A temperatura deve variar entre 18°C e 24°C.\n",
      "Você: Sair\n",
      "Encerrando chat. Até a próxima!\n"
     ]
    }
   ],
   "source": [
    "chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "cc7cd146",
   "metadata": {
    "id": "PlSsXqo2XMUo"
   },
   "source": [
    "## Vídeo 1.4 - Pegando dados de uma API"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3b7e18d1",
   "metadata": {
    "id": "R-bwgFFGYYNW"
   },
   "source": [
    "https://openweathermap.org/price"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "df811f31",
   "metadata": {
    "id": "Ili_ULsOWjG-"
   },
   "outputs": [],
   "source": [
    "import requests\n",
    "import json"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "81e85247",
   "metadata": {
    "id": "4gIrc5yAYnea"
   },
   "outputs": [],
   "source": [
    "def previsao_do_tempo(city, country):\n",
    "    WEATHER_API = userdata.get('WEATHER_API')\n",
    "    url = f\"http://api.openweathermap.org/data/2.5/weather?q={city},{country}&APPID={WEATHER_API}&lang=pt_br&units=metric\"\n",
    "    response = requests.get(url)\n",
    "    data = response.json()\n",
    "\n",
    "    return json.dumps(data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ad93025d",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 87
    },
    "id": "d4n-SZVQYpPG",
    "outputId": "4cd23f5a-f59d-4604-9214-87b5475a8ea2"
   },
   "outputs": [
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'{\"coord\": {\"lon\": -43.2075, \"lat\": -22.9028}, \"weather\": [{\"id\": 803, \"main\": \"Clouds\", \"description\": \"nublado\", \"icon\": \"04d\"}], \"base\": \"stations\", \"main\": {\"temp\": 25.2, \"feels_like\": 25.86, \"temp_min\": 24.98, \"temp_max\": 26.12, \"pressure\": 1016, \"humidity\": 80, \"sea_level\": 1016, \"grnd_level\": 1017}, \"visibility\": 10000, \"wind\": {\"speed\": 3.6, \"deg\": 190}, \"clouds\": {\"all\": 75}, \"dt\": 1729536174, \"sys\": {\"type\": 1, \"id\": 8429, \"country\": \"BR\", \"sunrise\": 1729498493, \"sunset\": 1729544374}, \"timezone\": -10800, \"id\": 3451190, \"name\": \"Rio de Janeiro\", \"cod\": 200}'"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "previsao_do_tempo('Rio de Janeiro', 'BR')"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "83b6f3d6",
   "metadata": {
    "id": "O5C4D3JwZFRE"
   },
   "source": [
    "# Aula 2 - Usando ferramentas"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "28ec1001",
   "metadata": {
    "id": "HPE-RY7_bxW-"
   },
   "source": [
    "## Vídeo 2.1 - Criando um dicionário"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e84dd119",
   "metadata": {
    "id": "XAZvr_-9ZkVS"
   },
   "outputs": [],
   "source": [
    "tools = [\n",
    "        {\n",
    "            \"type\": \"function\",\n",
    "            \"function\": {\n",
    "                \"name\": \"previsao_do_tempo\",\n",
    "                \"description\": \"Retorna a previsão do tempo em uma cidade específica\",\n",
    "                \"parameters\": {\n",
    "                    \"type\": \"object\",\n",
    "                    \"properties\": {\n",
    "                        \"city\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Nome da cidade\",\n",
    "                        },\n",
    "                        \"country\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Sigla do país\",\n",
    "                        },\n",
    "                    },\n",
    "                    \"required\": [\"city\", \"country\"],\n",
    "                },\n",
    "            }\n",
    "\n",
    "        }\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "7eb4132c",
   "metadata": {
    "id": "9iQwsTLOYqlj"
   },
   "outputs": [],
   "source": [
    "# Função para chamar a API com o histórico de mensagens\n",
    "def call_groq_api(messages, model=\"groq/llama3-groq-70b-8192-tool-use-preview\"):\n",
    "    global tools\n",
    "    response = completion(\n",
    "        model=model,\n",
    "        messages=messages,\n",
    "        tools=tools,\n",
    "        tool_choice=\"auto\",\n",
    "        api_key=GROQ_API_KEY,\n",
    "    )\n",
    "    resposta_texto = response.choices[0].message\n",
    "    chamada_ferramentas = resposta_texto.tool_calls\n",
    "    if chamada_ferramentas:\n",
    "      available_functions = {\n",
    "        \"previsao_do_tempo\": previsao_do_tempo,\n",
    "      }\n",
    "      for tool_call in chamada_ferramentas:\n",
    "        function_name = tool_call.function.name\n",
    "        function_to_call = available_functions[function_name]\n",
    "        function_args = json.loads(tool_call.function.arguments)\n",
    "        function_response = function_to_call(\n",
    "            city=function_args.get(\"city\"),\n",
    "            country=function_args.get(\"country\"),\n",
    "        )\n",
    "        return function_response\n",
    "\n",
    "    else:\n",
    "      return resposta_texto.content\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "88670e60",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "PbanNmxWcsA0",
    "outputId": "4f957f44-7e95-445c-ae6b-2bf2ae5cbdde"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Iniciando chat com o modelo. Digite 'sair' para encerrar.\n",
      "Você: Qual a previsão do tempo para São Paulo?\n",
      "Assistente: {\"coord\": {\"lon\": -46.6361, \"lat\": -23.5475}, \"weather\": [{\"id\": 803, \"main\": \"Clouds\", \"description\": \"nublado\", \"icon\": \"04d\"}], \"base\": \"stations\", \"main\": {\"temp\": 24.05, \"feels_like\": 24.5, \"temp_min\": 23.03, \"temp_max\": 24.94, \"pressure\": 1014, \"humidity\": 76, \"sea_level\": 1014, \"grnd_level\": 926}, \"visibility\": 10000, \"wind\": {\"speed\": 6.17, \"deg\": 170}, \"clouds\": {\"all\": 75}, \"dt\": 1729537054, \"sys\": {\"type\": 1, \"id\": 8394, \"country\": \"BR\", \"sunrise\": 1729499279, \"sunset\": 1729545233}, \"timezone\": -10800, \"id\": 3448439, \"name\": \"S\\u00e3o Paulo\", \"cod\": 200}\n",
      "Você: Devo levar um guarda chuva?\n",
      "Assistente: Sim, é recomendável levar um guarda-chuva, pois a previsão do tempo indica que está nublado em São Paulo.\n",
      "Você: Sair\n",
      "Encerrando chat. Até a próxima!\n"
     ]
    }
   ],
   "source": [
    "chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "7c9d291a",
   "metadata": {
    "id": "Z_zthxtLeiaY"
   },
   "source": [
    "## Vídeo 2.2 - Adicionando uma segunda API"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "13f7c44b",
   "metadata": {
    "id": "nAn-YDn7cu-L"
   },
   "outputs": [],
   "source": [
    "def verificar_tempestade_solar():\n",
    "    url = \"https://services.swpc.noaa.gov/products/noaa-planetary-k-index.json\"\n",
    "    response = requests.get(url)\n",
    "    if response.status_code == 200:\n",
    "        data = response.json()\n",
    "        latest_kp = float(data[-1][1])  # O último valor Kp\n",
    "        if latest_kp >= 5:\n",
    "            return f\"Alerta de tempestade solar! Índice Kp atual: {latest_kp}\"\n",
    "        else:\n",
    "            return f\"Sem tempestade solar no momento. Índice Kp atual: {latest_kp}\"\n",
    "    else:\n",
    "        return \"Não foi possível obter informações sobre tempestades solares no momento.\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "aea61638",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 35
    },
    "id": "IMxkzxmRe-xB",
    "outputId": "24058fb0-1536-48de-a01d-1e0fadeb5959"
   },
   "outputs": [
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'Sem tempestade solar no momento. Índice Kp atual: 1.67'"
      ]
     },
     "execution_count": 25,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "verificar_tempestade_solar()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f5119557",
   "metadata": {
    "id": "EE6z038-fAML"
   },
   "outputs": [],
   "source": [
    "tools = [\n",
    "        {\n",
    "            \"type\": \"function\",\n",
    "            \"function\": {\n",
    "                \"name\": \"previsao_do_tempo\",\n",
    "                \"description\": \"Retorna a previsão do tempo em uma cidade específica\",\n",
    "                \"parameters\": {\n",
    "                    \"type\": \"object\",\n",
    "                    \"properties\": {\n",
    "                        \"city\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Nome da cidade\",\n",
    "                        },\n",
    "                        \"country\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Sigla do país\",\n",
    "                        },\n",
    "                    },\n",
    "                    \"required\": [\"city\", \"country\"],\n",
    "                },\n",
    "            }\n",
    "\n",
    "        },\n",
    "\n",
    "        {\n",
    "        \"type\": \"function\",\n",
    "        \"function\": {\n",
    "            \"name\": \"verificar_tempestade_solar\",\n",
    "            \"description\": \"Verifica se há uma tempestade solar em andamento\",\n",
    "            \"parameters\": {\n",
    "                \"type\": \"object\",\n",
    "                \"properties\": {},\n",
    "                \"required\": [],\n",
    "            },\n",
    "        }\n",
    "    }\n",
    "\n",
    "\n",
    "\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c8b3131c",
   "metadata": {
    "id": "uMPSVniJgN25"
   },
   "outputs": [],
   "source": [
    "# Função para chamar a API com o histórico de mensagens\n",
    "def call_groq_api(messages, model=\"groq/llama3-groq-70b-8192-tool-use-preview\"):\n",
    "    global tools\n",
    "    response = completion(\n",
    "        model=model,\n",
    "        messages=messages,\n",
    "        tools=tools,\n",
    "        tool_choice=\"auto\",\n",
    "        api_key=GROQ_API_KEY,\n",
    "    )\n",
    "    resposta_texto = response.choices[0].message\n",
    "    chamada_ferramentas = resposta_texto.tool_calls\n",
    "    if chamada_ferramentas:\n",
    "      available_functions = {\n",
    "        \"previsao_do_tempo\": previsao_do_tempo,\n",
    "        \"verificar_tempestade_solar\": verificar_tempestade_solar\n",
    "      }\n",
    "      for tool_call in chamada_ferramentas:\n",
    "        function_name = tool_call.function.name\n",
    "        function_to_call = available_functions[function_name]\n",
    "        function_args = json.loads(tool_call.function.arguments)\n",
    "\n",
    "        match function_name:\n",
    "          case \"previsao_do_tempo\":\n",
    "\n",
    "            function_response = function_to_call(\n",
    "                city=function_args.get(\"city\"),\n",
    "                country=function_args.get(\"country\"),\n",
    "            )\n",
    "          case \"verificar_tempestade_solar\":\n",
    "            function_response = function_to_call()\n",
    "        return function_response\n",
    "\n",
    "    else:\n",
    "      return resposta_texto.content"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8f666a31",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "YAP6Jyjthw4a",
    "outputId": "225e83ec-ba30-4c03-8934-b3cbbb33bdb6"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Iniciando chat com o modelo. Digite 'sair' para encerrar.\n",
      "Você: Estamos tendo uma tempestade solar?\n",
      "Assistente: Sem tempestade solar no momento. Índice Kp atual: 1.67\n",
      "Você: Sair\n",
      "Encerrando chat. Até a próxima!\n"
     ]
    }
   ],
   "source": [
    "chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "404bee0b",
   "metadata": {
    "id": "MWac9OMXiReR"
   },
   "source": [
    "## Vídeo 2.3 - Carregando dados com a Pandas"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5d4a1450",
   "metadata": {
    "id": "zI0B354dibxv"
   },
   "outputs": [],
   "source": [
    "import pandas as pd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3715c0b5",
   "metadata": {
    "id": "LpcyEEQNhzhG"
   },
   "outputs": [],
   "source": [
    "def extrair_sismos():\n",
    "    # Fazer a requisição para obter o conteúdo da página\n",
    "    url = 'https://earthquake.usgs.gov/earthquakes/feed/v1.0/summary/4.5_day.csv'\n",
    "\n",
    "    df = pd.read_csv(url)\n",
    "\n",
    "    # Retornar o DataFrame com os dados\n",
    "    return df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4ceb4d76",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "id": "dVWV_VKRiryF",
    "outputId": "1b71337e-51e6-4d53-ce3b-e6070ebf4283"
   },
   "outputs": [
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "dataframe"
      },
      "text/html": [
       "\n",
       "  <div id=\"df-e1f15524-f47d-43c7-a29a-72d2f65a326d\" class=\"colab-df-container\">\n",
       "    <div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>time</th>\n",
       "      <th>latitude</th>\n",
       "      <th>longitude</th>\n",
       "      <th>depth</th>\n",
       "      <th>mag</th>\n",
       "      <th>magType</th>\n",
       "      <th>nst</th>\n",
       "      <th>gap</th>\n",
       "      <th>dmin</th>\n",
       "      <th>rms</th>\n",
       "      <th>...</th>\n",
       "      <th>updated</th>\n",
       "      <th>place</th>\n",
       "      <th>type</th>\n",
       "      <th>horizontalError</th>\n",
       "      <th>depthError</th>\n",
       "      <th>magError</th>\n",
       "      <th>magNst</th>\n",
       "      <th>status</th>\n",
       "      <th>locationSource</th>\n",
       "      <th>magSource</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2024-10-21T16:02:44.673Z</td>\n",
       "      <td>-17.4094</td>\n",
       "      <td>-178.4698</td>\n",
       "      <td>530.519</td>\n",
       "      <td>4.5</td>\n",
       "      <td>mb</td>\n",
       "      <td>61</td>\n",
       "      <td>72</td>\n",
       "      <td>2.265</td>\n",
       "      <td>0.72</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:24:53.040Z</td>\n",
       "      <td>245 km ENE of Levuka, Fiji</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>13.55</td>\n",
       "      <td>9.043</td>\n",
       "      <td>0.066</td>\n",
       "      <td>67</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2024-10-21T15:15:13.441Z</td>\n",
       "      <td>-1.1985</td>\n",
       "      <td>127.4176</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>42</td>\n",
       "      <td>96</td>\n",
       "      <td>1.958</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T15:40:07.040Z</td>\n",
       "      <td>63 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.47</td>\n",
       "      <td>1.914</td>\n",
       "      <td>0.100</td>\n",
       "      <td>30</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2024-10-21T14:11:32.834Z</td>\n",
       "      <td>42.7664</td>\n",
       "      <td>143.6458</td>\n",
       "      <td>117.705</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>98</td>\n",
       "      <td>76</td>\n",
       "      <td>0.834</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:48:45.004Z</td>\n",
       "      <td>4 km S of Honchō, Japan</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>8.04</td>\n",
       "      <td>1.493</td>\n",
       "      <td>0.046</td>\n",
       "      <td>140</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2024-10-21T08:35:08.646Z</td>\n",
       "      <td>-1.0930</td>\n",
       "      <td>127.4696</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>43</td>\n",
       "      <td>83</td>\n",
       "      <td>1.855</td>\n",
       "      <td>0.56</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:53:33.040Z</td>\n",
       "      <td>51 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.88</td>\n",
       "      <td>1.755</td>\n",
       "      <td>0.097</td>\n",
       "      <td>34</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2024-10-21T08:24:06.089Z</td>\n",
       "      <td>-1.1634</td>\n",
       "      <td>127.3904</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.2</td>\n",
       "      <td>mb</td>\n",
       "      <td>68</td>\n",
       "      <td>77</td>\n",
       "      <td>1.922</td>\n",
       "      <td>0.50</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:40:51.040Z</td>\n",
       "      <td>60 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.11</td>\n",
       "      <td>1.865</td>\n",
       "      <td>0.072</td>\n",
       "      <td>63</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2024-10-21T08:08:33.226Z</td>\n",
       "      <td>-1.1854</td>\n",
       "      <td>127.2924</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.9</td>\n",
       "      <td>mb</td>\n",
       "      <td>43</td>\n",
       "      <td>82</td>\n",
       "      <td>1.945</td>\n",
       "      <td>0.59</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:30:52.040Z</td>\n",
       "      <td>65 km SSW of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.74</td>\n",
       "      <td>1.886</td>\n",
       "      <td>0.087</td>\n",
       "      <td>41</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2024-10-21T06:32:35.642Z</td>\n",
       "      <td>-1.0991</td>\n",
       "      <td>127.4181</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.7</td>\n",
       "      <td>mb</td>\n",
       "      <td>72</td>\n",
       "      <td>78</td>\n",
       "      <td>1.859</td>\n",
       "      <td>0.67</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T18:27:22.640Z</td>\n",
       "      <td>52 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.15</td>\n",
       "      <td>1.841</td>\n",
       "      <td>0.049</td>\n",
       "      <td>150</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>2024-10-21T05:35:49.196Z</td>\n",
       "      <td>52.6667</td>\n",
       "      <td>172.4989</td>\n",
       "      <td>43.229</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>114</td>\n",
       "      <td>110</td>\n",
       "      <td>3.794</td>\n",
       "      <td>0.58</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T17:45:55.969Z</td>\n",
       "      <td>50 km WSW of Attu Station, Alaska</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>9.32</td>\n",
       "      <td>7.330</td>\n",
       "      <td>0.029</td>\n",
       "      <td>388</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>2024-10-21T04:12:15.553Z</td>\n",
       "      <td>-1.1187</td>\n",
       "      <td>127.4283</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.2</td>\n",
       "      <td>mww</td>\n",
       "      <td>72</td>\n",
       "      <td>78</td>\n",
       "      <td>1.879</td>\n",
       "      <td>0.68</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T04:31:53.040Z</td>\n",
       "      <td>54 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.80</td>\n",
       "      <td>1.859</td>\n",
       "      <td>0.083</td>\n",
       "      <td>14</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>2024-10-21T04:10:22.686Z</td>\n",
       "      <td>-1.1127</td>\n",
       "      <td>127.4636</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.9</td>\n",
       "      <td>mb</td>\n",
       "      <td>39</td>\n",
       "      <td>106</td>\n",
       "      <td>1.874</td>\n",
       "      <td>1.09</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T04:25:14.040Z</td>\n",
       "      <td>53 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.72</td>\n",
       "      <td>1.871</td>\n",
       "      <td>0.102</td>\n",
       "      <td>30</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>2024-10-21T03:24:15.474Z</td>\n",
       "      <td>-1.1757</td>\n",
       "      <td>127.4170</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.8</td>\n",
       "      <td>mww</td>\n",
       "      <td>75</td>\n",
       "      <td>60</td>\n",
       "      <td>1.935</td>\n",
       "      <td>0.77</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T18:25:45.973Z</td>\n",
       "      <td>61 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.68</td>\n",
       "      <td>1.829</td>\n",
       "      <td>0.055</td>\n",
       "      <td>32</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>2024-10-20T23:22:17.999Z</td>\n",
       "      <td>-12.4790</td>\n",
       "      <td>166.9805</td>\n",
       "      <td>205.772</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>54</td>\n",
       "      <td>53</td>\n",
       "      <td>8.431</td>\n",
       "      <td>0.55</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-20T23:38:25.040Z</td>\n",
       "      <td>166 km NNW of Sola, Vanuatu</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>11.19</td>\n",
       "      <td>6.214</td>\n",
       "      <td>0.083</td>\n",
       "      <td>46</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>2024-10-20T22:59:40.810Z</td>\n",
       "      <td>-30.8936</td>\n",
       "      <td>-68.7474</td>\n",
       "      <td>107.654</td>\n",
       "      <td>5.7</td>\n",
       "      <td>mww</td>\n",
       "      <td>32</td>\n",
       "      <td>89</td>\n",
       "      <td>0.652</td>\n",
       "      <td>0.53</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T01:02:58.107Z</td>\n",
       "      <td>63 km NNW of Albardón, Argentina</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.31</td>\n",
       "      <td>8.017</td>\n",
       "      <td>0.039</td>\n",
       "      <td>63</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>13 rows × 22 columns</p>\n",
       "</div>\n",
       "    <div class=\"colab-df-buttons\">\n",
       "\n",
       "  <div class=\"colab-df-container\">\n",
       "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-e1f15524-f47d-43c7-a29a-72d2f65a326d')\"\n",
       "            title=\"Convert this dataframe to an interactive table.\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
       "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
       "  </svg>\n",
       "    </button>\n",
       "\n",
       "  <style>\n",
       "    .colab-df-container {\n",
       "      display:flex;\n",
       "      gap: 12px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert {\n",
       "      background-color: #E8F0FE;\n",
       "      border: none;\n",
       "      border-radius: 50%;\n",
       "      cursor: pointer;\n",
       "      display: none;\n",
       "      fill: #1967D2;\n",
       "      height: 32px;\n",
       "      padding: 0 0 0 0;\n",
       "      width: 32px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert:hover {\n",
       "      background-color: #E2EBFA;\n",
       "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "      fill: #174EA6;\n",
       "    }\n",
       "\n",
       "    .colab-df-buttons div {\n",
       "      margin-bottom: 4px;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert {\n",
       "      background-color: #3B4455;\n",
       "      fill: #D2E3FC;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert:hover {\n",
       "      background-color: #434B5C;\n",
       "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
       "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
       "      fill: #FFFFFF;\n",
       "    }\n",
       "  </style>\n",
       "\n",
       "    <script>\n",
       "      const buttonEl =\n",
       "        document.querySelector('#df-e1f15524-f47d-43c7-a29a-72d2f65a326d button.colab-df-convert');\n",
       "      buttonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "\n",
       "      async function convertToInteractive(key) {\n",
       "        const element = document.querySelector('#df-e1f15524-f47d-43c7-a29a-72d2f65a326d');\n",
       "        const dataTable =\n",
       "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
       "                                                    [key], {});\n",
       "        if (!dataTable) return;\n",
       "\n",
       "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
       "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
       "          + ' to learn more about interactive tables.';\n",
       "        element.innerHTML = '';\n",
       "        dataTable['output_type'] = 'display_data';\n",
       "        await google.colab.output.renderOutput(dataTable, element);\n",
       "        const docLink = document.createElement('div');\n",
       "        docLink.innerHTML = docLinkHtml;\n",
       "        element.appendChild(docLink);\n",
       "      }\n",
       "    </script>\n",
       "  </div>\n",
       "\n",
       "\n",
       "<div id=\"df-88010c50-bde5-4360-96e2-76cd505a3f3c\">\n",
       "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-88010c50-bde5-4360-96e2-76cd505a3f3c')\"\n",
       "            title=\"Suggest charts\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
       "     width=\"24px\">\n",
       "    <g>\n",
       "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
       "    </g>\n",
       "</svg>\n",
       "  </button>\n",
       "\n",
       "<style>\n",
       "  .colab-df-quickchart {\n",
       "      --bg-color: #E8F0FE;\n",
       "      --fill-color: #1967D2;\n",
       "      --hover-bg-color: #E2EBFA;\n",
       "      --hover-fill-color: #174EA6;\n",
       "      --disabled-fill-color: #AAA;\n",
       "      --disabled-bg-color: #DDD;\n",
       "  }\n",
       "\n",
       "  [theme=dark] .colab-df-quickchart {\n",
       "      --bg-color: #3B4455;\n",
       "      --fill-color: #D2E3FC;\n",
       "      --hover-bg-color: #434B5C;\n",
       "      --hover-fill-color: #FFFFFF;\n",
       "      --disabled-bg-color: #3B4455;\n",
       "      --disabled-fill-color: #666;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart {\n",
       "    background-color: var(--bg-color);\n",
       "    border: none;\n",
       "    border-radius: 50%;\n",
       "    cursor: pointer;\n",
       "    display: none;\n",
       "    fill: var(--fill-color);\n",
       "    height: 32px;\n",
       "    padding: 0;\n",
       "    width: 32px;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart:hover {\n",
       "    background-color: var(--hover-bg-color);\n",
       "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "    fill: var(--button-hover-fill-color);\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart-complete:disabled,\n",
       "  .colab-df-quickchart-complete:disabled:hover {\n",
       "    background-color: var(--disabled-bg-color);\n",
       "    fill: var(--disabled-fill-color);\n",
       "    box-shadow: none;\n",
       "  }\n",
       "\n",
       "  .colab-df-spinner {\n",
       "    border: 2px solid var(--fill-color);\n",
       "    border-color: transparent;\n",
       "    border-bottom-color: var(--fill-color);\n",
       "    animation:\n",
       "      spin 1s steps(1) infinite;\n",
       "  }\n",
       "\n",
       "  @keyframes spin {\n",
       "    0% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "      border-left-color: var(--fill-color);\n",
       "    }\n",
       "    20% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    30% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    40% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    60% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    80% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "    90% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "  }\n",
       "</style>\n",
       "\n",
       "  <script>\n",
       "    async function quickchart(key) {\n",
       "      const quickchartButtonEl =\n",
       "        document.querySelector('#' + key + ' button');\n",
       "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
       "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
       "      try {\n",
       "        const charts = await google.colab.kernel.invokeFunction(\n",
       "            'suggestCharts', [key], {});\n",
       "      } catch (error) {\n",
       "        console.error('Error during call to suggestCharts:', error);\n",
       "      }\n",
       "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
       "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
       "    }\n",
       "    (() => {\n",
       "      let quickchartButtonEl =\n",
       "        document.querySelector('#df-88010c50-bde5-4360-96e2-76cd505a3f3c button');\n",
       "      quickchartButtonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "    })();\n",
       "  </script>\n",
       "</div>\n",
       "\n",
       "    </div>\n",
       "  </div>\n"
      ],
      "text/plain": [
       "                        time  latitude  longitude    depth  mag magType  nst  \\\n",
       "0   2024-10-21T16:02:44.673Z  -17.4094  -178.4698  530.519  4.5      mb   61   \n",
       "1   2024-10-21T15:15:13.441Z   -1.1985   127.4176   10.000  4.6      mb   42   \n",
       "2   2024-10-21T14:11:32.834Z   42.7664   143.6458  117.705  4.6      mb   98   \n",
       "3   2024-10-21T08:35:08.646Z   -1.0930   127.4696   10.000  5.0      mb   43   \n",
       "4   2024-10-21T08:24:06.089Z   -1.1634   127.3904   10.000  5.2      mb   68   \n",
       "5   2024-10-21T08:08:33.226Z   -1.1854   127.2924   10.000  4.9      mb   43   \n",
       "6   2024-10-21T06:32:35.642Z   -1.0991   127.4181   10.000  5.7      mb   72   \n",
       "7   2024-10-21T05:35:49.196Z   52.6667   172.4989   43.229  5.0      mb  114   \n",
       "8   2024-10-21T04:12:15.553Z   -1.1187   127.4283   10.000  5.2     mww   72   \n",
       "9   2024-10-21T04:10:22.686Z   -1.1127   127.4636   10.000  4.9      mb   39   \n",
       "10  2024-10-21T03:24:15.474Z   -1.1757   127.4170   10.000  5.8     mww   75   \n",
       "11  2024-10-20T23:22:17.999Z  -12.4790   166.9805  205.772  5.0      mb   54   \n",
       "12  2024-10-20T22:59:40.810Z  -30.8936   -68.7474  107.654  5.7     mww   32   \n",
       "\n",
       "    gap   dmin   rms  ...                   updated  \\\n",
       "0    72  2.265  0.72  ...  2024-10-21T16:24:53.040Z   \n",
       "1    96  1.958  0.52  ...  2024-10-21T15:40:07.040Z   \n",
       "2    76  0.834  0.52  ...  2024-10-21T16:48:45.004Z   \n",
       "3    83  1.855  0.56  ...  2024-10-21T08:53:33.040Z   \n",
       "4    77  1.922  0.50  ...  2024-10-21T08:40:51.040Z   \n",
       "5    82  1.945  0.59  ...  2024-10-21T08:30:52.040Z   \n",
       "6    78  1.859  0.67  ...  2024-10-21T18:27:22.640Z   \n",
       "7   110  3.794  0.58  ...  2024-10-21T17:45:55.969Z   \n",
       "8    78  1.879  0.68  ...  2024-10-21T04:31:53.040Z   \n",
       "9   106  1.874  1.09  ...  2024-10-21T04:25:14.040Z   \n",
       "10   60  1.935  0.77  ...  2024-10-21T18:25:45.973Z   \n",
       "11   53  8.431  0.55  ...  2024-10-20T23:38:25.040Z   \n",
       "12   89  0.652  0.53  ...  2024-10-21T01:02:58.107Z   \n",
       "\n",
       "                                place        type horizontalError depthError  \\\n",
       "0          245 km ENE of Levuka, Fiji  earthquake           13.55      9.043   \n",
       "1        63 km S of Labuha, Indonesia  earthquake            7.47      1.914   \n",
       "2             4 km S of Honchō, Japan  earthquake            8.04      1.493   \n",
       "3        51 km S of Labuha, Indonesia  earthquake            5.88      1.755   \n",
       "4        60 km S of Labuha, Indonesia  earthquake            7.11      1.865   \n",
       "5      65 km SSW of Labuha, Indonesia  earthquake            6.74      1.886   \n",
       "6        52 km S of Labuha, Indonesia  earthquake            5.15      1.841   \n",
       "7   50 km WSW of Attu Station, Alaska  earthquake            9.32      7.330   \n",
       "8        54 km S of Labuha, Indonesia  earthquake            6.80      1.859   \n",
       "9        53 km S of Labuha, Indonesia  earthquake            7.72      1.871   \n",
       "10       61 km S of Labuha, Indonesia  earthquake            5.68      1.829   \n",
       "11        166 km NNW of Sola, Vanuatu  earthquake           11.19      6.214   \n",
       "12   63 km NNW of Albardón, Argentina  earthquake            6.31      8.017   \n",
       "\n",
       "    magError  magNst    status  locationSource magSource  \n",
       "0      0.066      67  reviewed              us        us  \n",
       "1      0.100      30  reviewed              us        us  \n",
       "2      0.046     140  reviewed              us        us  \n",
       "3      0.097      34  reviewed              us        us  \n",
       "4      0.072      63  reviewed              us        us  \n",
       "5      0.087      41  reviewed              us        us  \n",
       "6      0.049     150  reviewed              us        us  \n",
       "7      0.029     388  reviewed              us        us  \n",
       "8      0.083      14  reviewed              us        us  \n",
       "9      0.102      30  reviewed              us        us  \n",
       "10     0.055      32  reviewed              us        us  \n",
       "11     0.083      46  reviewed              us        us  \n",
       "12     0.039      63  reviewed              us        us  \n",
       "\n",
       "[13 rows x 22 columns]"
      ]
     },
     "execution_count": 57,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "extrair_sismos()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "74fca71b",
   "metadata": {
    "id": "c46_xv1yisCY"
   },
   "outputs": [],
   "source": [
    "tools = [\n",
    "        {\n",
    "            \"type\": \"function\",\n",
    "            \"function\": {\n",
    "                \"name\": \"previsao_do_tempo\",\n",
    "                \"description\": \"Retorna a previsão do tempo em uma cidade específica\",\n",
    "                \"parameters\": {\n",
    "                    \"type\": \"object\",\n",
    "                    \"properties\": {\n",
    "                        \"city\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Nome da cidade\",\n",
    "                        },\n",
    "                        \"country\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Sigla do país\",\n",
    "                        },\n",
    "                    },\n",
    "                    \"required\": [\"city\", \"country\"],\n",
    "                },\n",
    "            }\n",
    "\n",
    "        },\n",
    "\n",
    "      {\n",
    "        \"type\": \"function\",\n",
    "        \"function\": {\n",
    "            \"name\": \"verificar_tempestade_solar\",\n",
    "            \"description\": \"Verifica se há uma tempestade solar em andamento\",\n",
    "            \"parameters\": {\n",
    "                \"type\": \"object\",\n",
    "                \"properties\": {},\n",
    "                \"required\": [],\n",
    "            },\n",
    "        }\n",
    "      },\n",
    "      {\n",
    "        \"type\": \"function\",\n",
    "        \"function\": {\n",
    "            \"name\": \"extrair_sismos\",\n",
    "            \"description\": \"Extrai dados de sismos da USGS\",\n",
    "            \"parameters\": {\n",
    "                \"type\": \"object\",\n",
    "                \"properties\": {},\n",
    "                \"required\": [],\n",
    "            },\n",
    "        }\n",
    "      }\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a5e1f37a",
   "metadata": {
    "id": "ai2QmCVPiwAD"
   },
   "outputs": [],
   "source": [
    "# Função para chamar a API com o histórico de mensagens\n",
    "def call_groq_api(messages, model=\"groq/llama3-groq-70b-8192-tool-use-preview\"):\n",
    "    global tools\n",
    "    response = completion(\n",
    "        model=model,\n",
    "        messages=messages,\n",
    "        tools=tools,\n",
    "        tool_choice=\"auto\",\n",
    "        api_key=GROQ_API_KEY,\n",
    "    )\n",
    "    resposta_texto = response.choices[0].message\n",
    "    chamada_ferramentas = resposta_texto.tool_calls\n",
    "    if chamada_ferramentas:\n",
    "      available_functions = {\n",
    "        \"previsao_do_tempo\": previsao_do_tempo,\n",
    "        \"verificar_tempestade_solar\": verificar_tempestade_solar,\n",
    "        \"extrair_sismos\": extrair_sismos\n",
    "      }\n",
    "      for tool_call in chamada_ferramentas:\n",
    "        function_name = tool_call.function.name\n",
    "        function_to_call = available_functions[function_name]\n",
    "        function_args = json.loads(tool_call.function.arguments)\n",
    "\n",
    "        match function_name:\n",
    "          case \"previsao_do_tempo\":\n",
    "\n",
    "            function_response = function_to_call(\n",
    "                city=function_args.get(\"city\"),\n",
    "                country=function_args.get(\"country\"),\n",
    "            )\n",
    "          case \"verificar_tempestade_solar\":\n",
    "            function_response = function_to_call()\n",
    "          case \"extrair_sismos\":\n",
    "            function_response = function_to_call()\n",
    "        return function_response\n",
    "\n",
    "    else:\n",
    "      return resposta_texto.content"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "635f83be",
   "metadata": {
    "id": "CbmRnVrJnwsW"
   },
   "outputs": [],
   "source": [
    "# Função para iniciar o chat, mantendo o histórico\n",
    "def chat():\n",
    "    print(\"Iniciando chat com o modelo. Digite 'sair' para encerrar.\")\n",
    "\n",
    "    # Histórico de mensagens\n",
    "    messages = [{\"role\": \"system\", \"content\": \"\"\"\n",
    "    Você é o Chat da Terra e do Universo e responde em português brasileiro\n",
    "    perguntas sobre a previsão do tempo na Terra e do espaço próximo à Terra, além de informações sobre terremotos.\n",
    "    \"\"\"}]\n",
    "\n",
    "    while True:\n",
    "        user_message = input(\"Você: \")\n",
    "        if user_message.lower() == \"sair\":\n",
    "            print(\"Encerrando chat. Até a próxima!\")\n",
    "            break\n",
    "\n",
    "        # Adicionar a mensagem do usuário ao histórico\n",
    "        messages.append({\"role\": \"user\", \"content\": user_message})\n",
    "\n",
    "        # Chamar a API com o histórico completo\n",
    "        model_response = call_groq_api(messages)\n",
    "\n",
    "        # Exibir a resposta do assistente\n",
    "        display(model_response)\n",
    "        if isinstance(model_response, pd.DataFrame):\n",
    "            print(\"O model_response é um DataFrame.\")\n",
    "            texto_corrido = \"\"\n",
    "            for index, row in model_response.iterrows():\n",
    "                texto_corrido += f\"Evento {index + 1}: Magnitude {row['mag']}, Local {row['place']}, Tempo {row['time']}\\n\"\n",
    "\n",
    "            model_response = texto_corrido\n",
    "\n",
    "\n",
    "        # Adicionar a resposta do modelo ao histórico\n",
    "        messages.append({\"role\": \"assistant\", \"content\": model_response})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d3ad72c2",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "id": "HPolN8_goccC",
    "outputId": "40556e62-1769-4fbd-e66f-b00b53b90333"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Iniciando chat com o modelo. Digite 'sair' para encerrar.\n",
      "Você: Tivemos muitos sismos recentemente?\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "dataframe"
      },
      "text/html": [
       "\n",
       "  <div id=\"df-63a7b60e-28e5-405f-9fab-998748e06c76\" class=\"colab-df-container\">\n",
       "    <div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>time</th>\n",
       "      <th>latitude</th>\n",
       "      <th>longitude</th>\n",
       "      <th>depth</th>\n",
       "      <th>mag</th>\n",
       "      <th>magType</th>\n",
       "      <th>nst</th>\n",
       "      <th>gap</th>\n",
       "      <th>dmin</th>\n",
       "      <th>rms</th>\n",
       "      <th>...</th>\n",
       "      <th>updated</th>\n",
       "      <th>place</th>\n",
       "      <th>type</th>\n",
       "      <th>horizontalError</th>\n",
       "      <th>depthError</th>\n",
       "      <th>magError</th>\n",
       "      <th>magNst</th>\n",
       "      <th>status</th>\n",
       "      <th>locationSource</th>\n",
       "      <th>magSource</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2024-10-21T19:22:50.440Z</td>\n",
       "      <td>-1.1979</td>\n",
       "      <td>127.4281</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>mb</td>\n",
       "      <td>29</td>\n",
       "      <td>107</td>\n",
       "      <td>1.957</td>\n",
       "      <td>0.85</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T19:49:37.040Z</td>\n",
       "      <td>63 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.91</td>\n",
       "      <td>1.921</td>\n",
       "      <td>0.121</td>\n",
       "      <td>20</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2024-10-21T16:02:44.673Z</td>\n",
       "      <td>-17.4094</td>\n",
       "      <td>-178.4698</td>\n",
       "      <td>530.519</td>\n",
       "      <td>4.5</td>\n",
       "      <td>mb</td>\n",
       "      <td>61</td>\n",
       "      <td>72</td>\n",
       "      <td>2.265</td>\n",
       "      <td>0.72</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:24:53.040Z</td>\n",
       "      <td>245 km ENE of Levuka, Fiji</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>13.55</td>\n",
       "      <td>9.043</td>\n",
       "      <td>0.066</td>\n",
       "      <td>67</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2024-10-21T15:15:13.441Z</td>\n",
       "      <td>-1.1985</td>\n",
       "      <td>127.4176</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>42</td>\n",
       "      <td>96</td>\n",
       "      <td>1.958</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T15:40:07.040Z</td>\n",
       "      <td>63 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.47</td>\n",
       "      <td>1.914</td>\n",
       "      <td>0.100</td>\n",
       "      <td>30</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2024-10-21T14:11:32.834Z</td>\n",
       "      <td>42.7664</td>\n",
       "      <td>143.6458</td>\n",
       "      <td>117.705</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>98</td>\n",
       "      <td>76</td>\n",
       "      <td>0.834</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:48:45.004Z</td>\n",
       "      <td>4 km S of Honchō, Japan</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>8.04</td>\n",
       "      <td>1.493</td>\n",
       "      <td>0.046</td>\n",
       "      <td>140</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2024-10-21T08:35:08.646Z</td>\n",
       "      <td>-1.0930</td>\n",
       "      <td>127.4696</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>43</td>\n",
       "      <td>83</td>\n",
       "      <td>1.855</td>\n",
       "      <td>0.56</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:53:33.040Z</td>\n",
       "      <td>51 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.88</td>\n",
       "      <td>1.755</td>\n",
       "      <td>0.097</td>\n",
       "      <td>34</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2024-10-21T08:24:06.089Z</td>\n",
       "      <td>-1.1634</td>\n",
       "      <td>127.3904</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.2</td>\n",
       "      <td>mb</td>\n",
       "      <td>68</td>\n",
       "      <td>77</td>\n",
       "      <td>1.922</td>\n",
       "      <td>0.50</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:40:51.040Z</td>\n",
       "      <td>60 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.11</td>\n",
       "      <td>1.865</td>\n",
       "      <td>0.072</td>\n",
       "      <td>63</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2024-10-21T08:08:33.226Z</td>\n",
       "      <td>-1.1854</td>\n",
       "      <td>127.2924</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.9</td>\n",
       "      <td>mb</td>\n",
       "      <td>43</td>\n",
       "      <td>82</td>\n",
       "      <td>1.945</td>\n",
       "      <td>0.59</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T08:30:52.040Z</td>\n",
       "      <td>65 km SSW of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.74</td>\n",
       "      <td>1.886</td>\n",
       "      <td>0.087</td>\n",
       "      <td>41</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>2024-10-21T06:32:35.642Z</td>\n",
       "      <td>-1.0991</td>\n",
       "      <td>127.4181</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.7</td>\n",
       "      <td>mb</td>\n",
       "      <td>72</td>\n",
       "      <td>78</td>\n",
       "      <td>1.859</td>\n",
       "      <td>0.67</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T18:27:22.640Z</td>\n",
       "      <td>52 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.15</td>\n",
       "      <td>1.841</td>\n",
       "      <td>0.049</td>\n",
       "      <td>150</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>2024-10-21T05:35:49.196Z</td>\n",
       "      <td>52.6667</td>\n",
       "      <td>172.4989</td>\n",
       "      <td>43.229</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>114</td>\n",
       "      <td>110</td>\n",
       "      <td>3.794</td>\n",
       "      <td>0.58</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T17:45:55.969Z</td>\n",
       "      <td>50 km WSW of Attu Station, Alaska</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>9.32</td>\n",
       "      <td>7.330</td>\n",
       "      <td>0.029</td>\n",
       "      <td>388</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>2024-10-21T04:12:15.553Z</td>\n",
       "      <td>-1.1187</td>\n",
       "      <td>127.4283</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.2</td>\n",
       "      <td>mww</td>\n",
       "      <td>72</td>\n",
       "      <td>78</td>\n",
       "      <td>1.879</td>\n",
       "      <td>0.68</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T04:31:53.040Z</td>\n",
       "      <td>54 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.80</td>\n",
       "      <td>1.859</td>\n",
       "      <td>0.083</td>\n",
       "      <td>14</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>2024-10-21T04:10:22.686Z</td>\n",
       "      <td>-1.1127</td>\n",
       "      <td>127.4636</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.9</td>\n",
       "      <td>mb</td>\n",
       "      <td>39</td>\n",
       "      <td>106</td>\n",
       "      <td>1.874</td>\n",
       "      <td>1.09</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T04:25:14.040Z</td>\n",
       "      <td>53 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.72</td>\n",
       "      <td>1.871</td>\n",
       "      <td>0.102</td>\n",
       "      <td>30</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>2024-10-21T03:24:15.474Z</td>\n",
       "      <td>-1.1757</td>\n",
       "      <td>127.4170</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.8</td>\n",
       "      <td>mww</td>\n",
       "      <td>75</td>\n",
       "      <td>60</td>\n",
       "      <td>1.935</td>\n",
       "      <td>0.77</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T18:25:45.973Z</td>\n",
       "      <td>61 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>5.68</td>\n",
       "      <td>1.829</td>\n",
       "      <td>0.055</td>\n",
       "      <td>32</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>2024-10-20T23:22:17.999Z</td>\n",
       "      <td>-12.4790</td>\n",
       "      <td>166.9805</td>\n",
       "      <td>205.772</td>\n",
       "      <td>5.0</td>\n",
       "      <td>mb</td>\n",
       "      <td>54</td>\n",
       "      <td>53</td>\n",
       "      <td>8.431</td>\n",
       "      <td>0.55</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-20T23:38:25.040Z</td>\n",
       "      <td>166 km NNW of Sola, Vanuatu</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>11.19</td>\n",
       "      <td>6.214</td>\n",
       "      <td>0.083</td>\n",
       "      <td>46</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>2024-10-20T22:59:40.810Z</td>\n",
       "      <td>-30.8936</td>\n",
       "      <td>-68.7474</td>\n",
       "      <td>107.654</td>\n",
       "      <td>5.7</td>\n",
       "      <td>mww</td>\n",
       "      <td>32</td>\n",
       "      <td>89</td>\n",
       "      <td>0.652</td>\n",
       "      <td>0.53</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T01:02:58.107Z</td>\n",
       "      <td>63 km NNW of Albardón, Argentina</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>6.31</td>\n",
       "      <td>8.017</td>\n",
       "      <td>0.039</td>\n",
       "      <td>63</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>14 rows × 22 columns</p>\n",
       "</div>\n",
       "    <div class=\"colab-df-buttons\">\n",
       "\n",
       "  <div class=\"colab-df-container\">\n",
       "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-63a7b60e-28e5-405f-9fab-998748e06c76')\"\n",
       "            title=\"Convert this dataframe to an interactive table.\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
       "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
       "  </svg>\n",
       "    </button>\n",
       "\n",
       "  <style>\n",
       "    .colab-df-container {\n",
       "      display:flex;\n",
       "      gap: 12px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert {\n",
       "      background-color: #E8F0FE;\n",
       "      border: none;\n",
       "      border-radius: 50%;\n",
       "      cursor: pointer;\n",
       "      display: none;\n",
       "      fill: #1967D2;\n",
       "      height: 32px;\n",
       "      padding: 0 0 0 0;\n",
       "      width: 32px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert:hover {\n",
       "      background-color: #E2EBFA;\n",
       "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "      fill: #174EA6;\n",
       "    }\n",
       "\n",
       "    .colab-df-buttons div {\n",
       "      margin-bottom: 4px;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert {\n",
       "      background-color: #3B4455;\n",
       "      fill: #D2E3FC;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert:hover {\n",
       "      background-color: #434B5C;\n",
       "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
       "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
       "      fill: #FFFFFF;\n",
       "    }\n",
       "  </style>\n",
       "\n",
       "    <script>\n",
       "      const buttonEl =\n",
       "        document.querySelector('#df-63a7b60e-28e5-405f-9fab-998748e06c76 button.colab-df-convert');\n",
       "      buttonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "\n",
       "      async function convertToInteractive(key) {\n",
       "        const element = document.querySelector('#df-63a7b60e-28e5-405f-9fab-998748e06c76');\n",
       "        const dataTable =\n",
       "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
       "                                                    [key], {});\n",
       "        if (!dataTable) return;\n",
       "\n",
       "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
       "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
       "          + ' to learn more about interactive tables.';\n",
       "        element.innerHTML = '';\n",
       "        dataTable['output_type'] = 'display_data';\n",
       "        await google.colab.output.renderOutput(dataTable, element);\n",
       "        const docLink = document.createElement('div');\n",
       "        docLink.innerHTML = docLinkHtml;\n",
       "        element.appendChild(docLink);\n",
       "      }\n",
       "    </script>\n",
       "  </div>\n",
       "\n",
       "\n",
       "<div id=\"df-413a28ac-f40d-45c3-a0f2-f217f2d0e216\">\n",
       "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-413a28ac-f40d-45c3-a0f2-f217f2d0e216')\"\n",
       "            title=\"Suggest charts\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
       "     width=\"24px\">\n",
       "    <g>\n",
       "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
       "    </g>\n",
       "</svg>\n",
       "  </button>\n",
       "\n",
       "<style>\n",
       "  .colab-df-quickchart {\n",
       "      --bg-color: #E8F0FE;\n",
       "      --fill-color: #1967D2;\n",
       "      --hover-bg-color: #E2EBFA;\n",
       "      --hover-fill-color: #174EA6;\n",
       "      --disabled-fill-color: #AAA;\n",
       "      --disabled-bg-color: #DDD;\n",
       "  }\n",
       "\n",
       "  [theme=dark] .colab-df-quickchart {\n",
       "      --bg-color: #3B4455;\n",
       "      --fill-color: #D2E3FC;\n",
       "      --hover-bg-color: #434B5C;\n",
       "      --hover-fill-color: #FFFFFF;\n",
       "      --disabled-bg-color: #3B4455;\n",
       "      --disabled-fill-color: #666;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart {\n",
       "    background-color: var(--bg-color);\n",
       "    border: none;\n",
       "    border-radius: 50%;\n",
       "    cursor: pointer;\n",
       "    display: none;\n",
       "    fill: var(--fill-color);\n",
       "    height: 32px;\n",
       "    padding: 0;\n",
       "    width: 32px;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart:hover {\n",
       "    background-color: var(--hover-bg-color);\n",
       "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "    fill: var(--button-hover-fill-color);\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart-complete:disabled,\n",
       "  .colab-df-quickchart-complete:disabled:hover {\n",
       "    background-color: var(--disabled-bg-color);\n",
       "    fill: var(--disabled-fill-color);\n",
       "    box-shadow: none;\n",
       "  }\n",
       "\n",
       "  .colab-df-spinner {\n",
       "    border: 2px solid var(--fill-color);\n",
       "    border-color: transparent;\n",
       "    border-bottom-color: var(--fill-color);\n",
       "    animation:\n",
       "      spin 1s steps(1) infinite;\n",
       "  }\n",
       "\n",
       "  @keyframes spin {\n",
       "    0% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "      border-left-color: var(--fill-color);\n",
       "    }\n",
       "    20% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    30% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    40% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    60% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    80% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "    90% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "  }\n",
       "</style>\n",
       "\n",
       "  <script>\n",
       "    async function quickchart(key) {\n",
       "      const quickchartButtonEl =\n",
       "        document.querySelector('#' + key + ' button');\n",
       "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
       "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
       "      try {\n",
       "        const charts = await google.colab.kernel.invokeFunction(\n",
       "            'suggestCharts', [key], {});\n",
       "      } catch (error) {\n",
       "        console.error('Error during call to suggestCharts:', error);\n",
       "      }\n",
       "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
       "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
       "    }\n",
       "    (() => {\n",
       "      let quickchartButtonEl =\n",
       "        document.querySelector('#df-413a28ac-f40d-45c3-a0f2-f217f2d0e216 button');\n",
       "      quickchartButtonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "    })();\n",
       "  </script>\n",
       "</div>\n",
       "\n",
       "    </div>\n",
       "  </div>\n"
      ],
      "text/plain": [
       "                        time  latitude  longitude    depth  mag magType  nst  \\\n",
       "0   2024-10-21T19:22:50.440Z   -1.1979   127.4281   10.000  4.5      mb   29   \n",
       "1   2024-10-21T16:02:44.673Z  -17.4094  -178.4698  530.519  4.5      mb   61   \n",
       "2   2024-10-21T15:15:13.441Z   -1.1985   127.4176   10.000  4.6      mb   42   \n",
       "3   2024-10-21T14:11:32.834Z   42.7664   143.6458  117.705  4.6      mb   98   \n",
       "4   2024-10-21T08:35:08.646Z   -1.0930   127.4696   10.000  5.0      mb   43   \n",
       "5   2024-10-21T08:24:06.089Z   -1.1634   127.3904   10.000  5.2      mb   68   \n",
       "6   2024-10-21T08:08:33.226Z   -1.1854   127.2924   10.000  4.9      mb   43   \n",
       "7   2024-10-21T06:32:35.642Z   -1.0991   127.4181   10.000  5.7      mb   72   \n",
       "8   2024-10-21T05:35:49.196Z   52.6667   172.4989   43.229  5.0      mb  114   \n",
       "9   2024-10-21T04:12:15.553Z   -1.1187   127.4283   10.000  5.2     mww   72   \n",
       "10  2024-10-21T04:10:22.686Z   -1.1127   127.4636   10.000  4.9      mb   39   \n",
       "11  2024-10-21T03:24:15.474Z   -1.1757   127.4170   10.000  5.8     mww   75   \n",
       "12  2024-10-20T23:22:17.999Z  -12.4790   166.9805  205.772  5.0      mb   54   \n",
       "13  2024-10-20T22:59:40.810Z  -30.8936   -68.7474  107.654  5.7     mww   32   \n",
       "\n",
       "    gap   dmin   rms  ...                   updated  \\\n",
       "0   107  1.957  0.85  ...  2024-10-21T19:49:37.040Z   \n",
       "1    72  2.265  0.72  ...  2024-10-21T16:24:53.040Z   \n",
       "2    96  1.958  0.52  ...  2024-10-21T15:40:07.040Z   \n",
       "3    76  0.834  0.52  ...  2024-10-21T16:48:45.004Z   \n",
       "4    83  1.855  0.56  ...  2024-10-21T08:53:33.040Z   \n",
       "5    77  1.922  0.50  ...  2024-10-21T08:40:51.040Z   \n",
       "6    82  1.945  0.59  ...  2024-10-21T08:30:52.040Z   \n",
       "7    78  1.859  0.67  ...  2024-10-21T18:27:22.640Z   \n",
       "8   110  3.794  0.58  ...  2024-10-21T17:45:55.969Z   \n",
       "9    78  1.879  0.68  ...  2024-10-21T04:31:53.040Z   \n",
       "10  106  1.874  1.09  ...  2024-10-21T04:25:14.040Z   \n",
       "11   60  1.935  0.77  ...  2024-10-21T18:25:45.973Z   \n",
       "12   53  8.431  0.55  ...  2024-10-20T23:38:25.040Z   \n",
       "13   89  0.652  0.53  ...  2024-10-21T01:02:58.107Z   \n",
       "\n",
       "                                place        type horizontalError depthError  \\\n",
       "0        63 km S of Labuha, Indonesia  earthquake            7.91      1.921   \n",
       "1          245 km ENE of Levuka, Fiji  earthquake           13.55      9.043   \n",
       "2        63 km S of Labuha, Indonesia  earthquake            7.47      1.914   \n",
       "3             4 km S of Honchō, Japan  earthquake            8.04      1.493   \n",
       "4        51 km S of Labuha, Indonesia  earthquake            5.88      1.755   \n",
       "5        60 km S of Labuha, Indonesia  earthquake            7.11      1.865   \n",
       "6      65 km SSW of Labuha, Indonesia  earthquake            6.74      1.886   \n",
       "7        52 km S of Labuha, Indonesia  earthquake            5.15      1.841   \n",
       "8   50 km WSW of Attu Station, Alaska  earthquake            9.32      7.330   \n",
       "9        54 km S of Labuha, Indonesia  earthquake            6.80      1.859   \n",
       "10       53 km S of Labuha, Indonesia  earthquake            7.72      1.871   \n",
       "11       61 km S of Labuha, Indonesia  earthquake            5.68      1.829   \n",
       "12        166 km NNW of Sola, Vanuatu  earthquake           11.19      6.214   \n",
       "13   63 km NNW of Albardón, Argentina  earthquake            6.31      8.017   \n",
       "\n",
       "    magError  magNst    status  locationSource magSource  \n",
       "0      0.121      20  reviewed              us        us  \n",
       "1      0.066      67  reviewed              us        us  \n",
       "2      0.100      30  reviewed              us        us  \n",
       "3      0.046     140  reviewed              us        us  \n",
       "4      0.097      34  reviewed              us        us  \n",
       "5      0.072      63  reviewed              us        us  \n",
       "6      0.087      41  reviewed              us        us  \n",
       "7      0.049     150  reviewed              us        us  \n",
       "8      0.029     388  reviewed              us        us  \n",
       "9      0.083      14  reviewed              us        us  \n",
       "10     0.102      30  reviewed              us        us  \n",
       "11     0.055      32  reviewed              us        us  \n",
       "12     0.083      46  reviewed              us        us  \n",
       "13     0.039      63  reviewed              us        us  \n",
       "\n",
       "[14 rows x 22 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "O model_response é um DataFrame.\n",
      "Você: Qual desses tem a maior magnitude?\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'O evento com a maior magnitude é o Evento 14, com magnitude 5.7, localizado 63 km NNW de Albardón, Argentina.'"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Você: Sair\n",
      "Encerrando chat. Até a próxima!\n"
     ]
    }
   ],
   "source": [
    "chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "766bee87",
   "metadata": {
    "id": "0q7_GiB-pNvJ"
   },
   "source": [
    "# Aula 3 - Adicionando uma base de conhecimento"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bb42dd2a",
   "metadata": {
    "id": "x4GXlQ36mEEH"
   },
   "source": [
    "## Vídeo 3.1 - Criando um index com o Pinecone"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "59039f04",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "UrWHBbg5o3cL",
    "outputId": "b6a4ee81-3429-410c-df70-17d7df67e522"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting pinecone\n",
      "  Downloading pinecone-5.3.1-py3-none-any.whl.metadata (19 kB)\n",
      "Requirement already satisfied: certifi>=2019.11.17 in /usr/local/lib/python3.10/dist-packages (from pinecone) (2024.8.30)\n",
      "Collecting pinecone-plugin-inference<2.0.0,>=1.1.0 (from pinecone)\n",
      "  Downloading pinecone_plugin_inference-1.1.0-py3-none-any.whl.metadata (2.2 kB)\n",
      "Collecting pinecone-plugin-interface<0.0.8,>=0.0.7 (from pinecone)\n",
      "  Downloading pinecone_plugin_interface-0.0.7-py3-none-any.whl.metadata (1.2 kB)\n",
      "Requirement already satisfied: python-dateutil>=2.5.3 in /usr/local/lib/python3.10/dist-packages (from pinecone) (2.8.2)\n",
      "Requirement already satisfied: tqdm>=4.64.1 in /usr/local/lib/python3.10/dist-packages (from pinecone) (4.66.5)\n",
      "Requirement already satisfied: typing-extensions>=3.7.4 in /usr/local/lib/python3.10/dist-packages (from pinecone) (4.12.2)\n",
      "Requirement already satisfied: urllib3>=1.26.0 in /usr/local/lib/python3.10/dist-packages (from pinecone) (2.2.3)\n",
      "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.5.3->pinecone) (1.16.0)\n",
      "Downloading pinecone-5.3.1-py3-none-any.whl (419 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m419.8/419.8 kB\u001b[0m \u001b[31m13.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading pinecone_plugin_inference-1.1.0-py3-none-any.whl (85 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m85.4/85.4 kB\u001b[0m \u001b[31m6.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading pinecone_plugin_interface-0.0.7-py3-none-any.whl (6.2 kB)\n",
      "Installing collected packages: pinecone-plugin-interface, pinecone-plugin-inference, pinecone\n",
      "Successfully installed pinecone-5.3.1 pinecone-plugin-inference-1.1.0 pinecone-plugin-interface-0.0.7\n"
     ]
    }
   ],
   "source": [
    "!pip install pinecone"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d10d3ef8",
   "metadata": {
    "id": "G7d_V8WNplqD"
   },
   "outputs": [],
   "source": [
    "PINECONE_API = userdata.get('PINECONE_API')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5be1ba12",
   "metadata": {
    "id": "kGS8vRvMpnUk"
   },
   "outputs": [],
   "source": [
    "from pinecone import Pinecone\n",
    "\n",
    "pc = Pinecone(api_key=PINECONE_API)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fffd6c43",
   "metadata": {
    "id": "l7iDWFUApo0W"
   },
   "outputs": [],
   "source": [
    "data = [\n",
    "    {\n",
    "        \"id\": \"occurrence1\",\n",
    "        \"text\": \"Ouro presente em veios de quartzo em uma formação hidrotermal. Observa-se alta concentração de ouro em zonas de fraturas e falhas.\"\n",
    "    },\n",
    "    {\n",
    "        \"id\": \"occurrence2\",\n",
    "        \"text\": \"Associação do ouro com sulfetos, especialmente pirita e arsenopirita, em ambiente de rochas metavulcânicas. Indica potencial para depósito orogênico de ouro.\"\n",
    "    },\n",
    "    {\n",
    "        \"id\": \"occurrence3\",\n",
    "        \"text\": \"Ouro aluvial encontrado em depósitos de cascalho próximo a rios e córregos. Indica transporte e concentração secundária de ouro.\"\n",
    "    },\n",
    "    {\n",
    "        \"id\": \"occurrence4\",\n",
    "        \"text\": \"Presença de ouro em formações de skarn associadas a intrusões ígneas graníticas. Indica formação relacionada a processos de metamorfismo de contato.\"\n",
    "    },\n",
    "    {\n",
    "        \"id\": \"occurrence5\",\n",
    "        \"text\": \"Ouro disseminado em formações sedimentares de origem marinha, em conglomerados ricos em minerais pesados. Potencial para depósito do tipo placer ou paleoplacer.\"\n",
    "    },\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f55c0d5a",
   "metadata": {
    "id": "KGa4u3AOpqD_"
   },
   "outputs": [],
   "source": [
    "index = pc.Index(\"geologia\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4ed6aa71",
   "metadata": {
    "id": "PT-AiSfmpwPp"
   },
   "outputs": [],
   "source": [
    "embeddings = pc.inference.embed(\n",
    "    \"multilingual-e5-large\",\n",
    "    inputs=[d['text'] for d in data],\n",
    "    parameters={\n",
    "        \"input_type\": \"passage\"\n",
    "    }\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "036b4a4f",
   "metadata": {
    "id": "2raVAjoBpxX1"
   },
   "outputs": [],
   "source": [
    "vectors = []\n",
    "for d, e in zip(data, embeddings):\n",
    "    vectors.append({\n",
    "        \"id\": d['id'],\n",
    "        \"values\": e['values'],\n",
    "        \"metadata\": {'text': d['text']}\n",
    "    })\n",
    "\n",
    "index.upsert(\n",
    "    vectors=vectors,\n",
    "    namespace=\"ns1\"\n",
    ")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "1ec6c03a",
   "metadata": {
    "id": "20w9z96Pp08d"
   },
   "source": [
    "## Vídeo 3.2 - Perguntando sobre os dados"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c11406e5",
   "metadata": {
    "id": "6VgFpKzhp4Yb"
   },
   "outputs": [],
   "source": [
    "query = \"o ouro ocorre em sedimentos de origem marinha?\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fc6f1d0e",
   "metadata": {
    "id": "VXfMbVF3rSvF"
   },
   "outputs": [],
   "source": [
    "x = pc.inference.embed(\n",
    "    model=\"multilingual-e5-large\",\n",
    "    inputs=[query],\n",
    "    parameters={\n",
    "        \"input_type\": \"query\"\n",
    "    }\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1a88f314",
   "metadata": {
    "id": "_Bl1lVvVrUjQ"
   },
   "outputs": [],
   "source": [
    "results = index.query(\n",
    "    namespace=\"ns1\",\n",
    "    vector=x[0].values,\n",
    "    top_k=3,\n",
    "    include_values=False,\n",
    "    include_metadata=True\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "807b52f4",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "Z5Nhkug_rWu5",
    "outputId": "8cecacd7-6eec-4b48-a635-38c1ed17372a"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'matches': [{'id': 'occurrence5',\n",
       "              'metadata': {'text': 'Ouro disseminado em formações sedimentares '\n",
       "                                   'de origem marinha, em conglomerados ricos '\n",
       "                                   'em minerais pesados. Potencial para '\n",
       "                                   'depósito do tipo placer ou paleoplacer.'},\n",
       "              'score': 0.901498914,\n",
       "              'values': []},\n",
       "             {'id': 'occurrence1',\n",
       "              'metadata': {'text': 'Ouro presente em veios de quartzo em uma '\n",
       "                                   'formação hidrotermal. Observa-se alta '\n",
       "                                   'concentração de ouro em zonas de fraturas '\n",
       "                                   'e falhas.'},\n",
       "              'score': 0.869495034,\n",
       "              'values': []},\n",
       "             {'id': 'occurrence3',\n",
       "              'metadata': {'text': 'Ouro aluvial encontrado em depósitos de '\n",
       "                                   'cascalho próximo a rios e córregos. Indica '\n",
       "                                   'transporte e concentração secundária de '\n",
       "                                   'ouro.'},\n",
       "              'score': 0.867000878,\n",
       "              'values': []}],\n",
       " 'namespace': 'ns1',\n",
       " 'usage': {'read_units': 6}}"
      ]
     },
     "execution_count": 68,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "results"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "cdc7f19a",
   "metadata": {
    "id": "hdr6CpZasHUa"
   },
   "source": [
    "## Vídeo 3.3 - Carregando um livro"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d0af29e5",
   "metadata": {
    "id": "p75btjQOrYmP"
   },
   "outputs": [],
   "source": [
    "!pip install pypdf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "080fc640",
   "metadata": {
    "id": "e3BsQZI5sPJk"
   },
   "outputs": [],
   "source": [
    "import pypdf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8e24d6a4",
   "metadata": {
    "id": "5Z2gg3OJsQlf"
   },
   "outputs": [],
   "source": [
    "# Função para dividir o texto em chunks\n",
    "def chunk_text(text, chunk_size=500):\n",
    "    # Divide o texto em partes menores\n",
    "    return [text[i:i+chunk_size] for i in range(0, len(text), chunk_size)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "04fc752d",
   "metadata": {
    "id": "4DuvbRaysSEX"
   },
   "outputs": [],
   "source": [
    "# Abrir e ler o arquivo PDF\n",
    "with open(\"livro.pdf\", \"rb\") as file:\n",
    "    reader = pypdf.PdfReader(file)\n",
    "    text = \"\"\n",
    "\n",
    "    # Itera por todas as páginas do PDF\n",
    "    for page_num in range(len(reader.pages)):\n",
    "        page = reader.pages[page_num]\n",
    "        text += page.extract_text()  # Extrai o texto de cada página"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fd1c2a08",
   "metadata": {
    "id": "2TDYfq-8sTRR"
   },
   "outputs": [],
   "source": [
    "# Dividir o texto em chunks de 500 caracteres\n",
    "chunks = chunk_text(text)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a83fef8b",
   "metadata": {
    "id": "5sTdWMMSsUqP"
   },
   "outputs": [],
   "source": [
    "chunks[1412]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "26d16f87",
   "metadata": {
    "id": "svgXE_vXsWMu"
   },
   "outputs": [],
   "source": [
    "# Formatar os chunks no formato desejado\n",
    "data = []\n",
    "for i, chunk in enumerate(chunks[:90], start=1):\n",
    "    data.append({\n",
    "        \"id\": f\"chunk{i}\",\n",
    "        \"text\": chunk.strip()  # Remove espaços em branco desnecessários\n",
    "    })\n",
    "\n",
    "# Exibindo os primeiros 5 chunks como exemplo\n",
    "for chunk in data[:5]:\n",
    "    print(chunk)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "26fa4d57",
   "metadata": {
    "id": "h0xi-vOnsYlg"
   },
   "outputs": [],
   "source": [
    "embeddings = pc.inference.embed(\n",
    "    \"multilingual-e5-large\",\n",
    "    inputs=[d['text'] for d in data],\n",
    "    parameters={\n",
    "        \"input_type\": \"passage\"\n",
    "    }\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3c7e2f73",
   "metadata": {
    "id": "2x9aLuC3sZt8"
   },
   "outputs": [],
   "source": [
    "vectors = []\n",
    "for d, e in zip(data, embeddings):\n",
    "    vectors.append({\n",
    "        \"id\": d['id'],\n",
    "        \"values\": e['values'],\n",
    "        \"metadata\": {'text': d['text']}\n",
    "    })\n",
    "\n",
    "index.upsert(\n",
    "    vectors=vectors,\n",
    "    namespace=\"ns1\"\n",
    ")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "c0d00086",
   "metadata": {
    "id": "5DO927gVsazT"
   },
   "source": [
    "# Aula 4 - Utilizando o conhecimento no chat"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "6156d370",
   "metadata": {
    "id": "et2n2Zrwsmf9"
   },
   "source": [
    "## Vídeo 4.1 Chamando o modelo uma segunda vez"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "42b13690",
   "metadata": {
    "id": "vzlTEDdUsfD-"
   },
   "outputs": [],
   "source": [
    "def info_geologia(query):\n",
    "    x = pc.inference.embed(\n",
    "        model=\"multilingual-e5-large\",\n",
    "        inputs=[query],\n",
    "        parameters={\n",
    "            \"input_type\": \"query\"\n",
    "        }\n",
    "    )\n",
    "    results = index.query(\n",
    "        namespace=\"ns1\",\n",
    "        vector=x[0].values,\n",
    "        top_k=3,\n",
    "        include_values=False,\n",
    "        include_metadata=True\n",
    "    )\n",
    "    return results"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "010eef0a",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "oC547YVVulhu",
    "outputId": "2954f590-6906-4654-9548-c25b4a57572b"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'matches': [{'id': 'occurrence5',\n",
       "              'metadata': {'text': 'Ouro disseminado em formações sedimentares '\n",
       "                                   'de origem marinha, em conglomerados ricos '\n",
       "                                   'em minerais pesados. Potencial para '\n",
       "                                   'depósito do tipo placer ou paleoplacer.'},\n",
       "              'score': 0.901498914,\n",
       "              'values': []},\n",
       "             {'id': 'occurrence1',\n",
       "              'metadata': {'text': 'Ouro presente em veios de quartzo em uma '\n",
       "                                   'formação hidrotermal. Observa-se alta '\n",
       "                                   'concentração de ouro em zonas de fraturas '\n",
       "                                   'e falhas.'},\n",
       "              'score': 0.869495034,\n",
       "              'values': []},\n",
       "             {'id': 'occurrence3',\n",
       "              'metadata': {'text': 'Ouro aluvial encontrado em depósitos de '\n",
       "                                   'cascalho próximo a rios e córregos. Indica '\n",
       "                                   'transporte e concentração secundária de '\n",
       "                                   'ouro.'},\n",
       "              'score': 0.867000878,\n",
       "              'values': []}],\n",
       " 'namespace': 'ns1',\n",
       " 'usage': {'read_units': 6}}"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "query = \"o ouro ocorre em sedimentos de origem marinha?\"\n",
    "info_geologia(query)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3497df18",
   "metadata": {
    "id": "NnK4hbU2laoU"
   },
   "outputs": [],
   "source": [
    "resposta = info_geologia(query)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c354fbdc",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 35
    },
    "id": "I5iIDfSflfT9",
    "outputId": "29cfb7fd-100c-4a97-bcc6-39d70a8ac256"
   },
   "outputs": [
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'Ouro disseminado em formações sedimentares de origem marinha, em conglomerados ricos em minerais pesados. Potencial para depósito do tipo placer ou paleoplacer.'"
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "resposta.matches[0].metadata['text']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "db3dd1a9",
   "metadata": {
    "id": "uOwuA7U-zaLR"
   },
   "outputs": [],
   "source": [
    "# Função para iniciar o chat, mantendo o histórico\n",
    "def chat():\n",
    "    print(\"Iniciando chat com o modelo. Digite 'sair' para encerrar.\")\n",
    "\n",
    "    # Histórico de mensagens\n",
    "    messages = [{\"role\": \"system\", \"content\": \"\"\"\n",
    "    Você é o Chat da Terra e do Universo e responde em português brasileiro\n",
    "    perguntas sobre a previsão do tempo na Terra e do espaço próximo à Terra, além de informações sobre terremotos.\n",
    "    \"\"\"}]\n",
    "\n",
    "    while True:\n",
    "        user_message = input(\"Você: \")\n",
    "        if user_message.lower() == \"sair\":\n",
    "            print(\"Encerrando chat. Até a próxima!\")\n",
    "            break\n",
    "\n",
    "        # Adicionar a mensagem do usuário ao histórico\n",
    "        messages.append({\"role\": \"user\", \"content\": user_message})\n",
    "\n",
    "        # Verificar se o tema é geologia\n",
    "        if \"geologia\" in user_message.lower():\n",
    "            # Chamar a função info_geologia para obter informações adicionais\n",
    "            resposta = info_geologia(user_message)\n",
    "\n",
    "            # Adicionar as informações de geologia ao histórico como contexto extra\n",
    "            geologia_info = f\"Informações adicionais sobre geologia: {resposta.matches[0].metadata['text']}\"\n",
    "            messages.append({\"role\": \"system\", \"content\": geologia_info})\n",
    "\n",
    "        # Chamar a API com o histórico completo\n",
    "        model_response = call_groq_api(messages)\n",
    "\n",
    "        # Exibir a resposta do assistente\n",
    "        display(model_response)\n",
    "        if isinstance(model_response, pd.DataFrame):\n",
    "            print(\"O model_response é um DataFrame.\")\n",
    "            texto_corrido = \"\"\n",
    "            for index, row in model_response.iterrows():\n",
    "                texto_corrido += f\"Evento {index + 1}: Magnitude {row['mag']}, Local {row['place']}, Tempo {row['time']}\\n\"\n",
    "\n",
    "            model_response = texto_corrido\n",
    "\n",
    "        # Adicionar a resposta do modelo ao histórico\n",
    "        messages.append({\"role\": \"assistant\", \"content\": model_response})\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3bc3424e",
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 510
    },
    "id": "keBHvBLekhv-",
    "outputId": "878507e6-361b-47ce-af90-16363b1e8f34"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Iniciando chat com o modelo. Digite 'sair' para encerrar.\n",
      "Você: Quais são os terremotos mais recentes?\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "dataframe"
      },
      "text/html": [
       "\n",
       "  <div id=\"df-7be44c8b-e8aa-49e4-9c84-dca3e48bec3d\" class=\"colab-df-container\">\n",
       "    <div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>time</th>\n",
       "      <th>latitude</th>\n",
       "      <th>longitude</th>\n",
       "      <th>depth</th>\n",
       "      <th>mag</th>\n",
       "      <th>magType</th>\n",
       "      <th>nst</th>\n",
       "      <th>gap</th>\n",
       "      <th>dmin</th>\n",
       "      <th>rms</th>\n",
       "      <th>...</th>\n",
       "      <th>updated</th>\n",
       "      <th>place</th>\n",
       "      <th>type</th>\n",
       "      <th>horizontalError</th>\n",
       "      <th>depthError</th>\n",
       "      <th>magError</th>\n",
       "      <th>magNst</th>\n",
       "      <th>status</th>\n",
       "      <th>locationSource</th>\n",
       "      <th>magSource</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2024-10-22T02:36:18.744Z</td>\n",
       "      <td>-6.0572</td>\n",
       "      <td>130.3925</td>\n",
       "      <td>10.000</td>\n",
       "      <td>5.1</td>\n",
       "      <td>mb</td>\n",
       "      <td>68</td>\n",
       "      <td>43</td>\n",
       "      <td>1.600</td>\n",
       "      <td>0.59</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-22T03:06:24.040Z</td>\n",
       "      <td>265 km W of Tual, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>8.62</td>\n",
       "      <td>1.757</td>\n",
       "      <td>0.080</td>\n",
       "      <td>51</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2024-10-22T01:12:37.703Z</td>\n",
       "      <td>36.2205</td>\n",
       "      <td>141.2627</td>\n",
       "      <td>35.000</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>40</td>\n",
       "      <td>151</td>\n",
       "      <td>2.485</td>\n",
       "      <td>0.60</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-22T02:44:12.040Z</td>\n",
       "      <td>61 km E of Ōarai, Japan</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.55</td>\n",
       "      <td>1.897</td>\n",
       "      <td>0.107</td>\n",
       "      <td>26</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2024-10-21T20:25:35.394Z</td>\n",
       "      <td>-17.7498</td>\n",
       "      <td>-178.6849</td>\n",
       "      <td>550.271</td>\n",
       "      <td>4.7</td>\n",
       "      <td>mb</td>\n",
       "      <td>85</td>\n",
       "      <td>78</td>\n",
       "      <td>2.249</td>\n",
       "      <td>0.68</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T20:49:50.040Z</td>\n",
       "      <td>214 km E of Levuka, Fiji</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>10.39</td>\n",
       "      <td>8.217</td>\n",
       "      <td>0.029</td>\n",
       "      <td>369</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2024-10-21T19:22:50.440Z</td>\n",
       "      <td>-1.1979</td>\n",
       "      <td>127.4281</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>mb</td>\n",
       "      <td>29</td>\n",
       "      <td>107</td>\n",
       "      <td>1.957</td>\n",
       "      <td>0.85</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T19:49:37.040Z</td>\n",
       "      <td>63 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.91</td>\n",
       "      <td>1.921</td>\n",
       "      <td>0.121</td>\n",
       "      <td>20</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2024-10-21T16:02:44.673Z</td>\n",
       "      <td>-17.4094</td>\n",
       "      <td>-178.4698</td>\n",
       "      <td>530.519</td>\n",
       "      <td>4.5</td>\n",
       "      <td>mb</td>\n",
       "      <td>61</td>\n",
       "      <td>72</td>\n",
       "      <td>2.265</td>\n",
       "      <td>0.72</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:24:53.040Z</td>\n",
       "      <td>245 km ENE of Levuka, Fiji</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>13.55</td>\n",
       "      <td>9.043</td>\n",
       "      <td>0.066</td>\n",
       "      <td>67</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>2024-10-21T15:15:13.441Z</td>\n",
       "      <td>-1.1985</td>\n",
       "      <td>127.4176</td>\n",
       "      <td>10.000</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>42</td>\n",
       "      <td>96</td>\n",
       "      <td>1.958</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T15:40:07.040Z</td>\n",
       "      <td>63 km S of Labuha, Indonesia</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>7.47</td>\n",
       "      <td>1.914</td>\n",
       "      <td>0.100</td>\n",
       "      <td>30</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>2024-10-21T14:11:32.834Z</td>\n",
       "      <td>42.7664</td>\n",
       "      <td>143.6458</td>\n",
       "      <td>117.705</td>\n",
       "      <td>4.6</td>\n",
       "      <td>mb</td>\n",
       "      <td>98</td>\n",
       "      <td>76</td>\n",
       "      <td>0.834</td>\n",
       "      <td>0.52</td>\n",
       "      <td>...</td>\n",
       "      <td>2024-10-21T16:48:45.004Z</td>\n",
       "      <td>4 km S of Honchō, Japan</td>\n",
       "      <td>earthquake</td>\n",
       "      <td>8.04</td>\n",
       "      <td>1.493</td>\n",
       "      <td>0.046</td>\n",
       "      <td>140</td>\n",
       "      <td>reviewed</td>\n",
       "      <td>us</td>\n",
       "      <td>us</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>7 rows × 22 columns</p>\n",
       "</div>\n",
       "    <div class=\"colab-df-buttons\">\n",
       "\n",
       "  <div class=\"colab-df-container\">\n",
       "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-7be44c8b-e8aa-49e4-9c84-dca3e48bec3d')\"\n",
       "            title=\"Convert this dataframe to an interactive table.\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
       "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
       "  </svg>\n",
       "    </button>\n",
       "\n",
       "  <style>\n",
       "    .colab-df-container {\n",
       "      display:flex;\n",
       "      gap: 12px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert {\n",
       "      background-color: #E8F0FE;\n",
       "      border: none;\n",
       "      border-radius: 50%;\n",
       "      cursor: pointer;\n",
       "      display: none;\n",
       "      fill: #1967D2;\n",
       "      height: 32px;\n",
       "      padding: 0 0 0 0;\n",
       "      width: 32px;\n",
       "    }\n",
       "\n",
       "    .colab-df-convert:hover {\n",
       "      background-color: #E2EBFA;\n",
       "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "      fill: #174EA6;\n",
       "    }\n",
       "\n",
       "    .colab-df-buttons div {\n",
       "      margin-bottom: 4px;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert {\n",
       "      background-color: #3B4455;\n",
       "      fill: #D2E3FC;\n",
       "    }\n",
       "\n",
       "    [theme=dark] .colab-df-convert:hover {\n",
       "      background-color: #434B5C;\n",
       "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
       "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
       "      fill: #FFFFFF;\n",
       "    }\n",
       "  </style>\n",
       "\n",
       "    <script>\n",
       "      const buttonEl =\n",
       "        document.querySelector('#df-7be44c8b-e8aa-49e4-9c84-dca3e48bec3d button.colab-df-convert');\n",
       "      buttonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "\n",
       "      async function convertToInteractive(key) {\n",
       "        const element = document.querySelector('#df-7be44c8b-e8aa-49e4-9c84-dca3e48bec3d');\n",
       "        const dataTable =\n",
       "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
       "                                                    [key], {});\n",
       "        if (!dataTable) return;\n",
       "\n",
       "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
       "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
       "          + ' to learn more about interactive tables.';\n",
       "        element.innerHTML = '';\n",
       "        dataTable['output_type'] = 'display_data';\n",
       "        await google.colab.output.renderOutput(dataTable, element);\n",
       "        const docLink = document.createElement('div');\n",
       "        docLink.innerHTML = docLinkHtml;\n",
       "        element.appendChild(docLink);\n",
       "      }\n",
       "    </script>\n",
       "  </div>\n",
       "\n",
       "\n",
       "<div id=\"df-abf44cab-9896-46e3-9a39-f10d689b19b0\">\n",
       "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-abf44cab-9896-46e3-9a39-f10d689b19b0')\"\n",
       "            title=\"Suggest charts\"\n",
       "            style=\"display:none;\">\n",
       "\n",
       "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
       "     width=\"24px\">\n",
       "    <g>\n",
       "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
       "    </g>\n",
       "</svg>\n",
       "  </button>\n",
       "\n",
       "<style>\n",
       "  .colab-df-quickchart {\n",
       "      --bg-color: #E8F0FE;\n",
       "      --fill-color: #1967D2;\n",
       "      --hover-bg-color: #E2EBFA;\n",
       "      --hover-fill-color: #174EA6;\n",
       "      --disabled-fill-color: #AAA;\n",
       "      --disabled-bg-color: #DDD;\n",
       "  }\n",
       "\n",
       "  [theme=dark] .colab-df-quickchart {\n",
       "      --bg-color: #3B4455;\n",
       "      --fill-color: #D2E3FC;\n",
       "      --hover-bg-color: #434B5C;\n",
       "      --hover-fill-color: #FFFFFF;\n",
       "      --disabled-bg-color: #3B4455;\n",
       "      --disabled-fill-color: #666;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart {\n",
       "    background-color: var(--bg-color);\n",
       "    border: none;\n",
       "    border-radius: 50%;\n",
       "    cursor: pointer;\n",
       "    display: none;\n",
       "    fill: var(--fill-color);\n",
       "    height: 32px;\n",
       "    padding: 0;\n",
       "    width: 32px;\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart:hover {\n",
       "    background-color: var(--hover-bg-color);\n",
       "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
       "    fill: var(--button-hover-fill-color);\n",
       "  }\n",
       "\n",
       "  .colab-df-quickchart-complete:disabled,\n",
       "  .colab-df-quickchart-complete:disabled:hover {\n",
       "    background-color: var(--disabled-bg-color);\n",
       "    fill: var(--disabled-fill-color);\n",
       "    box-shadow: none;\n",
       "  }\n",
       "\n",
       "  .colab-df-spinner {\n",
       "    border: 2px solid var(--fill-color);\n",
       "    border-color: transparent;\n",
       "    border-bottom-color: var(--fill-color);\n",
       "    animation:\n",
       "      spin 1s steps(1) infinite;\n",
       "  }\n",
       "\n",
       "  @keyframes spin {\n",
       "    0% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "      border-left-color: var(--fill-color);\n",
       "    }\n",
       "    20% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    30% {\n",
       "      border-color: transparent;\n",
       "      border-left-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    40% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-top-color: var(--fill-color);\n",
       "    }\n",
       "    60% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "    }\n",
       "    80% {\n",
       "      border-color: transparent;\n",
       "      border-right-color: var(--fill-color);\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "    90% {\n",
       "      border-color: transparent;\n",
       "      border-bottom-color: var(--fill-color);\n",
       "    }\n",
       "  }\n",
       "</style>\n",
       "\n",
       "  <script>\n",
       "    async function quickchart(key) {\n",
       "      const quickchartButtonEl =\n",
       "        document.querySelector('#' + key + ' button');\n",
       "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
       "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
       "      try {\n",
       "        const charts = await google.colab.kernel.invokeFunction(\n",
       "            'suggestCharts', [key], {});\n",
       "      } catch (error) {\n",
       "        console.error('Error during call to suggestCharts:', error);\n",
       "      }\n",
       "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
       "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
       "    }\n",
       "    (() => {\n",
       "      let quickchartButtonEl =\n",
       "        document.querySelector('#df-abf44cab-9896-46e3-9a39-f10d689b19b0 button');\n",
       "      quickchartButtonEl.style.display =\n",
       "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
       "    })();\n",
       "  </script>\n",
       "</div>\n",
       "\n",
       "    </div>\n",
       "  </div>\n"
      ],
      "text/plain": [
       "                       time  latitude  longitude    depth  mag magType  nst  \\\n",
       "0  2024-10-22T02:36:18.744Z   -6.0572   130.3925   10.000  5.1      mb   68   \n",
       "1  2024-10-22T01:12:37.703Z   36.2205   141.2627   35.000  4.6      mb   40   \n",
       "2  2024-10-21T20:25:35.394Z  -17.7498  -178.6849  550.271  4.7      mb   85   \n",
       "3  2024-10-21T19:22:50.440Z   -1.1979   127.4281   10.000  4.5      mb   29   \n",
       "4  2024-10-21T16:02:44.673Z  -17.4094  -178.4698  530.519  4.5      mb   61   \n",
       "5  2024-10-21T15:15:13.441Z   -1.1985   127.4176   10.000  4.6      mb   42   \n",
       "6  2024-10-21T14:11:32.834Z   42.7664   143.6458  117.705  4.6      mb   98   \n",
       "\n",
       "   gap   dmin   rms  ...                   updated  \\\n",
       "0   43  1.600  0.59  ...  2024-10-22T03:06:24.040Z   \n",
       "1  151  2.485  0.60  ...  2024-10-22T02:44:12.040Z   \n",
       "2   78  2.249  0.68  ...  2024-10-21T20:49:50.040Z   \n",
       "3  107  1.957  0.85  ...  2024-10-21T19:49:37.040Z   \n",
       "4   72  2.265  0.72  ...  2024-10-21T16:24:53.040Z   \n",
       "5   96  1.958  0.52  ...  2024-10-21T15:40:07.040Z   \n",
       "6   76  0.834  0.52  ...  2024-10-21T16:48:45.004Z   \n",
       "\n",
       "                          place        type horizontalError depthError  \\\n",
       "0   265 km W of Tual, Indonesia  earthquake            8.62      1.757   \n",
       "1       61 km E of Ōarai, Japan  earthquake            7.55      1.897   \n",
       "2      214 km E of Levuka, Fiji  earthquake           10.39      8.217   \n",
       "3  63 km S of Labuha, Indonesia  earthquake            7.91      1.921   \n",
       "4    245 km ENE of Levuka, Fiji  earthquake           13.55      9.043   \n",
       "5  63 km S of Labuha, Indonesia  earthquake            7.47      1.914   \n",
       "6       4 km S of Honchō, Japan  earthquake            8.04      1.493   \n",
       "\n",
       "   magError  magNst    status  locationSource magSource  \n",
       "0     0.080      51  reviewed              us        us  \n",
       "1     0.107      26  reviewed              us        us  \n",
       "2     0.029     369  reviewed              us        us  \n",
       "3     0.121      20  reviewed              us        us  \n",
       "4     0.066      67  reviewed              us        us  \n",
       "5     0.100      30  reviewed              us        us  \n",
       "6     0.046     140  reviewed              us        us  \n",
       "\n",
       "[7 rows x 22 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "O model_response é um DataFrame.\n",
      "Você: Terremotos são frequentes no Brasil?\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'O Brasil não é conhecido por terremotos frequentes ou intensos. Isso ocorre porque o país está localizado em uma região tectonicamente estável, longe das fronteiras entre placas tectônicas, que são as principais zonas de ocorrência de terremotos.'"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Você: Ouro ocorre em sedimentos de origem marinha? geologia\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'Sim, o ouro pode ocorrer em sedimentos de origem marinha, como em conglomerados ricos em minerais pesados. Isso pode formar depósitos do tipo placer ou paleoplacer.'"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Você: Qual a previsão do tempo para São Paulo?\n"
     ]
    },
    {
     "data": {
      "application/vnd.google.colaboratory.intrinsic+json": {
       "type": "string"
      },
      "text/plain": [
       "'{\"coord\": {\"lon\": -46.6361, \"lat\": -23.5475}, \"weather\": [{\"id\": 800, \"main\": \"Clear\", \"description\": \"c\\\\u00e9u limpo\", \"icon\": \"01d\"}], \"base\": \"stations\", \"main\": {\"temp\": 20.4, \"feels_like\": 20.92, \"temp_min\": 19.37, \"temp_max\": 21.16, \"pressure\": 1019, \"humidity\": 93, \"sea_level\": 1019, \"grnd_level\": 929}, \"visibility\": 10000, \"wind\": {\"speed\": 1.54, \"deg\": 20}, \"clouds\": {\"all\": 0}, \"dt\": 1729590251, \"sys\": {\"type\": 1, \"id\": 8394, \"country\": \"BR\", \"sunrise\": 1729585632, \"sunset\": 1729631663}, \"timezone\": -10800, \"id\": 3448439, \"name\": \"S\\\\u00e3o Paulo\", \"cod\": 200}'"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Você: Sair\n",
      "Encerrando chat. Até a próxima!\n"
     ]
    }
   ],
   "source": [
    "chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "756f67a4",
   "metadata": {
    "id": "1mgLM8xT0M5b"
   },
   "source": [
    "## Vídeo 4.2 - Transformando em um app Gradio"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "62dfd324",
   "metadata": {
    "id": "TSwM0Zrv2fFR"
   },
   "outputs": [],
   "source": [
    "!pip install litellm gradio pinecone"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6ef72980",
   "metadata": {
    "id": "HW8VlF2ceeXB"
   },
   "outputs": [],
   "source": [
    "from litellm import completion\n",
    "import requests\n",
    "import json\n",
    "import gradio as gr\n",
    "import pandas as pd\n",
    "from pinecone import Pinecone\n",
    "import os\n",
    "from google.colab import userdata\n",
    "\n",
    "def info_geologia(query):\n",
    "    PINECONE_API = userdata.get('PINECONE_API')\n",
    "    pc = Pinecone(api_key=PINECONE_API)\n",
    "    index = pc.Index(\"geologia\")\n",
    "    x = pc.inference.embed(\n",
    "        model=\"multilingual-e5-large\",\n",
    "        inputs=[query],\n",
    "        parameters={\n",
    "            \"input_type\": \"query\"\n",
    "        }\n",
    "    )\n",
    "    results = index.query(\n",
    "        namespace=\"ns1\",\n",
    "        vector=x[0].values,\n",
    "        top_k=3,\n",
    "        include_values=False,\n",
    "        include_metadata=True\n",
    "    )\n",
    "    return results\n",
    "\n",
    "def previsao_do_tempo(city, country):\n",
    "    WEATHER_API = userdata.get('WEATHER_API')\n",
    "    url = f\"http://api.openweathermap.org/data/2.5/weather?q={city},{country}&APPID={WEATHER_API}&lang=pt_br&units=metric\"\n",
    "    response = requests.get(url)\n",
    "    data = response.json()\n",
    "\n",
    "    return json.dumps(data)\n",
    "\n",
    "\n",
    "def verificar_tempestade_solar():\n",
    "    url = \"https://services.swpc.noaa.gov/products/noaa-planetary-k-index.json\"\n",
    "    response = requests.get(url)\n",
    "    if response.status_code == 200:\n",
    "        data = response.json()\n",
    "        latest_kp = float(data[-1][1])  # O último valor Kp\n",
    "        if latest_kp >= 5:\n",
    "            return f\"Alerta de tempestade solar! Índice Kp atual: {latest_kp}\"\n",
    "        else:\n",
    "            return f\"Sem tempestade solar no momento. Índice Kp atual: {latest_kp}\"\n",
    "    else:\n",
    "        return \"Não foi possível obter informações sobre tempestades solares no momento.\"\n",
    "\n",
    "def extrair_sismos():\n",
    "    # Fazer a requisição para obter o conteúdo da página\n",
    "    url = 'https://earthquake.usgs.gov/earthquakes/feed/v1.0/summary/4.5_day.csv'\n",
    "\n",
    "    df = pd.read_csv(url)\n",
    "\n",
    "    # Retornar o DataFrame com os dados\n",
    "    return df\n",
    "\n",
    "\n",
    "tools = [\n",
    "        {\n",
    "            \"type\": \"function\",\n",
    "            \"function\": {\n",
    "                \"name\": \"previsao_do_tempo\",\n",
    "                \"description\": \"Retorna a previsão do tempo em uma cidade específica\",\n",
    "                \"parameters\": {\n",
    "                    \"type\": \"object\",\n",
    "                    \"properties\": {\n",
    "                        \"city\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Nome da cidade\",\n",
    "                        },\n",
    "                        \"country\": {\n",
    "                            \"type\": \"string\",\n",
    "                            \"description\": \"Sigla do país\",\n",
    "                        },\n",
    "                    },\n",
    "                    \"required\": [\"city\", \"country\"],\n",
    "                },\n",
    "            }\n",
    "\n",
    "        },\n",
    "\n",
    "      {\n",
    "        \"type\": \"function\",\n",
    "        \"function\": {\n",
    "            \"name\": \"verificar_tempestade_solar\",\n",
    "            \"description\": \"Verifica se há uma tempestade solar em andamento\",\n",
    "            \"parameters\": {\n",
    "                \"type\": \"object\",\n",
    "                \"properties\": {},\n",
    "                \"required\": [],\n",
    "            },\n",
    "        }\n",
    "      },\n",
    "      {\n",
    "        \"type\": \"function\",\n",
    "        \"function\": {\n",
    "            \"name\": \"extrair_sismos\",\n",
    "            \"description\": \"Extrai dados de sismos da USGS\",\n",
    "            \"parameters\": {\n",
    "                \"type\": \"object\",\n",
    "                \"properties\": {},\n",
    "                \"required\": [],\n",
    "            },\n",
    "        }\n",
    "      }\n",
    "]\n",
    "\n",
    "\n",
    "# Função para chamar a API com o histórico de mensagens\n",
    "def call_groq_api(messages, model=\"groq/llama3-groq-70b-8192-tool-use-preview\"):\n",
    "    global tools\n",
    "    GROQ_API_KEY = userdata.get('GROQ_API_KEY')\n",
    "    response = completion(\n",
    "        model=model,\n",
    "        messages=messages,\n",
    "        tools=tools,\n",
    "        tool_choice=\"auto\",\n",
    "        api_key=GROQ_API_KEY,\n",
    "    )\n",
    "    resposta_texto = response.choices[0].message\n",
    "    chamada_ferramentas = resposta_texto.tool_calls\n",
    "    if chamada_ferramentas:\n",
    "      available_functions = {\n",
    "        \"previsao_do_tempo\": previsao_do_tempo,\n",
    "        \"verificar_tempestade_solar\": verificar_tempestade_solar,\n",
    "        \"extrair_sismos\": extrair_sismos\n",
    "      }\n",
    "      for tool_call in chamada_ferramentas:\n",
    "        function_name = tool_call.function.name\n",
    "        function_to_call = available_functions[function_name]\n",
    "        function_args = json.loads(tool_call.function.arguments)\n",
    "\n",
    "        match function_name:\n",
    "          case \"previsao_do_tempo\":\n",
    "\n",
    "            function_response = function_to_call(\n",
    "                city=function_args.get(\"city\"),\n",
    "                country=function_args.get(\"country\"),\n",
    "            )\n",
    "          case \"verificar_tempestade_solar\":\n",
    "            function_response = function_to_call()\n",
    "          case \"extrair_sismos\":\n",
    "            function_response = function_to_call()\n",
    "        return function_response\n",
    "\n",
    "    else:\n",
    "      return resposta_texto.content\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "216aa476",
   "metadata": {
    "id": "3SRMTQjIefs7"
   },
   "outputs": [],
   "source": [
    "def response(message, history):\n",
    "    messages = [{\"role\": \"system\", \"content\": \"\"\"\n",
    "    Você é o Chat da Terra e do Universo e responde em português brasileiro\n",
    "    perguntas sobre a previsão do tempo na Terra e do espaço próximo à Terra, além de informações sobre terremotos.\n",
    "    \"\"\"}]\n",
    "\n",
    "    # Adicionar o histórico anterior ao histórico de mensagens\n",
    "    for user_msg, bot_msg in history:\n",
    "        messages.append({\"role\": \"user\", \"content\": user_msg})\n",
    "        messages.append({\"role\": \"assistant\", \"content\": bot_msg})\n",
    "\n",
    "    # Adicionar a nova mensagem do usuário\n",
    "    messages.append({\"role\": \"user\", \"content\": message})\n",
    "\n",
    "    # Verificar se o tema é geologia\n",
    "    if \"geologia\" in message.lower():\n",
    "        resposta = info_geologia(message)\n",
    "        geologia_info = f\"Informações adicionais sobre geologia: {resposta.matches[0].metadata['text']}\"\n",
    "        messages.append({\"role\": \"system\", \"content\": geologia_info})\n",
    "\n",
    "    # Obter a resposta do modelo\n",
    "    model_response = call_groq_api(messages)\n",
    "\n",
    "    # Se a resposta for um DataFrame, convertê-la para texto\n",
    "    if isinstance(model_response, pd.DataFrame):\n",
    "        texto_corrido = \"\"\n",
    "        for index, row in model_response.iterrows():\n",
    "            texto_corrido += f\"Evento {index + 1}: Magnitude {row['mag']}, Local {row['place']}, Tempo {row['time']}\\n\"\n",
    "        model_response = texto_corrido\n",
    "\n",
    "    # Retornar a resposta como string para Gradio\n",
    "    return model_response"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3e863a62",
   "metadata": {
    "id": "E3-ZXcM7elFU"
   },
   "outputs": [],
   "source": [
    "# Interface Gradio com ChatInterface\n",
    "gr.ChatInterface(\n",
    "    response,  # Função que gera a resposta\n",
    "    title='🌍☀️🌧️ Chat da Terra e do Universo',\n",
    "    textbox=gr.Textbox(placeholder=\"Digite sua mensagem aqui...\"),\n",
    "    type='messages'\n",
    ").launch(debug=True)"
   ]
  }
 ],
 "metadata": {},
 "nbformat": 4,
 "nbformat_minor": 5
}