added check to onCloseWebsocket event

src/frontend/src/controllers/API/index.ts

@@ -312,7 +312,7 @@ export async function getHealth() {
  */
 export async function getBuildStatus(
   flowId: string
-): Promise<BuildStatusTypeAPI> {
+): Promise<AxiosResponse<BuildStatusTypeAPI>> {
   return await api.get(`${BASE_URL_API}build/${flowId}/status`);
 }
 

src/frontend/src/modals/formModal/index.tsx

@@ -8,7 +8,7 @@ import { classNames } from "../../utils/utils";
 import ChatInput from "./chatInput";
 import ChatMessage from "./chatMessage";
 
-import _ from "lodash";
+import _, { set } from "lodash";
 import AccordionComponent from "../../components/AccordionComponent";
 import IconComponent from "../../components/genericIconComponent";
 import ToggleShadComponent from "../../components/toggleShadComponent";
@@ -27,6 +27,7 @@ import { AuthContext } from "../../contexts/authContext";
 import { TabsContext } from "../../contexts/tabsContext";
 import { TabsState } from "../../types/tabs";
 import { validateNodes } from "../../utils/reactflowUtils";
+import { getBuildStatus } from "../../controllers/API";
 
 export default function FormModal({
   flow,
@@ -155,9 +156,21 @@ export default function FormModal({
 
   function handleOnClose(event: CloseEvent): void {
     if (isOpen.current) {
+      getBuildStatus(flow.id).then((response) => {
+        if (response.data.built) {
+          connectWS();
+        }
+        else {
+          setErrorData({
+            title: "Please build the flow again before using the chat."
+          })
+        }
+      }).catch((error) => {
+        setErrorData({title:error.data?.detail?error.data.detail:error.message})
+
+      });
       setErrorData({ title: event.reason });
       setTimeout(() => {
-        connectWS();
         setLockChat(false);
       }, 1000);
     }
@@ -173,8 +186,7 @@ export default function FormModal({
     const host = isDevelopment ? "localhost:7860" : window.location.host;
     const chatEndpoint = `/api/v1/chat/${chatId}`;
 
-    return `${
-      isDevelopment ? "ws" : webSocketProtocol
+    return `${isDevelopment ? "ws" : webSocketProtocol
       }://${host}${chatEndpoint}?token=${encodeURIComponent(accessToken!)}`;
   }
 
@@ -260,7 +272,6 @@ export default function FormModal({
       };
       newWs.onmessage = (event) => {
         const data = JSON.parse(event.data);
-        console.log("Received data:", data);
         handleWsMessage(data);
         //get chat history
       };
@@ -268,7 +279,6 @@ export default function FormModal({
         handleOnClose(event);
       };
       newWs.onerror = (ev) => {
-        console.log(ev, "error");
         if (flow.id === "") {
           connectWS();
         } else {
@@ -294,7 +304,6 @@ export default function FormModal({
   useEffect(() => {
     connectWS();
     return () => {
-      console.log("unmount");
       console.log(ws);
       if (ws.current) {
         ws.current.close();
@@ -513,7 +522,7 @@ export default function FormModal({
                           <div className="-mb-1">
                             <ToggleShadComponent
                               enabled={chatKey === key}
-                              setEnabled={() => {}}
+                              setEnabled={() => { }}
                               size="small"
                               disabled={true}
                             />