Update src/index.js

src/index.js

@@ -3,6 +3,7 @@ for (let sign of ['unhandledRejection', 'uncaughtException']) process.on(sign, c
 const os = require('os')
 const util = require('util')
 const uuid = require('uuid')
+const logger = require('morgan')
 const express = require('express')
 const puppeteer = require('puppeteer')
 const rateLimit = require('express-rate-limit')
@@ -14,20 +15,15 @@ const sessions = new Map()
 const app = express()
 app.set('trust proxy', 1)
 app.set('json spaces', 4)
+app.use(logger('dev'))
 app.use(express.urlencoded({ extended: true }))
 app.use(async (req, res, next) => {
-	console.log(req.method, req.path, res.statusCode)
-	
-	const sessionId = req.query.sessionId || req.body.sessionId
-	if (sessions.has(sessionId)) {
-		const characterAI = sessions.get(sessionId)
-		
-		if (Date.now() - characterAI.lastfetch > 10 * 60 * 1000) {
+	for (let [sessionId, characterAI] of sessions) {
+		if (Date.now() - characterAI.lastFetch > 10 * 60 * 1000) {
 			await characterAI.unauthenticate()
 			sessions.delete(sessionId)
 		}
 	}
-	
 	next()
 })
 
@@ -64,7 +60,7 @@ app.get('/ev', async (req, res) => {
 	}
 })
 
-app.get('/sessions', (_, res) => res.json({ sessions: eval(sessions.values()) }))
+app.get('/sessions', (_, res) => res.json({ sessions: eval(Array.from(sessions)) }))
 
 app.all('/api', rateLimit(), async (req, res) => {
 	if (!['GET', 'POST'].includes(req.method)) return res.json({ message: 'Method not allowed' })
@@ -80,9 +76,10 @@ app.all('/api', rateLimit(), async (req, res) => {
 			
 			const chat = await characterAI.createOrContinueChat(characterId)
 			const response = await chat.sendAndAwaitResponse(text, true)
+			const urlAvatar = `https://characterai.io/i/80/static/avatars/${response.srcAvatarFileName}`
 			
 			delete response.chat
-			res.json({ ...response, sessionId })
+			res.json({ ...response, urlAvatar, sessionId })
 			return
 		}
 		
@@ -94,12 +91,13 @@ app.all('/api', rateLimit(), async (req, res) => {
 		
 		const chat = await characterAI.createOrContinueChat(characterId)
 		const response = await chat.sendAndAwaitResponse(text, true)
+		const urlAvatar = `https://characterai.io/i/80/static/avatars/${response.srcAvatarFileName}`
 		
 		delete response.chat
-		res.json({ ...response, sessionId: id })
+		res.json({ ...response, urlAvatar, sessionId: id })
 	} catch (e) {
 		console.log(e)
-		e = e.toString()
+		e = String(e)
 		res.json({ message: e === '[object Object]' ? 'Internal Server Error' : e })
 	}
 })
@@ -107,7 +105,6 @@ app.all('/api', rateLimit(), async (req, res) => {
 const PORT = process.env.PORT || 7860
 app.listen(PORT, () => console.log('App running on port', PORT))
 
-function findChromium() {
-	let cmd = execSync('nix eval nixpkgs.chromium.outPath --raw').toString()
-	return `${cmd}/bin/chromium`
-}
+// const { REPL_SLUG, REPL_OWNER } = process.env
+// if (REPL_SLUG && REPL_OWNER)
+	setInterval(() => fetch(`https://c-ai.team-skizo.repl.co`, { method: 'head' }).catch(console.log), 20 * 1000)