juji_chat.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "757667b9",
   "metadata": {},
   "outputs": [],
   "source": [
    "import requests\n",
    "import websocket\n",
    "import json\n",
    "import logging\n",
    "from time import sleep"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "4c279a7e",
   "metadata": {},
   "source": [
    "## Request to create participation and get websocket url"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "10756005",
   "metadata": {},
   "outputs": [],
   "source": [
    "r = requests.post('https://juji.ai/pre-chat/62886f95-94d1-408f-8c19-70706611271c', \n",
    "                  data={'firstName': 'John'})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c258dd29",
   "metadata": {},
   "outputs": [],
   "source": [
    "r.raise_for_status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "11c3096c",
   "metadata": {},
   "outputs": [],
   "source": [
    "chat_info = r.json()\n",
    "chat_info"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "04273300",
   "metadata": {},
   "outputs": [],
   "source": [
    "chat_info['websocketUrl']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "002ad7fb",
   "metadata": {},
   "outputs": [],
   "source": [
    "def on_message(ws, message):\n",
    "    # print(message)\n",
    "    parsed = json.loads(message)\n",
    "#     print(message)\n",
    "    if \"data\" in parsed:\n",
    "        if \"chat\" in parsed[\"data\"]:\n",
    "            chat_data = parsed[\"data\"][\"chat\"]\n",
    "            if \"role\" in chat_data and chat_data[\"role\"] == \"rep\":\n",
    "                if chat_data[\"type\"] == \"normal\":\n",
    "                    print(chat_data[\"text\"])\n",
    "                elif chat_data[\"type\"] == \"user-joined\":\n",
    "                    print(\"=== Welcome to Juji Bot ===\")\n",
    "\n",
    "def on_error(ws, error):\n",
    "    print(error)\n",
    "\n",
    "def on_close(ws, close_status_code, close_msg):\n",
    "    print(\"### closed ###\")\n",
    "\n",
    "def on_open(ws):\n",
    "    print(\"Opened connection\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "15300d09",
   "metadata": {},
   "source": [
    "## Method1: Initiate Websocket App"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3316efdc",
   "metadata": {},
   "outputs": [],
   "source": [
    "websocket.enableTrace(True)\n",
    "ws = websocket.WebSocketApp(chat_info['websocketUrl'],\n",
    "                            on_open=on_open,\n",
    "                            on_message=on_message,\n",
    "                            on_error=on_error,\n",
    "                            on_close=on_close)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9290d194",
   "metadata": {},
   "outputs": [],
   "source": [
    "ws.run_forever()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d0eabb24",
   "metadata": {},
   "source": [
    "## Mehtod2: Initiate Websocket"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "80b0222f",
   "metadata": {},
   "outputs": [],
   "source": [
    "ws = websocket.create_connection(chat_info['websocketUrl'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6b45a4cd",
   "metadata": {},
   "outputs": [],
   "source": [
    "ws.send(\"\"\"\n",
    "        subscription {{\n",
    "                chat(input: {{\n",
    "                    participationId: \"{0}\"\n",
    "                }}) {{\n",
    "                    role\n",
    "                    text\n",
    "                    type\n",
    "                    display{{\n",
    "                        data{{\n",
    "                            questions{{\n",
    "                                heading\n",
    "                                kind\n",
    "                            }}\n",
    "                        }}\n",
    "                    }}\n",
    "                }}\n",
    "            }}\"\"\".format(chat_info[\"participationId\"]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "940bc8c8",
   "metadata": {},
   "outputs": [],
   "source": [
    "while True:\n",
    "    sleep(1)\n",
    "    result = ws.recv()\n",
    "    on_message(ws, result)\n",
    "    # use keep-alive message to estimate end of message from Chatbot\n",
    "    # this is inefficient because keep-alive message takes a long time to arrive\n",
    "    if \"type\" in result and result[\"type\"] == \"keep-alive\":\n",
    "        break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "69eb3623",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python [conda env:juji-client] *",
   "language": "python",
   "name": "conda-env-juji-client-py"
  },
  "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.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}