nltk_chat.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "%run google_stt.ipynb\n",
    "%run google_tts.ipynb\n",
    "%run google_nlp.ipynb\n",
    "\n",
    "# Natural Language Toolkit: Chatbot Utilities\n",
    "#\n",
    "# Copyright (C) 2001-2020 NLTK Project\n",
    "# Authors: Steven Bird <stevenbird1@gmail.com>\n",
    "# URL: <http://nltk.org/>\n",
    "# For license information, see LICENSE.TXT\n",
    "\n",
    "# Based on an Eliza implementation by Joe Strout <joe@strout.net>,\n",
    "# Jeff Epler <jepler@inetnebr.com> and Jez Higgins <jez@jezuk.co.uk>.\n",
    "\n",
    "import re\n",
    "import random\n",
    "import wikipedia\n",
    "\n",
    "reflections = {\n",
    "    \"i am\": \"you are\",\n",
    "    \"i was\": \"you were\",\n",
    "    \"i\": \"you\",\n",
    "    \"i'm\": \"you are\",\n",
    "    \"i'd\": \"you would\",\n",
    "    \"i've\": \"you have\",\n",
    "    \"i'll\": \"you will\",\n",
    "    \"my\": \"your\",\n",
    "    \"you are\": \"I am\",\n",
    "    \"you were\": \"I was\",\n",
    "    \"you've\": \"I have\",\n",
    "    \"you'll\": \"I will\",\n",
    "    \"your\": \"my\",\n",
    "    \"yours\": \"mine\",\n",
    "    \"you\": \"me\",\n",
    "    \"me\": \"you\",\n",
    "}\n",
    "\n",
    "\n",
    "class Chat(object):\n",
    "    def __init__(self, pairs, reflections={}):\n",
    "        \"\"\"\n",
    "        Initialize the chatbot.  Pairs is a list of patterns and responses.  Each\n",
    "        pattern is a regular expression matching the user's statement or question,\n",
    "        e.g. r'I like (.*)'.  For each such pattern a list of possible responses\n",
    "        is given, e.g. ['Why do you like %1', 'Did you ever dislike %1'].  Material\n",
    "        which is matched by parenthesized sections of the patterns (e.g. .*) is mapped to\n",
    "        the numbered positions in the responses, e.g. %1.\n",
    "\n",
    "        :type pairs: list of tuple\n",
    "        :param pairs: The patterns and responses\n",
    "        :type reflections: dict\n",
    "        :param reflections: A mapping between first and second person expressions\n",
    "        :rtype: None\n",
    "        \"\"\"\n",
    "\n",
    "        self._pairs = [(re.compile(x, re.IGNORECASE), y) for (x, y) in pairs]\n",
    "        self._reflections = reflections\n",
    "        self._regex = self._compile_reflections()\n",
    "\n",
    "    def _compile_reflections(self):\n",
    "        sorted_refl = sorted(self._reflections, key=len, reverse=True)\n",
    "        return re.compile(\n",
    "            r\"\\b({0})\\b\".format(\"|\".join(map(re.escape, sorted_refl))), re.IGNORECASE\n",
    "        )\n",
    "\n",
    "    def _substitute(self, str):\n",
    "        \"\"\"\n",
    "        Substitute words in the string, according to the specified reflections,\n",
    "        e.g. \"I'm\" -> \"you are\"\n",
    "\n",
    "        :type str: str\n",
    "        :param str: The string to be mapped\n",
    "        :rtype: str\n",
    "        \"\"\"\n",
    "\n",
    "        return self._regex.sub(\n",
    "            lambda mo: self._reflections[mo.string[mo.start() : mo.end()]], str.lower()\n",
    "        )\n",
    "\n",
    "    def _wildcards(self, response, match):\n",
    "        pos = response.find(\"%\")\n",
    "        while pos >= 0:\n",
    "            num = int(response[pos + 1 : pos + 2])\n",
    "            response = (\n",
    "                response[:pos]\n",
    "                + self._substitute(match.group(num))\n",
    "                + response[pos + 2 :]\n",
    "            )\n",
    "            pos = response.find(\"%\")\n",
    "        return response\n",
    "\n",
    "    def respond(self, str):\n",
    "            \"\"\"\n",
    "            Generate a response to the user input.\n",
    "\n",
    "            :type str: str\n",
    "            :param str: The string to be mapped\n",
    "            :rtype: str\n",
    "            \"\"\"\n",
    "\n",
    "            # check each pattern\n",
    "            for (pattern, response) in self._pairs:\n",
    "                match = pattern.match(str)\n",
    "\n",
    "                # did the pattern match?\n",
    "                if match:\n",
    "                    resp = random.choice(response)  # pick a random response\n",
    "                    resp = self._wildcards(resp, match)  # process wildcards\n",
    "\n",
    "                    # fix munged punctuation at the end\n",
    "                    if resp[-2:] == \"?.\":\n",
    "                        resp = resp[:-2] + \".\"\n",
    "                    if resp[-2:] == \"??\":\n",
    "                        resp = resp[:-2] + \"?\"\n",
    "                    return resp\n",
    "\n",
    "\n",
    "        # Hold a conversation with a chatbot\n",
    "    def converse(self, quit=\"quit\"):\n",
    "        user_input = \"\"\n",
    "        while user_input != quit:\n",
    "            user_input = speech_to_text()\n",
    "\n",
    "            if user_input:\n",
    "                while user_input[-1] in \"!.\":\n",
    "                    user_input = user_input[:-1]\n",
    "\n",
    "                print(\"Chatty: \", end='')\n",
    "                response = self.respond(user_input)\n",
    "\n",
    "                if (response):\n",
    "                    print(response)\n",
    "                    text_to_speech(response)\n",
    "                    if(response == \"I'm not sure. Let me look that up for you.\"):\n",
    "                        time.sleep(2)\n",
    "                        entity = sample_analyze_entities(user_input)\n",
    "                        wiki = re.sub('listen', '', wikipedia.summary(entity, sentences=1))\n",
    "                        print(wiki)\n",
    "                        text_to_speech(wiki)\n",
    "                else:\n",
    "                    print(\"I don't understand that :(\")\n",
    "                    text_to_speech(\"I don't understand that :(\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}