From 15fc56f4cbf3f9fcc48ebec9bbf1d1c474a12d21 Mon Sep 17 00:00:00 2001
From: Liza Shulyayeva <65890040+lazeratops@users.noreply.github.com>
Date: Wed, 3 May 2023 09:50:55 +0200
Subject: [PATCH] Account for possible timing issues with model load

---
 src/components/ChatTile.vue | 30 +++++++++++++++++++++---------
 1 file changed, 21 insertions(+), 9 deletions(-)

diff --git a/src/components/ChatTile.vue b/src/components/ChatTile.vue
index e0e949c..dce788d 100644
--- a/src/components/ChatTile.vue
+++ b/src/components/ChatTile.vue
@@ -1,8 +1,11 @@
 <template>
   <div :class="chatIsOpen ? 'chat-wrapper open' : 'chat-wrapper'">
     <div>
-      <button class="chat" @click="handleChatClick">
-        <template v-if="chatIsOpen">
+      <button class="chat" @click="handleChatClick" :disabled="!model">
+        <template v-if="!model">
+          <img class="icon" :src="close" alt="" />
+        </template>
+        <template v-else-if="chatIsOpen">
           <img class="icon" :src="close" alt="" />
         </template>
         <template v-else>
@@ -23,7 +26,8 @@
       </div>
 
       <p v-if="displayWarning" class="warning-message">
-        Oops! The message you're trying to send was flagged as potentially offensive. If you think this was flagged in error, please contact us!
+        Oops! The message you're trying to send was flagged as potentially
+        offensive. If you think this was flagged in error, please contact us!
       </p>
 
       <form @submit="submitForm">
@@ -47,7 +51,7 @@
 </template>
 
 <script>
-import { load } from '@tensorflow-models/toxicity';
+import { load } from "@tensorflow-models/toxicity";
 
 import close from "../assets/x.svg";
 import chat from "../assets/chat.svg";
@@ -69,9 +73,13 @@ export default {
   },
   beforeCreate() {
     const threshold = 0.9;
-    load(threshold).then(model => {
-      this.model = Object.freeze(model);
-    });
+    load(threshold)
+      .then((model) => {
+        this.model = Object.freeze(model);
+      })
+      .catch((e) => {
+        console.error("failed to load toxicity model", e);
+      });
   },
   methods: {
     // Toggle chat's view (open/closed)
@@ -84,12 +92,16 @@ export default {
       this.displayWarning = false;
 
       const hasToxicContent = await this.analyzeInputText(this.text);
-      hasToxicContent ? this.displayWarning = true : this.sendMessage(this.text);
+      hasToxicContent
+        ? (this.displayWarning = true)
+        : this.sendMessage(this.text);
       this.text = "";
     },
     async analyzeInputText(text) {
       const predictions = await this.model.classify(text);
-      const containsToxicContent = predictions.some(prediction => prediction.results[0].match);
+      const containsToxicContent = predictions.some(
+        (prediction) => prediction.results[0].match
+      );
       return containsToxicContent;
     },
   },