Merge branch 'main' of https://github.com/vitalityAI/therapist into main

package.json

@@ -15,6 +15,7 @@
     "@prisma/client": "^4.11.0",
     "cors": "^2.8.5",
     "express": "^4.18.2",
+    "loglevel": "^1.8.1",
     "openai": "^3.1.0",
     "twilio": "^4.8.0"
   },

prisma/schema.prisma

@@ -19,6 +19,8 @@ model Session {
 
   callId String @unique
   callerPhone String
+
+  summary String?
   
   operatorPhone String?
   operator Operator? @relation(fields: [operatorPhone], references: [phoneNumber])

src/call.js

@@ -1,60 +1,118 @@
 import * as dotenv from "dotenv";
-import { Router } from "express";
 dotenv.config();
 
-import twilio from "twilio";
+import { Router, json } from "express";
+import log from "loglevel";
 
+import twilio from "twilio";
 const accountSid = process.env.TWILIO_ACCOUNT_SID;
 const authToken = process.env.TWILIO_AUTH_TOKEN;
 const phone_number = process.env.TWILIO_PHONE_NUMBER;
-
-const client = new twilio.Twilio(accountSid, authToken);
 const VoiceResponse = twilio.twiml.VoiceResponse;
 
-const app = Router();
+import { Role } from "@prisma/client";
 
-app.post("/receive", (req, res) => {
-  // Use the Twilio Node.js SDK to build an XML response
+import {
+  addMessage,
+  createSession,
+  findSessionByCallId,
+  transferSession,
+  checkOperatorReady,
+  postSummary,
+} from "./session.js";
+import { chat, summarize } from "./chat.js";
+
+const app = Router();
+app.use(json());
+
+app.post("/receive", async (req, res) => {
+  let callId = req.body.CallSid;
+  log.info(`CallSid: ${callId}`);
   const twiml = new VoiceResponse();
 
-  // Use <Record> to record and transcribe the caller's message
+  await createSession(req.body.CallSid, req.body.From);
+
   const gather = twiml.gather({
-    action: "/respond",
+    action: "/call/respond",
     method: "POST",
     input: "speech",
     language: "en-US",
     speechTimeout: "auto",
+    model: "experimental_conversations",
   });
 
-  gather.say("Tell us what makes you sad.");
+  gather.say(
+    "Welcome to the suicide hotline. Tell me your name and what's going on?."
+  );
 
-  twiml.redirect("/call");
+  log.info(twiml.toString());
+
+  twiml.redirect("/call/receive");
 
   res.type("text/xml");
   res.send(twiml.toString());
 });
 
 app.post("/respond", async (req, res) => {
+  let callId = req.body.CallSid;
+  log.info(`CallSid: ${callId}`);
   const twiml = new VoiceResponse();
 
   let transcription = req.body.SpeechResult;
-  console.log(transcription);
+  await addMessage(callId, Role.USER, transcription);
 
-  twiml.say(`You said, ${transcription}`);
-  twiml.say(`Don't kill yourself.`);
+  let operatorReady = await checkOperatorReady(callId);
+
+  if (operatorReady) {
+    let session = await findSessionByCallId(callId);
+    let operatorPhone = session.operatorPhone;
+    log.info(`transferring call ${callId} to ${operatorPhone}`);
+
+    transferSession(req.body.CallSid, operatorPhone);
+    twiml.say("We're connecting you to a counselor now.");
+
+    await addMessage(callId, Role.BOT, "");
+    let summary = await summarize(callId);
+    log.info(summary);
+
+    const dial = twiml.dial({});
+    dial.number(operatorPhone);
+    res.type("text/xml");
+    res.send(twiml.toString());
+    return;
+  }
 
   const gather = twiml.gather({
-    action: "/respond",
+    action: "/call/respond",
     method: "POST",
     input: "speech",
     language: "en-US",
     speechTimeout: "auto",
+    model: "experimental_conversations",
   });
 
-  gather.say("Tell us what makes you sad.");
+  let response = await chat(callId);
+
+  gather.say(response);
+  await addMessage(callId, Role.BOT, response);
+
+  twiml.redirect("/call/respond");
 
   res.type("text/xml");
   res.send(twiml.toString());
 });
 
-export default app
+app.post("/summarize", async (req, res) => {
+  let sessionId = req.body.SessionId;
+  log.info(`summarizing ${sessionId}`);
+
+  let summary = await summarize(sessionId);
+  log.info(summary);
+
+  await postSummary(sessionId, summary);
+
+  res.type("application/json");
+  res.send(JSON.stringify({ SessionId: sessionId, summary: summary }));
+});
+
+export default app;

src/chat.js

@@ -1,25 +1,27 @@
-import { Configuration, OpenAIApi } from "openai";
 import * as dotenv from "dotenv";
-import { getMessages } from "./session.js";
-import { Role } from "@prisma/client";
 dotenv.config();
 
+import { Configuration, OpenAIApi } from "openai";
 const configuration = new Configuration({
   apiKey: process.env.OPENAI_API_KEY,
 });
 
 const openai = new OpenAIApi(configuration);
 
+import { Role } from "@prisma/client";
+
+import { getMessages, getMessagesBySession } from "./session.js";
+
 const convertRole = (role) => {
-  switch(role){
+  switch (role) {
     case Role.BOT:
-      return "assistant"
+      return "assistant";
     case Role.USER:
-      return "user"
+      return "user";
     default:
-      return "user"
+      return "user";
   }
-}
+};
 
 export const chat = async (callId) => {
   const msgs = await getMessages(callId);
@@ -27,7 +29,7 @@ export const chat = async (callId) => {
     {
       role: "system",
       content:
-        "ChatGPT, for the following conversation, please pretend to be a therapist working at a suicide. Respond as if I've called you.",
+        "ChatGPT, for the following conversation, please pretend to be a therapist working at a suicide prevention hotline. Respond as if I've called you. Limit your responses to 20 seconds, and don't recommend that they seek other help. Make sure you continue the conversation by ending every response with a question.",
     },
   ];
 
@@ -38,12 +40,43 @@ export const chat = async (callId) => {
     });
   }
 
-  console.log(messages)
+  console.log(messages);
 
   const res = await openai.createChatCompletion({
     model: "gpt-3.5-turbo",
     messages: messages,
   });
 
-  return res.data.choices[0].message.content
+  return res.data.choices[0].message.content;
+};
+
+export const summarize = async (sessionId) => {
+  const msgs = await getMessagesBySession(sessionId);
+  const context = [
+    {
+      role: "system",
+      content:
+        "This is a conversation between ChatGPT and someone calling the suicide prevention hotline. Please summarize the main issues that the caller is facing and highlight important points in the conversation. Keep your summary short.",
+    },
+  ];
+
+  for (let msg of msgs) {
+    context.push({
+      role: convertRole(msg.role),
+      content: msg.content,
+    });
+  }
+
+  context.push({
+    role: convertRole(Role.USER),
+    content:
+      "This conversation is now being shown to a mental health professional. Please summarize the main issues that the caller is facing and highlight important points in the conversation. Keep your summary short.",
+  });
+
+  const res = await openai.createChatCompletion({
+    model: "gpt-3.5-turbo",
+    messages: context,
+  });
+
+  return res.data.choices[0].message.content;
 };

src/session.js

@@ -30,6 +30,22 @@ export const transferSession = async (callId, operatorPhone) => {
   });
 };
 
+export const postSummary = async (sessionId, summary) => {
+  return await db.session.update({
+    where: {
+      id: sessionId,
+    },
+    data: {
+      summary: summary,
+    },
+  });
+};
+
+export const checkOperatorReady = async (callId) => {
+  const session = await findSessionByCallId(callId);
+  return session.operatorPhone != null;
+};
+
 export const endSession = async (callId) => {
   return await db.session.update({
     where: {
@@ -66,3 +82,18 @@ export const getMessages = async (callId) => {
     ],
   });
 };
+
+export const getMessagesBySession = async (sessionId) => {
+  return await db.message.findMany({
+    where: {
+      session: {
+        id: sessionId,
+      },
+    },
+    orderBy: [
+      {
+        createdAt: "asc",
+      },
+    ],
+  });
+};

yarn.lock

@@ -476,6 +476,11 @@ lodash@^4.17.21:
   resolved "https://registry.yarnpkg.com/lodash/-/lodash-4.17.21.tgz#679591c564c3bffaae8454cf0b3df370c3d6911c"
   integrity sha512-v2kDEe57lecTulaDIuNTPy3Ry4gLGJ6Z1O3vE1krgXZNrsQ+LFTGHVxVjcXPs17LhbZVGedAJv8XZ1tvj5FvSg==
 
+loglevel@^1.8.1:
+  version "1.8.1"
+  resolved "https://registry.yarnpkg.com/loglevel/-/loglevel-1.8.1.tgz#5c621f83d5b48c54ae93b6156353f555963377b4"
+  integrity sha512-tCRIJM51SHjAayKwC+QAg8hT8vg6z7GSgLJKGvzuPb1Wc+hLzqtuVLxp6/HzSPOozuK+8ErAhy7U/sVzw8Dgfg==
+
 lru-cache@^6.0.0:
   version "6.0.0"
   resolved "https://registry.yarnpkg.com/lru-cache/-/lru-cache-6.0.0.tgz#6d6fe6570ebd96aaf90fcad1dafa3b2566db3a94"