Merge branch 'dev' into frontChanges

src/backend/langflow/api/callback.py

@@ -1,12 +1,13 @@
+import asyncio
 from typing import Any
 
-from langchain.callbacks.base import AsyncCallbackHandler
+from langchain.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
 
 from langflow.api.schemas import ChatResponse
 
 
 # https://github.com/hwchase17/chat-langchain/blob/master/callback.py
-class StreamingLLMCallbackHandler(AsyncCallbackHandler):
+class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
     """Callback handler for streaming LLM responses."""
 
     def __init__(self, websocket):
@@ -15,3 +16,17 @@ class StreamingLLMCallbackHandler(AsyncCallbackHandler):
     async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
         resp = ChatResponse(message=token, type="stream", intermediate_steps="")
         await self.websocket.send_json(resp.dict())
+
+
+class StreamingLLMCallbackHandler(BaseCallbackHandler):
+    """Callback handler for streaming LLM responses."""
+
+    def __init__(self, websocket):
+        self.websocket = websocket
+
+    def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
+        resp = ChatResponse(message=token, type="stream", intermediate_steps="")
+
+        loop = asyncio.get_event_loop()
+        coroutine = self.websocket.send_json(resp.dict())
+        asyncio.run_coroutine_threadsafe(coroutine, loop)

src/backend/langflow/api/chat.py

@@ -1,4 +1,10 @@
-from fastapi import APIRouter, WebSocket
+from fastapi import (
+    APIRouter,
+    WebSocket,
+    WebSocketDisconnect,
+    WebSocketException,
+    status,
+)
 
 from langflow.api.chat_manager import ChatManager
 from langflow.utils.logger import logger
@@ -12,7 +18,9 @@ async def websocket_endpoint(client_id: str, websocket: WebSocket):
     """Websocket endpoint for chat."""
     try:
         await chat_manager.handle_websocket(client_id, websocket)
-    except Exception as e:
+    except WebSocketException as exc:
-        # Log stack trace
+        logger.error(exc)
-        logger.exception(e)
+        await websocket.close(code=status.WS_1011_INTERNAL_ERROR, reason=str(exc))
-        raise e
+    except WebSocketDisconnect as exc:
+        logger.error(exc)
+        await websocket.close(code=status.WS_1000_NORMAL_CLOSURE, reason=str(exc))

src/backend/langflow/api/chat_manager.py

@@ -3,7 +3,7 @@ import json
 from collections import defaultdict
 from typing import Dict, List
 
-from fastapi import WebSocket
+from fastapi import WebSocket, status
 
 from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
 from langflow.cache import cache_manager
@@ -47,7 +47,6 @@ class ChatManager:
     def __init__(self):
         self.active_connections: Dict[str, WebSocket] = {}
         self.chat_history = ChatHistory()
-        self.chat_history.attach(self.on_chat_history_update)
         self.cache_manager = cache_manager
         self.cache_manager.attach(self.update)
 
@@ -91,7 +90,7 @@ class ChatManager:
         self.active_connections[client_id] = websocket
 
     def disconnect(self, client_id: str):
-        del self.active_connections[client_id]
+        self.active_connections.pop(client_id, None)
 
     async def send_message(self, client_id: str, message: str):
         websocket = self.active_connections[client_id]
@@ -109,7 +108,7 @@ class ChatManager:
 
         graph_data = payload
         start_resp = ChatResponse(message=None, type="start", intermediate_steps="")
-        self.chat_history.add_message(client_id, start_resp)
+        await self.send_json(client_id, start_resp)
 
         is_first_message = len(self.chat_history.get_history(client_id=client_id)) == 0
         # Generate result and thought
@@ -143,11 +142,12 @@ class ChatManager:
                     break
 
         response = ChatResponse(
-            message=result or "",
+            message=result,
             intermediate_steps=intermediate_steps.strip(),
             type="end",
             files=file_responses,
         )
+        await self.send_json(client_id, response)
         self.chat_history.add_message(client_id, response)
 
     async def handle_websocket(self, client_id: str, websocket: WebSocket):
@@ -171,17 +171,24 @@ class ChatManager:
 
                 with self.cache_manager.set_client_id(client_id):
                     await self.process_message(client_id, payload)
+
         except Exception as e:
             # Handle any exceptions that might occur
             logger.exception(e)
             # send a message to the client
-            await self.send_message(client_id, str(e))
-            raise e
-        finally:
             await self.active_connections[client_id].close(
-                code=1000, reason="Client disconnected"
+                code=status.WS_1011_INTERNAL_ERROR, reason=str(e)[:120]
             )
             self.disconnect(client_id)
+        finally:
+            try:
+                connection = self.active_connections.get(client_id)
+                if connection:
+                    await connection.close(code=1000, reason="Client disconnected")
+                    self.disconnect(client_id)
+            except Exception as e:
+                logger.exception(e)
+            self.disconnect(client_id)
 
 
 async def process_graph(
@@ -203,8 +210,8 @@ async def process_graph(
     # Generate result and thought
     try:
         logger.debug("Generating result and thought")
-        result, intermediate_steps = get_result_and_steps(
+        result, intermediate_steps = await get_result_and_steps(
-            langchain_object, chat_message.message or ""
+            langchain_object, chat_message.message or "", websocket=websocket
         )
         logger.debug("Generated result and intermediate_steps")
         return result, intermediate_steps

src/backend/langflow/api/endpoints.py

@@ -1,8 +1,13 @@
 import logging
-from typing import Any, Dict
 
 from fastapi import APIRouter, HTTPException
 
+from langflow.api.schemas import (
+    ExportedFlow,
+    GraphData,
+    PredictRequest,
+    PredictResponse,
+)
 from langflow.interface.run import process_graph_cached
 from langflow.interface.types import build_langchain_types_dict
 
@@ -16,10 +21,14 @@ def get_all():
     return build_langchain_types_dict()
 
 
-@router.post("/predict")
+@router.post("/predict", response_model=PredictResponse)
-def get_load(data: Dict[str, Any]):
+async def get_load(predict_request: PredictRequest):
     try:
-        return process_graph_cached(data)
+        exported_flow: ExportedFlow = predict_request.exported_flow
+        graph_data: GraphData = exported_flow.data
+        data = graph_data.dict()
+        response = process_graph_cached(data, predict_request.message)
+        return PredictResponse(result=response.get("result", ""))
     except Exception as e:
         # Log stack trace
         logger.exception(e)

src/backend/langflow/api/schemas.py

@@ -1,8 +1,37 @@
-from typing import Any, Union
+from typing import Any, Dict, List, Union
 
 from pydantic import BaseModel, validator
 
 
+class GraphData(BaseModel):
+    """Data inside the exported flow."""
+
+    nodes: List[Dict[str, Any]]
+    edges: List[Dict[str, Any]]
+
+
+class ExportedFlow(BaseModel):
+    """Exported flow from LangFlow."""
+
+    description: str
+    name: str
+    id: str
+    data: GraphData
+
+
+class PredictRequest(BaseModel):
+    """Predict request schema."""
+
+    message: str
+    exported_flow: ExportedFlow
+
+
+class PredictResponse(BaseModel):
+    """Predict response schema."""
+
+    result: str
+
+
 class ChatMessage(BaseModel):
     """Chat message schema."""
 

src/backend/langflow/interface/agents/custom.py

@@ -1,3 +1,4 @@
+from abc import ABC
 from typing import Any, List, Optional
 
 from langchain import LLMChain
@@ -27,14 +28,31 @@ from langchain.agents.agent_toolkits.vectorstore.prompt import (
 from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS
 from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS as SQL_FORMAT_INSTRUCTIONS
 from langchain.base_language import BaseLanguageModel
-
 from langchain.memory.chat_memory import BaseChatMemory
 from langchain.sql_database import SQLDatabase
 from langchain.tools.python.tool import PythonAstREPLTool
 from langchain.tools.sql_database.prompt import QUERY_CHECKER
 
 
-class JsonAgent(AgentExecutor):
+class CustomAgentExecutor(AgentExecutor, ABC):
+    """Custom agent executor"""
+
+    @staticmethod
+    def function_name():
+        return "CustomAgentExecutor"
+
+    @classmethod
+    def initialize(cls, *args, **kwargs):
+        pass
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def run(self, *args, **kwargs):
+        return super().run(*args, **kwargs)
+
+
+class JsonAgent(CustomAgentExecutor):
     """Json agent"""
 
     @staticmethod
@@ -70,7 +88,7 @@ class JsonAgent(AgentExecutor):
         return super().run(*args, **kwargs)
 
 
-class CSVAgent(AgentExecutor):
+class CSVAgent(CustomAgentExecutor):
     """CSV agent"""
 
     @staticmethod
@@ -118,7 +136,7 @@ class CSVAgent(AgentExecutor):
         return super().run(*args, **kwargs)
 
 
-class VectorStoreAgent(AgentExecutor):
+class VectorStoreAgent(CustomAgentExecutor):
     """Vector Store agent"""
 
     @staticmethod
@@ -156,7 +174,7 @@ class VectorStoreAgent(AgentExecutor):
         return super().run(*args, **kwargs)
 
 
-class SQLAgent(AgentExecutor):
+class SQLAgent(CustomAgentExecutor):
     """SQL agent"""
 
     @staticmethod
@@ -228,7 +246,7 @@ class SQLAgent(AgentExecutor):
         return super().run(*args, **kwargs)
 
 
-class VectorStoreRouterAgent(AgentExecutor):
+class VectorStoreRouterAgent(CustomAgentExecutor):
     """Vector Store Router Agent"""
 
     @staticmethod
@@ -267,7 +285,7 @@ class VectorStoreRouterAgent(AgentExecutor):
         return super().run(*args, **kwargs)
 
 
-class InitializeAgent(AgentExecutor):
+class InitializeAgent(CustomAgentExecutor):
     """Implementation of initialize_agent function"""
 
     @staticmethod

src/backend/langflow/interface/importing/utils.py

@@ -5,9 +5,9 @@ from typing import Any, Type
 
 from langchain import PromptTemplate
 from langchain.agents import Agent
+from langchain.base_language import BaseLanguageModel
 from langchain.chains.base import Chain
 from langchain.chat_models.base import BaseChatModel
-from langchain.base_language import BaseLanguageModel
 from langchain.tools import BaseTool
 
 

src/backend/langflow/interface/loading.py

@@ -13,9 +13,9 @@ from langchain.agents.load_tools import (
 )
 from langchain.agents.loading import load_agent_from_config
 from langchain.agents.tools import Tool
+from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.chains.loading import load_chain_from_config
-from langchain.base_language import BaseLanguageModel
 from langchain.llms.loading import load_llm_from_config
 from pydantic import ValidationError
 
@@ -30,88 +30,19 @@ from langflow.utils import util, validate
 
 def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
     """Instantiate class from module type and key, and params"""
+    params = convert_params_to_sets(params)
+
     if node_type in CUSTOM_AGENTS:
-        if custom_agent := CUSTOM_AGENTS.get(node_type):
+        custom_agent = CUSTOM_AGENTS.get(node_type)
-            return custom_agent.initialize(**params)  # type: ignore
+        if custom_agent:
-    params = process_params(params)
+            return custom_agent.initialize(**params)
+
     class_object = import_by_type(_type=base_type, name=node_type)
-    # check if it is a class before using issubclass
+    return instantiate_based_on_type(class_object, base_type, node_type, params)
-
-    # if isinstance(class_object, type) and issubclass(class_object, BaseModel):
-    #     # validate params
-    #     fields = class_object.__fields__
-    #     params = {key: value for key, value in params.items() if key in fields}
-
-    if base_type == "agents":
-        # We need to initialize it differently
-        return load_agent_executor(class_object, params)
-    elif base_type == "prompts":
-        if node_type == "ZeroShotPrompt":
-            if "tools" not in params:
-                params["tools"] = []
-            return ZeroShotAgent.create_prompt(**params)
-    elif base_type == "tools":
-        if node_type == "JsonSpec":
-            params["dict_"] = load_file_into_dict(params.pop("path"))
-            return class_object(**params)
-        elif node_type == "PythonFunction":
-            # If the node_type is "PythonFunction"
-            # we need to get the function from the params
-            # which will be a str containing a python function
-            # and then we need to compile it and return the function
-            # as the instance
-            function_string = params["code"]
-            if isinstance(function_string, str):
-                return validate.eval_function(function_string)
-            raise ValueError("Function should be a string")
-        elif node_type.lower() == "tool":
-            return class_object(**params)
-    elif base_type == "toolkits":
-        loaded_toolkit = class_object(**params)
-        # Check if node_type has a loader
-        if toolkits_creator.has_create_function(node_type):
-            return load_toolkits_executor(node_type, loaded_toolkit, params)
-        return loaded_toolkit
-    elif base_type == "embeddings":
-        # ? Why remove model from params?
-        try:
-            params.pop("model")
-        except KeyError:
-            pass
-        # remove all params that are not in class_object.__fields__
-        try:
-            return class_object(**params)
-        except ValidationError:
-            params = {
-                key: value
-                for key, value in params.items()
-                if key in class_object.__fields__
-            }
-            return class_object(**params)
-    elif base_type == "vectorstores":
-        if len(params.get("documents", [])) == 0:
-            # Error when the pdf or other source was not correctly
-            # loaded.
-            raise ValueError(
-                "The source you provided did not load correctly or was empty."
-                "This may cause an error in the vectorstore."
-            )
-        return class_object.from_documents(**params)
-    elif base_type == "documentloaders":
-        return class_object(**params).load()
-    elif base_type == "textsplitters":
-        documents = params.pop("documents")
-        text_splitter = class_object(**params)
-        return text_splitter.split_documents(documents)
-    elif base_type == "utilities":
-        if node_type == "SQLDatabase":
-            return class_object.from_uri(params.pop("uri"))
-
-    return class_object(**params)
 
 
-def process_params(params):
+def convert_params_to_sets(params):
-    """Process params"""
+    """Convert certain params to sets"""
     if "allowed_special" in params:
         params["allowed_special"] = set(params["allowed_special"])
     if "disallowed_special" in params:
@@ -119,6 +50,100 @@ def process_params(params):
     return params
 
 
+def instantiate_based_on_type(class_object, base_type, node_type, params):
+    if base_type == "agents":
+        return instantiate_agent(class_object, params)
+    elif base_type == "prompts":
+        return instantiate_prompt(node_type, params)
+    elif base_type == "tools":
+        return instantiate_tool(node_type, class_object, params)
+    elif base_type == "toolkits":
+        return instantiate_toolkit(node_type, class_object, params)
+    elif base_type == "embeddings":
+        return instantiate_embedding(class_object, params)
+    elif base_type == "vectorstores":
+        return instantiate_vectorstore(class_object, params)
+    elif base_type == "documentloaders":
+        return instantiate_documentloader(class_object, params)
+    elif base_type == "textsplitters":
+        return instantiate_textsplitter(class_object, params)
+    elif base_type == "utilities":
+        return instantiate_utility(node_type, class_object, params)
+    else:
+        return class_object(**params)
+
+
+def instantiate_agent(class_object, params):
+    return load_agent_executor(class_object, params)
+
+
+def instantiate_prompt(node_type, params):
+    if node_type == "ZeroShotPrompt":
+        if "tools" not in params:
+            params["tools"] = []
+        return ZeroShotAgent.create_prompt(**params)
+    return None  # Or some other default action
+
+
+def instantiate_tool(node_type, class_object, params):
+    if node_type == "JsonSpec":
+        params["dict_"] = load_file_into_dict(params.pop("path"))
+        return class_object(**params)
+    elif node_type == "PythonFunction":
+        function_string = params["code"]
+        if isinstance(function_string, str):
+            return validate.eval_function(function_string)
+        raise ValueError("Function should be a string")
+    elif node_type.lower() == "tool":
+        return class_object(**params)
+    return None  # Or some other default action
+
+
+def instantiate_toolkit(node_type, class_object, params):
+    loaded_toolkit = class_object(**params)
+    if toolkits_creator.has_create_function(node_type):
+        return load_toolkits_executor(node_type, loaded_toolkit, params)
+    return loaded_toolkit
+
+
+def instantiate_embedding(class_object, params):
+    params.pop("model", None)
+    try:
+        return class_object(**params)
+    except ValidationError:
+        params = {
+            key: value
+            for key, value in params.items()
+            if key in class_object.__fields__
+        }
+        return class_object(**params)
+
+
+def instantiate_vectorstore(class_object, params):
+    if len(params.get("documents", [])) == 0:
+        raise ValueError(
+            "The source you provided did not load correctly or was empty."
+            "This may cause an error in the vectorstore."
+        )
+    return class_object.from_documents(**params)
+
+
+def instantiate_documentloader(class_object, params):
+    return class_object(**params).load()
+
+
+def instantiate_textsplitter(class_object, params):
+    documents = params.pop("documents")
+    text_splitter = class_object(**params)
+    return text_splitter.split_documents(documents)
+
+
+def instantiate_utility(node_type, class_object, params):
+    if node_type == "SQLDatabase":
+        return class_object.from_uri(params.pop("uri"))
+    return class_object(**params)
+
+
 def load_flow_from_json(path: str, build=True):
     # This is done to avoid circular imports
     from langflow.graph import Graph

src/backend/langflow/interface/run.py

@@ -4,9 +4,9 @@ from typing import Any, Dict
 
 from chromadb.errors import NotEnoughElementsException  # type: ignore
 
+from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler  # type: ignore
 from langflow.cache.base import compute_dict_hash, load_cache, memoize_dict
 from langflow.graph.graph import Graph
-from langflow.interface import loading
 from langflow.utils.logger import logger
 
 
@@ -66,47 +66,12 @@ def build_langchain_object(data_graph):
     return graph.build()
 
 
-def process_graph(data_graph: Dict[str, Any]):
+def process_graph_cached(data_graph: Dict[str, Any], message: str):
     """
     Process graph by extracting input variables and replacing ZeroShotPrompt
     with PromptTemplate,then run the graph and return the result and thought.
     """
     # Load langchain object
-    logger.debug("Loading langchain object")
-    message = data_graph.pop("message", "")
-    is_first_message = len(data_graph.get("chatHistory", [])) == 0
-    computed_hash, langchain_object = load_langchain_object(
-        data_graph, is_first_message
-    )
-    logger.debug("Loaded langchain object")
-
-    if langchain_object is None:
-        # Raise user facing error
-        raise ValueError(
-            "There was an error loading the langchain_object. Please, check all the nodes and try again."
-        )
-
-    # Generate result and thought
-    logger.debug("Generating result and thought")
-    result, thought = get_result_and_steps(langchain_object, message)
-    logger.debug("Generated result and thought")
-
-    # Save langchain_object to cache
-    # We have to save it here because if the
-    # memory is updated we need to keep the new values
-    logger.debug("Saving langchain object to cache")
-    # save_cache(computed_hash, langchain_object, is_first_message)
-    logger.debug("Saved langchain object to cache")
-    return {"result": str(result), "thought": thought.strip()}
-
-
-def process_graph_cached(data_graph: Dict[str, Any]):
-    """
-    Process graph by extracting input variables and replacing ZeroShotPrompt
-    with PromptTemplate,then run the graph and return the result and thought.
-    """
-    # Load langchain object
-    message = data_graph.pop("message", "")
     is_first_message = len(data_graph.get("chatHistory", [])) == 0
     langchain_object = load_or_build_langchain_object(data_graph, is_first_message)
     logger.debug("Loaded langchain object")
@@ -119,7 +84,7 @@ def process_graph_cached(data_graph: Dict[str, Any]):
 
     # Generate result and thought
     logger.debug("Generating result and thought")
-    result, thought = get_result_and_steps(langchain_object, message)
+    result, thought = get_result_and_thought(langchain_object, message)
     logger.debug("Generated result and thought")
     return {"result": str(result), "thought": thought.strip()}
 
@@ -185,8 +150,9 @@ def fix_memory_inputs(langchain_object):
             update_memory_keys(langchain_object, possible_new_mem_key)
 
 
-def get_result_and_steps(langchain_object, message: str):
+async def get_result_and_steps(langchain_object, message: str, **kwargs):
     """Get result and thought from extracted json"""
+
     try:
         if hasattr(langchain_object, "verbose"):
             langchain_object.verbose = True
@@ -206,17 +172,21 @@ def get_result_and_steps(langchain_object, message: str):
             # https://github.com/hwchase17/langchain/issues/2068
             # Deactivating until we have a frontend solution
             # to display intermediate steps
-            langchain_object.return_intermediate_steps = False
+            langchain_object.return_intermediate_steps = True
 
         fix_memory_inputs(langchain_object)
 
         with io.StringIO() as output_buffer, contextlib.redirect_stdout(output_buffer):
             try:
-                output = langchain_object(chat_input)
+                async_callbacks = [AsyncStreamingLLMCallbackHandler(**kwargs)]
-            except ValueError as exc:
+                output = await langchain_object.acall(
+                    chat_input, callbacks=async_callbacks
+                )
+            except Exception as exc:
                 # make the error message more informative
                 logger.debug(f"Error: {str(exc)}")
-                output = langchain_object.run(chat_input)
+                sync_callbacks = [StreamingLLMCallbackHandler(**kwargs)]
+                output = langchain_object(chat_input, callbacks=sync_callbacks)
 
             intermediate_steps = (
                 output.get("intermediate_steps", []) if isinstance(output, dict) else []
@@ -241,7 +211,7 @@ def get_result_and_steps(langchain_object, message: str):
     return result, thought
 
 
-def async_get_result_and_steps(langchain_object, message: str):
+def get_result_and_thought(langchain_object, message: str):
     """Get result and thought from extracted json"""
     try:
         if hasattr(langchain_object, "verbose"):
@@ -296,34 +266,6 @@ def async_get_result_and_steps(langchain_object, message: str):
     return result, thought
 
 
-def get_result_and_thought(extracted_json: Dict[str, Any], message: str):
-    """Get result and thought from extracted json"""
-    try:
-        langchain_object = loading.load_langchain_type_from_config(
-            config=extracted_json
-        )
-        with io.StringIO() as output_buffer, contextlib.redirect_stdout(output_buffer):
-            output = langchain_object(message)
-            intermediate_steps = (
-                output.get("intermediate_steps", []) if isinstance(output, dict) else []
-            )
-            result = (
-                output.get(langchain_object.output_keys[0])
-                if isinstance(output, dict)
-                else output
-            )
-
-            if intermediate_steps:
-                thought = format_intermediate_steps(intermediate_steps)
-            else:
-                thought = output_buffer.getvalue()
-
-    except Exception as e:
-        result = f"Error: {str(e)}"
-        thought = ""
-    return result, thought
-
-
 def format_intermediate_steps(intermediate_steps):
     formatted_chain = "> Entering new AgentExecutor chain...\n"
     for step in intermediate_steps:

src/backend/langflow/interface/utils.py

@@ -4,13 +4,9 @@ import os
 from io import BytesIO
 
 import yaml
-from langchain.callbacks.manager import AsyncCallbackManager
+from langchain.base_language import BaseLanguageModel
-from langchain.chat_models import AzureChatOpenAI, ChatOpenAI
-from langchain.llms import AzureOpenAI, OpenAI
 from PIL.Image import Image
 
-from langflow.api.callback import StreamingLLMCallbackHandler
-
 
 def load_file_into_dict(file_path: str) -> dict:
     if not os.path.exists(file_path):
@@ -48,10 +44,7 @@ def try_setting_streaming_options(langchain_object, websocket):
         langchain_object.llm_chain, "llm"
     ):
         llm = langchain_object.llm_chain.llm
-    if isinstance(llm, (OpenAI, ChatOpenAI, AzureOpenAI, AzureChatOpenAI)):
+    if isinstance(llm, BaseLanguageModel):
         llm.streaming = bool(hasattr(llm, "streaming"))
-        stream_handler = StreamingLLMCallbackHandler(websocket)
-        stream_manager = AsyncCallbackManager([stream_handler])
-        llm.callback_manager = stream_manager
 
     return langchain_object

src/backend/langflow/template/nodes.py

@@ -123,6 +123,13 @@ class MidJourneyPromptChainNode(FrontendNode):
                 multiline=False,
                 name="llm",
             ),
+            TemplateField(
+                field_type="BaseChatMemory",
+                required=False,
+                show=True,
+                name="memory",
+                advanced=False,
+            ),
         ],
     )
     description: str = "MidJourneyPromptChain is a chain you can use to generate new MidJourney prompts."

src/frontend/dev.Dockerfile

@@ -15,7 +15,10 @@ WORKDIR /home/node/app
 COPY --chown=node:node . ./
 
 COPY ./set_proxy.sh .
-RUN chmod +x set_proxy.sh && ./set_proxy.sh
+RUN chmod +x set_proxy.sh && \
+    cat set_proxy.sh | tr -d '\r' > set_proxy_unix.sh && \
+    chmod +x set_proxy_unix.sh && \
+    ./set_proxy_unix.sh
 
 USER node
 

src/frontend/package.json

@@ -30,11 +30,16 @@
         "react-error-boundary": "^4.0.2",
         "react-icons": "^4.8.0",
         "react-laag": "^2.0.5",
+        "react-markdown": "^8.0.7",
         "react-router-dom": "^6.8.1",
         "react-scripts": "5.0.1",
         "react-syntax-highlighter": "^15.5.0",
+        "react-syntax-highlighter": "^15.5.0",
         "react-tabs": "^6.0.0",
         "reactflow": "^11.5.5",
+        "rehype-mathjax": "^4.0.2",
+        "remark-gfm": "^3.0.1",
+        "remark-math": "^5.1.1",
         "tailwindcss": "^3.2.6",
         "typescript": "^4.9.5",
         "web-vitals": "^2.1.4"
@@ -63,5 +68,8 @@
             "last 1 safari version"
         ]
     },
-    "proxy": "http://127.0.0.1:7860"
+    "proxy": "http://127.0.0.1:7860",
+    "devDependencies": {
+        "@tailwindcss/typography": "^0.5.9"
+    }
 }

src/frontend/set_proxy.sh

@@ -7,4 +7,4 @@ packagejson=$(cat package.json)
 
 packagejson=$(echo "$packagejson" | jq ".proxy = \"$backend_url\"")
 
-echo "$packagejson" > package.json
+echo "$packagejson" > package.json

src/frontend/src/App.css

@@ -3,45 +3,45 @@
 @tailwind utilities;
 
 .App {
-  text-align: center;
+	text-align: center;
 }
 
 .App-logo {
-  height: 40vmin;
+	height: 40vmin;
-  pointer-events: none;
+	pointer-events: none;
 }
 
 @media (prefers-reduced-motion: no-preference) {
-  .App-logo {
+	.App-logo {
-    animation: App-logo-spin infinite 20s linear;
+		animation: App-logo-spin infinite 20s linear;
-  }
+	}
 }
 
 .App-header {
-  background-color: #282c34;
+	background-color: #282c34;
-  min-height: 100vh;
+	min-height: 100vh;
-  display: flex;
+	display: flex;
-  flex-direction: column;
+	flex-direction: column;
-  align-items: center;
+	align-items: center;
-  justify-content: center;
+	justify-content: center;
-  font-size: calc(10px + 2vmin);
+	font-size: calc(10px + 2vmin);
-  color: white;
+	color: white;
 }
 
 .App-link {
-  color: #61dafb;
+	color: #61dafb;
 }
 
 @keyframes App-logo-spin {
-  from {
+	from {
-    transform: rotate(0deg);
+		transform: rotate(0deg);
-  }
+	}
-  to {
+	to {
-    transform: rotate(360deg);
+		transform: rotate(360deg);
-  }
+	}
 }
 
-@font-face{
+@font-face {
-  font-family: text-security-disc;
+	font-family: text-security-disc;
-  src: url("assets/text-security-disc.woff") format("woff");
+	src: url("assets/text-security-disc.woff") format("woff");
-}
+}

src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx

@@ -198,7 +198,9 @@ export default function ParameterComponent({
 							save();
 						}}
 					/>
-				):(<></>)}
+				) : (
+					<></>
+				)}
 			</>
 		</div>
 	);

src/frontend/src/CustomNodes/GenericNode/index.tsx

@@ -1,10 +1,5 @@
 import { BugAntIcon, Cog6ToothIcon, ExclamationCircleIcon, InformationCircleIcon, TrashIcon } from "@heroicons/react/24/outline";
-import {
+import { classNames, nodeColors, nodeIcons, toNormalCase } from "../../utils";
-  classNames,
-  nodeColors,
-  nodeIcons,
-  toNormalCase,
-} from "../../utils";
 import ParameterComponent from "./components/parameterComponent";
 import { typesContext } from "../../contexts/typesContext";
 import { useContext, useState, useEffect, useRef, Fragment } from "react";
@@ -17,11 +12,11 @@ import { TabsContext } from "../../contexts/tabsContext";
 import { debounce } from "../../utils";
 import Tooltip from "../../components/TooltipComponent";
 export default function GenericNode({
-  data,
+	data,
-  selected,
+	selected,
 }: {
-  data: NodeDataType;
+	data: NodeDataType;
-  selected: boolean;
+	selected: boolean;
 }) {
   const { setErrorData } = useContext(alertContext);
   const showError = useRef(true);
@@ -35,23 +30,22 @@ export default function GenericNode({
   const { reactFlowInstance } = useContext(typesContext);
   const [params, setParams] = useState([]);
 
+	useEffect(() => {
+		if (reactFlowInstance) {
+			setParams(Object.values(reactFlowInstance.toObject()));
+		}
+	}, [save]);
 
-  useEffect(() => {
+	const validateNode = useCallback(
-    if (reactFlowInstance) {
+		debounce(async () => {
-      setParams(Object.values(reactFlowInstance.toObject()));
+			try {
-    }
+				const response = await fetch(`/validate/node/${data.id}`, {
-  }, [save]);
+					method: "POST",
-
+					headers: {
-  const validateNode = useCallback(
+						"Content-Type": "application/json",
-    debounce(async () => {
+					},
-      try {
+					body: JSON.stringify(reactFlowInstance.toObject()),
-        const response = await fetch(`/validate/node/${data.id}`, {
+				});
-          method: "POST",
-          headers: {
-            "Content-Type": "application/json",
-          },
-          body: JSON.stringify(reactFlowInstance.toObject()),
-        });
 
         if (response.status === 200) {
           let jsonResponse = await response.json();
@@ -84,18 +78,18 @@ export default function GenericNode({
     }
   }, [validationStatus]);
 
-  if (!Icon) {
+	if (!Icon) {
-    if (showError.current) {
+		if (showError.current) {
-      setErrorData({
+			setErrorData({
-        title: data.type
+				title: data.type
-          ? `The ${data.type} node could not be rendered, please review your json file`
+					? `The ${data.type} node could not be rendered, please review your json file`
-          : "There was a node that can't be rendered, please review your json file",
+					: "There was a node that can't be rendered, please review your json file",
-      });
+			});
-      showError.current = false;
+			showError.current = false;
-    }
+		}
-    deleteNode(data.id);
+		deleteNode(data.id);
-    return;
+		return;
-  }
+	}
 
   return (
     <div
@@ -168,12 +162,12 @@ export default function GenericNode({
           {data.node.description}
         </div>
 
-        <>
+				<>
-          {Object.keys(data.node.template)
+					{Object.keys(data.node.template)
-            .filter((t) => t.charAt(0) !== "_")
+						.filter((t) => t.charAt(0) !== "_")
-            .map((t: string, idx) => (
+						.map((t: string, idx) => (
-              <div key={idx}>
+							<div key={idx}>
-                {/* {idx === 0 ? (
+								{/* {idx === 0 ? (
 									<div
 										className={classNames(
 											"px-5 py-2 mt-2 dark:text-white text-center",
@@ -192,59 +186,59 @@ export default function GenericNode({
 								) : (
 									<></>
 								)} */}
-                {data.node.template[t].show &&
+								{data.node.template[t].show &&
-                !data.node.template[t].advanced ? (
+								!data.node.template[t].advanced ? (
-                  <ParameterComponent
+									<ParameterComponent
-                    data={data}
+										data={data}
-                    color={
+										color={
-                      nodeColors[types[data.node.template[t].type]] ??
+											nodeColors[types[data.node.template[t].type]] ??
-                      nodeColors.unknown
+											nodeColors.unknown
-                    }
+										}
-                    title={
+										title={
-                      data.node.template[t].display_name
+											data.node.template[t].display_name
-                        ? data.node.template[t].display_name
+												? data.node.template[t].display_name
-                        : data.node.template[t].name
+												: data.node.template[t].name
-                        ? toNormalCase(data.node.template[t].name)
+												? toNormalCase(data.node.template[t].name)
-                        : toNormalCase(t)
+												: toNormalCase(t)
-                    }
+										}
-                    name={t}
+										name={t}
-                    tooltipTitle={
+										tooltipTitle={
-                      "Type: " +
+											"Type: " +
-                      data.node.template[t].type +
+											data.node.template[t].type +
-                      (data.node.template[t].list ? " list" : "")
+											(data.node.template[t].list ? " list" : "")
-                    }
+										}
-                    required={data.node.template[t].required}
+										required={data.node.template[t].required}
-                    id={data.node.template[t].type + "|" + t + "|" + data.id}
+										id={data.node.template[t].type + "|" + t + "|" + data.id}
-                    left={true}
+										left={true}
-                    type={data.node.template[t].type}
+										type={data.node.template[t].type}
-                  />
+									/>
-                ) : (
+								) : (
-                  <></>
+									<></>
-                )}
+								)}
-              </div>
+							</div>
-            ))}
+						))}
-          <div
+					<div
-            className={classNames(
+						className={classNames(
-              Object.keys(data.node.template).length < 1 ? "hidden" : "",
+							Object.keys(data.node.template).length < 1 ? "hidden" : "",
-              "w-full flex justify-center"
+							"w-full flex justify-center"
-            )}
+						)}
-          >
+					>
-            {" "}
+						{" "}
-          </div>
+					</div>
-          {/* <div className="px-5 py-2 mt-2 dark:text-white text-center">
+					{/* <div className="px-5 py-2 mt-2 dark:text-white text-center">
 						Output
 					</div> */}
-          <ParameterComponent
+					<ParameterComponent
-            data={data}
+						data={data}
-            color={nodeColors[types[data.type]] ?? nodeColors.unknown}
+						color={nodeColors[types[data.type]] ?? nodeColors.unknown}
-            title={data.type}
+						title={data.type}
-            tooltipTitle={`Type: ${data.node.base_classes.join(" | ")}`}
+						tooltipTitle={`Type: ${data.node.base_classes.join(" | ")}`}
-            id={[data.type, data.id, ...data.node.base_classes].join("|")}
+						id={[data.type, data.id, ...data.node.base_classes].join("|")}
-            type={data.node.base_classes.join("|")}
+						type={data.node.base_classes.join("|")}
-            left={false}
+						left={false}
-          />
+					/>
-        </>
+				</>
-      </div>
+			</div>
-    </div>
+		</div>
-  );
+	);
 }

src/frontend/src/alerts/alertDropDown/components/singleAlertComponent/index.tsx

@@ -1,8 +1,8 @@
 import {
-  XCircleIcon,
+	XCircleIcon,
-  XMarkIcon,
+	XMarkIcon,
-  InformationCircleIcon,
+	InformationCircleIcon,
-  CheckCircleIcon,
+	CheckCircleIcon,
 } from "@heroicons/react/24/outline";
 import { Link } from "react-router-dom";
 import { Transition } from "@headlessui/react";
@@ -10,145 +10,145 @@ import { useState } from "react";
 import { SingleAlertComponentType } from "../../../../types/alerts";
 
 export default function SingleAlert({
-  dropItem,
+	dropItem,
-  removeAlert,
+	removeAlert,
 }: SingleAlertComponentType) {
-  const [show, setShow] = useState(true);
+	const [show, setShow] = useState(true);
-  const type = dropItem.type;
+	const type = dropItem.type;
 
-  return (
+	return (
-    <Transition
+		<Transition
-      className="relative"
+			className="relative"
-      show={show}
+			show={show}
-      appear={true}
+			appear={true}
-      enter="transition-transform duration-500 ease-out"
+			enter="transition-transform duration-500 ease-out"
-      enterFrom={"transform translate-x-[-100%]"}
+			enterFrom={"transform translate-x-[-100%]"}
-      enterTo={"transform translate-x-0"}
+			enterTo={"transform translate-x-0"}
-      leave="transition-transform duration-500 ease-in"
+			leave="transition-transform duration-500 ease-in"
-      leaveFrom={"transform translate-x-0"}
+			leaveFrom={"transform translate-x-0"}
-      leaveTo={"transform translate-x-[-100%]"}
+			leaveTo={"transform translate-x-[-100%]"}
-    >
+		>
-      {type === "error" ? (
+			{type === "error" ? (
-        <div
+				<div
-          className="flex bg-red-50 rounded-md p-3 mb-2 mx-2"
+					className="flex bg-red-50 rounded-md p-3 mb-2 mx-2"
-          key={dropItem.id}
+					key={dropItem.id}
-        >
+				>
-          <div className="flex-shrink-0">
+					<div className="flex-shrink-0">
-            <XCircleIcon className="h-5 w-5 text-red-400" aria-hidden="true" />
+						<XCircleIcon className="h-5 w-5 text-red-400" aria-hidden="true" />
-          </div>
+					</div>
-          <div className="ml-3">
+					<div className="ml-3">
-            <h3 className="text-sm font-medium text-red-800">
+						<h3 className="text-sm font-medium text-red-800">
-              {dropItem.title}
+							{dropItem.title}
-            </h3>
+						</h3>
-            {dropItem.list ? (
+						{dropItem.list ? (
-              <div className="mt-2 text-sm text-red-700">
+							<div className="mt-2 text-sm text-red-700">
-                <ul className="list-disc space-y-1 pl-5">
+								<ul className="list-disc space-y-1 pl-5">
-                  {dropItem.list.map((item, idx) => (
+									{dropItem.list.map((item, idx) => (
-                    <li key={idx}>{item}</li>
+										<li key={idx}>{item}</li>
-                  ))}
+									))}
-                </ul>
+								</ul>
-              </div>
+							</div>
-            ) : (
+						) : (
-              <></>
+							<></>
-            )}
+						)}
-          </div>
+					</div>
-          <div className="ml-auto pl-3">
+					<div className="ml-auto pl-3">
-            <div className="-mx-1.5 -my-1.5">
+						<div className="-mx-1.5 -my-1.5">
-              <button
+							<button
-                type="button"
+								type="button"
-                onClick={() => {
+								onClick={() => {
-                  setShow(false);
+									setShow(false);
-                  setTimeout(() => {
+									setTimeout(() => {
-                    removeAlert(dropItem.id);
+										removeAlert(dropItem.id);
-                  }, 500);
+									}, 500);
-                }}
+								}}
-                className="inline-flex rounded-md bg-red-50 p-1.5 text-red-500"
+								className="inline-flex rounded-md bg-red-50 p-1.5 text-red-500"
-              >
+							>
-                <span className="sr-only">Dismiss</span>
+								<span className="sr-only">Dismiss</span>
-                <XMarkIcon className="h-5 w-5" aria-hidden="true" />
+								<XMarkIcon className="h-5 w-5" aria-hidden="true" />
-              </button>
+							</button>
-            </div>
+						</div>
-          </div>
+					</div>
-        </div>
+				</div>
-      ) : type === "notice" ? (
+			) : type === "notice" ? (
-        <div
+				<div
-          className="flex rounded-md bg-blue-50 p-3 mb-2 mx-2"
+					className="flex rounded-md bg-blue-50 p-3 mb-2 mx-2"
-          key={dropItem.id}
+					key={dropItem.id}
-        >
+				>
-          <div className="flex-shrink-0">
+					<div className="flex-shrink-0">
-            <InformationCircleIcon
+						<InformationCircleIcon
-              className="h-5 w-5 text-blue-400"
+							className="h-5 w-5 text-blue-400"
-              aria-hidden="true"
+							aria-hidden="true"
-            />
+						/>
-          </div>
+					</div>
-          <div className="ml-3 flex-1 md:flex md:justify-between">
+					<div className="ml-3 flex-1 md:flex md:justify-between">
-            <p className="text-sm text-blue-700">{dropItem.title}</p>
+						<p className="text-sm text-blue-700">{dropItem.title}</p>
-            <p className="mt-3 text-sm md:mt-0 md:ml-6">
+						<p className="mt-3 text-sm md:mt-0 md:ml-6">
-              {dropItem.link ? (
+							{dropItem.link ? (
-                <Link
+								<Link
-                  to={dropItem.link}
+									to={dropItem.link}
-                  className="whitespace-nowrap font-medium text-blue-700 hover:text-blue-600"
+									className="whitespace-nowrap font-medium text-blue-700 hover:text-blue-600"
-                >
+								>
-                  Details
+									Details
-                </Link>
+								</Link>
-              ) : (
+							) : (
-                <></>
+								<></>
-              )}
+							)}
-            </p>
+						</p>
-          </div>
+					</div>
-          <div className="ml-auto pl-3">
+					<div className="ml-auto pl-3">
-            <div className="-mx-1.5 -my-1.5">
+						<div className="-mx-1.5 -my-1.5">
-              <button
+							<button
-                type="button"
+								type="button"
-                onClick={() => {
+								onClick={() => {
-                  setShow(false);
+									setShow(false);
-                  setTimeout(() => {
+									setTimeout(() => {
-                    removeAlert(dropItem.id);
+										removeAlert(dropItem.id);
-                  }, 500);
+									}, 500);
-                }}
+								}}
-                className="inline-flex rounded-md bg-blue-50 p-1.5 text-blue-500"
+								className="inline-flex rounded-md bg-blue-50 p-1.5 text-blue-500"
-              >
+							>
-                <span className="sr-only">Dismiss</span>
+								<span className="sr-only">Dismiss</span>
-                <XMarkIcon className="h-5 w-5" aria-hidden="true" />
+								<XMarkIcon className="h-5 w-5" aria-hidden="true" />
-              </button>
+							</button>
-            </div>
+						</div>
-          </div>
+					</div>
-        </div>
+				</div>
-      ) : (
+			) : (
-        <div
+				<div
-          className="flex bg-green-50 p-3 mb-2 mx-2 rounded-md"
+					className="flex bg-green-50 p-3 mb-2 mx-2 rounded-md"
-          key={dropItem.id}
+					key={dropItem.id}
-        >
+				>
-          <div className="flex-shrink-0">
+					<div className="flex-shrink-0">
-            <CheckCircleIcon
+						<CheckCircleIcon
-              className="h-5 w-5 text-green-400"
+							className="h-5 w-5 text-green-400"
-              aria-hidden="true"
+							aria-hidden="true"
-            />
+						/>
-          </div>
+					</div>
-          <div className="ml-3">
+					<div className="ml-3">
-            <p className="text-sm font-medium text-green-800">
+						<p className="text-sm font-medium text-green-800">
-              {dropItem.title}
+							{dropItem.title}
-            </p>
+						</p>
-          </div>
+					</div>
-          <div className="ml-auto pl-3">
+					<div className="ml-auto pl-3">
-            <div className="-mx-1.5 -my-1.5">
+						<div className="-mx-1.5 -my-1.5">
-              <button
+							<button
-                type="button"
+								type="button"
-                onClick={() => {
+								onClick={() => {
-                  setShow(false);
+									setShow(false);
-                  setTimeout(() => {
+									setTimeout(() => {
-                    removeAlert(dropItem.id);
+										removeAlert(dropItem.id);
-                  }, 500);
+									}, 500);
-                }}
+								}}
-                className="inline-flex rounded-md bg-green-50 p-1.5 text-green-500"
+								className="inline-flex rounded-md bg-green-50 p-1.5 text-green-500"
-              >
+							>
-                <span className="sr-only">Dismiss</span>
+								<span className="sr-only">Dismiss</span>
-                <XMarkIcon className="h-5 w-5" aria-hidden="true" />
+								<XMarkIcon className="h-5 w-5" aria-hidden="true" />
-              </button>
+							</button>
-            </div>
+						</div>
-          </div>
+					</div>
-        </div>
+				</div>
-      )}
+			)}
-    </Transition>
+		</Transition>
-  );
+	);
 }

src/frontend/src/alerts/alertDropDown/index.tsx

@@ -7,60 +7,60 @@ import { AlertDropdownType } from "../../types/alerts";
 import { PopUpContext } from "../../contexts/popUpContext";
 import { useOnClickOutside } from "../hooks/useOnClickOutside";
 export default function AlertDropdown({}: AlertDropdownType) {
-  const { closePopUp } = useContext(PopUpContext);
+	const { closePopUp } = useContext(PopUpContext);
-  const componentRef = useRef<HTMLDivElement>(null);
+	const componentRef = useRef<HTMLDivElement>(null);
 
-  // Use the custom hook
+	// Use the custom hook
-  useOnClickOutside(componentRef, () => {
+	useOnClickOutside(componentRef, () => {
-    closePopUp();
+		closePopUp();
-  });
+	});
 
-  const {
+	const {
-    notificationList,
+		notificationList,
-    clearNotificationList,
+		clearNotificationList,
-    removeFromNotificationList,
+		removeFromNotificationList,
-  } = useContext(alertContext);
+	} = useContext(alertContext);
 
-  return (
+	return (
-    <div
+		<div
-      ref={componentRef}
+			ref={componentRef}
-      className="z-10 py-3 pb-4 rounded-md bg-white dark:bg-gray-800 ring-1 ring-black ring-opacity-5 shadow-lg focus:outline-none overflow-hidden w-[16rem] h-[28rem] flex flex-col"
+			className="z-10 py-3 pb-4 rounded-md bg-white dark:bg-gray-800 ring-1 ring-black ring-opacity-5 shadow-lg focus:outline-none overflow-hidden w-[16rem] h-[28rem] flex flex-col"
-    >
+		>
-      <div className="flex pl-3 flex-row justify-between text-md font-medium text-gray-800 dark:text-gray-200">
+			<div className="flex pl-3 flex-row justify-between text-md font-medium text-gray-800 dark:text-gray-200">
-        Notifications
+				Notifications
-        <div className="flex gap-2 pr-3 ">
+				<div className="flex gap-2 pr-3 ">
-          <button
+					<button
-            className="text-gray-800 hover:text-red-500 dark:text-gray-200 dark:hover:text-red-500"
+						className="text-gray-800 hover:text-red-500 dark:text-gray-200 dark:hover:text-red-500"
-            onClick={() => {
+						onClick={() => {
-              closePopUp();
+							closePopUp();
-              setTimeout(clearNotificationList, 100);
+							setTimeout(clearNotificationList, 100);
-            }}
+						}}
-          >
+					>
-            <TrashIcon className="w-[1.1rem] h-[1.1rem]" />
+						<TrashIcon className="w-[1.1rem] h-[1.1rem]" />
-          </button>
+					</button>
-          <button
+					<button
-            className="text-gray-800 hover:text-red-500 dark:text-gray-200 dark:hover:text-red-500"
+						className="text-gray-800 hover:text-red-500 dark:text-gray-200 dark:hover:text-red-500"
-            onClick={closePopUp}
+						onClick={closePopUp}
-          >
+					>
-            <XMarkIcon className="h-5 w-5" />
+						<XMarkIcon className="h-5 w-5" />
-          </button>
+					</button>
-        </div>
+				</div>
-      </div>
+			</div>
-      <div className="mt-2 flex flex-col overflow-y-scroll w-full h-full scrollbar-hide text-gray-900 dark:text-gray-300">
+			<div className="mt-2 flex flex-col overflow-y-scroll w-full h-full scrollbar-hide text-gray-900 dark:text-gray-300">
-        {notificationList.length !== 0 ? (
+				{notificationList.length !== 0 ? (
-          notificationList.map((alertItem, index) => (
+					notificationList.map((alertItem, index) => (
-            <SingleAlert
+						<SingleAlert
-              key={alertItem.id}
+							key={alertItem.id}
-              dropItem={alertItem}
+							dropItem={alertItem}
-              removeAlert={removeFromNotificationList}
+							removeAlert={removeFromNotificationList}
-            />
+						/>
-          ))
+					))
-        ) : (
+				) : (
-          <div className="h-full w-full pb-16 text-gray-500 dark:text-gray-500 flex justify-center items-center">
+					<div className="h-full w-full pb-16 text-gray-500 dark:text-gray-500 flex justify-center items-center">
-            No new notifications
+						No new notifications
-          </div>
+					</div>
-        )}
+				)}
-      </div>
+			</div>
-    </div>
+		</div>
-  );
+	);
 }

src/frontend/src/alerts/hooks/useOnClickOutside/index.ts

@@ -1,33 +1,33 @@
 import { useEffect } from "react";
 
 export function useOnClickOutside(ref, handler) {
-  useEffect(() => {
+	useEffect(() => {
-    const listener = (event) => {
+		const listener = (event) => {
-      // Do nothing if clicking ref's element or its children
+			// Do nothing if clicking ref's element or its children
-      if (!ref.current || ref.current.contains(event.target)) {
+			if (!ref.current || ref.current.contains(event.target)) {
-        return;
+				return;
-      }
+			}
 
-      handler(event);
+			handler(event);
-    };
+		};
 
-    // Attach the listener to the document
+		// Attach the listener to the document
-    document.addEventListener("mousedown", listener, { passive: true });
+		document.addEventListener("mousedown", listener, { passive: true });
 
-    // Attach the listener to the react-flow instance
+		// Attach the listener to the react-flow instance
-    const reactFlowContainer = document.querySelector(".react-flow");
+		const reactFlowContainer = document.querySelector(".react-flow");
-    if (reactFlowContainer) {
+		if (reactFlowContainer) {
-      reactFlowContainer.addEventListener("mousedown", listener, {
+			reactFlowContainer.addEventListener("mousedown", listener, {
-        passive: true,
+				passive: true,
-      });
+			});
-    }
+		}
 
-    // Clean up the listener when the component is unmounted
+		// Clean up the listener when the component is unmounted
-    return () => {
+		return () => {
-      document.removeEventListener("mousedown", listener);
+			document.removeEventListener("mousedown", listener);
-      if (reactFlowContainer) {
+			if (reactFlowContainer) {
-        reactFlowContainer.removeEventListener("mousedown", listener);
+				reactFlowContainer.removeEventListener("mousedown", listener);
-      }
+			}
-    };
+		};
-  }, [ref, handler]); // Rerun only if ref or handler changes
+	}, [ref, handler]); // Rerun only if ref or handler changes
 }

src/frontend/src/components/CrashErrorComponent/index.tsx

@@ -2,12 +2,14 @@ export default function CrashErrorComponent({ error, resetErrorBoundary }) {
 	return (
 		<div className="fixed top-0 left-0 w-full h-full flex items-center justify-center bg-gray-800 bg-opacity-50 z-50">
 			<div className="bg-white max-w-4xl h-1/3 min-h-fit rounded-lg shadow-lg p-8 text-start flex flex-col justify-evenly">
-				<h1 className="text-red-500 text-3xl mb-4">Oops! An unknown error has occurred.</h1>
+				<h1 className="text-red-500 text-3xl mb-4">
+					Oops! An unknown error has occurred.
+				</h1>
 				<p className="text-gray-700 mb-4 text-xl">
-					Please click the 'Reset Application' button
+					Please click the 'Reset Application' button to restore the
-					to restore the application's state. If the error persists, please
+					application's state. If the error persists, please create an issue on
-					create an issue on our GitHub page. We apologize for any inconvenience
+					our GitHub page. We apologize for any inconvenience this may have
-					this may have caused.   
+					caused.
 				</p>
 				<div className="flex justify-center">
 					<button

src/frontend/src/components/ExtraSidebarComponent/index.tsx

@@ -6,119 +6,119 @@ import { classNames } from "../../utils";
 import { locationContext } from "../../contexts/locationContext";
 
 export default function ExtraSidebar() {
-  const {
+	const {
-    current,
+		current,
-    isStackedOpen,
+		isStackedOpen,
-    setIsStackedOpen,
+		setIsStackedOpen,
-    extraNavigation,
+		extraNavigation,
-    extraComponent,
+		extraComponent,
-  } = useContext(locationContext);
+	} = useContext(locationContext);
-  return (
+	return (
-    <>
+		<>
-      <aside
+			<aside
-        className={` ${
+				className={` ${
-          isStackedOpen ? "w-52" : "w-0 "
+					isStackedOpen ? "w-52" : "w-0 "
-        } flex-shrink-0 flex overflow-hidden flex-col border-r dark:border-r-gray-700 transition-all duration-500`}
+				} flex-shrink-0 flex overflow-hidden flex-col border-r dark:border-r-gray-700 transition-all duration-500`}
-      >
+			>
-        <div className="w-52 dark:bg-gray-800 border dark:border-gray-700  overflow-y-auto scrollbar-hide h-full flex flex-col items-start">
+				<div className="w-52 dark:bg-gray-800 border dark:border-gray-700  overflow-y-auto scrollbar-hide h-full flex flex-col items-start">
-          <div className="flex pt-1 px-4 justify-between align-middle w-full">
+					<div className="flex pt-1 px-4 justify-between align-middle w-full">
-            <span className="text-gray-900 dark:text-white py-[2px] font-medium ">
+						<span className="text-gray-900 dark:text-white py-[2px] font-medium ">
-              {extraNavigation.title}
+							{extraNavigation.title}
-            </span>
+						</span>
-          </div>
+					</div>
-          <div className="flex flex-grow flex-col w-full">
+					<div className="flex flex-grow flex-col w-full">
-            {extraNavigation.options ? (
+						{extraNavigation.options ? (
-              <div className="p-4">
+							<div className="p-4">
-                <nav className="flex-1 space-y-1">
+								<nav className="flex-1 space-y-1">
-                  {extraNavigation.options.map((item) =>
+									{extraNavigation.options.map((item) =>
-                    !item.children ? (
+										!item.children ? (
-                      <div key={item.name}>
+											<div key={item.name}>
-                        <Link
+												<Link
-                          to={item.href}
+													to={item.href}
-                          className={classNames(
+													className={classNames(
-                            item.href.split("/")[2] === current[4]
+														item.href.split("/")[2] === current[4]
-                              ? "bg-gray-100 text-gray-900"
+															? "bg-gray-100 text-gray-900"
-                              : "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
+															: "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
-                            "group w-full flex items-center pl-2 py-2 text-sm font-medium rounded-md"
+														"group w-full flex items-center pl-2 py-2 text-sm font-medium rounded-md"
-                          )}
+													)}
-                        >
+												>
-                          <item.icon
+													<item.icon
-                            className={classNames(
+														className={classNames(
-                              item.href.split("/")[2] === current[4]
+															item.href.split("/")[2] === current[4]
-                                ? "text-gray-500"
+																? "text-gray-500"
-                                : "text-gray-400 group-hover:text-gray-500",
+																: "text-gray-400 group-hover:text-gray-500",
-                              "mr-3 flex-shrink-0 h-6 w-6"
+															"mr-3 flex-shrink-0 h-6 w-6"
-                            )}
+														)}
-                          />
+													/>
-                          {item.name}
+													{item.name}
-                        </Link>
+												</Link>
-                      </div>
+											</div>
-                    ) : (
+										) : (
-                      <Disclosure
+											<Disclosure
-                        as="div"
+												as="div"
-                        key={item.name}
+												key={item.name}
-                        className="space-y-1"
+												className="space-y-1"
-                      >
+											>
-                        {({ open }) => (
+												{({ open }) => (
-                          <>
+													<>
-                            <Disclosure.Button
+														<Disclosure.Button
-                              className={classNames(
+															className={classNames(
-                                item.href.split("/")[2] === current[4]
+																item.href.split("/")[2] === current[4]
-                                  ? "bg-gray-100 text-gray-900"
+																	? "bg-gray-100 text-gray-900"
-                                  : "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
+																	: "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
-                                "group w-full flex items-center pl-2 pr-1 py-2 text-left text-sm font-medium rounded-md focus:outline-none focus:ring-2 focus:ring-indigo-500"
+																"group w-full flex items-center pl-2 pr-1 py-2 text-left text-sm font-medium rounded-md focus:outline-none focus:ring-2 focus:ring-indigo-500"
-                              )}
+															)}
-                            >
+														>
-                              <item.icon
+															<item.icon
-                                className="mr-3 h-6 w-6 flex-shrink-0 text-gray-400 group-hover:text-gray-500"
+																className="mr-3 h-6 w-6 flex-shrink-0 text-gray-400 group-hover:text-gray-500"
-                                aria-hidden="true"
+																aria-hidden="true"
-                              />
+															/>
-                              <span className="flex-1">{item.name}</span>
+															<span className="flex-1">{item.name}</span>
-                              <svg
+															<svg
-                                className={classNames(
+																className={classNames(
-                                  open
+																	open
-                                    ? "text-gray-400 rotate-90"
+																		? "text-gray-400 rotate-90"
-                                    : "text-gray-300",
+																		: "text-gray-300",
-                                  "ml-3 h-5 w-5 flex-shrink-0 transition-rotate duration-150 ease-in-out group-hover:text-gray-400"
+																	"ml-3 h-5 w-5 flex-shrink-0 transition-rotate duration-150 ease-in-out group-hover:text-gray-400"
-                                )}
+																)}
-                                viewBox="0 0 20 20"
+																viewBox="0 0 20 20"
-                                aria-hidden="true"
+																aria-hidden="true"
-                              >
+															>
-                                <path
+																<path
-                                  d="M6 6L14 10L6 14V6Z"
+																	d="M6 6L14 10L6 14V6Z"
-                                  fill="currentColor"
+																	fill="currentColor"
-                                />
+																/>
-                              </svg>
+															</svg>
-                            </Disclosure.Button>
+														</Disclosure.Button>
-                            <Disclosure.Panel className="space-y-1">
+														<Disclosure.Panel className="space-y-1">
-                              {item.children.map((subItem) => (
+															{item.children.map((subItem) => (
-                                <Link
+																<Link
-                                  key={subItem.name}
+																	key={subItem.name}
-                                  to={subItem.href}
+																	to={subItem.href}
-                                  className={classNames(
+																	className={classNames(
-                                    subItem.href.split("/")[3] === current[5]
+																		subItem.href.split("/")[3] === current[5]
-                                      ? "bg-gray-100 text-gray-900"
+																			? "bg-gray-100 text-gray-900"
-                                      : "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
+																			: "bg-white text-gray-600 hover:bg-gray-50 hover:text-gray-900",
-                                    "group flex w-full items-center rounded-md py-2 pl-11 pr-2 text-sm font-medium"
+																		"group flex w-full items-center rounded-md py-2 pl-11 pr-2 text-sm font-medium"
-                                  )}
+																	)}
-                                >
+																>
-                                  {subItem.name}
+																	{subItem.name}
-                                </Link>
+																</Link>
-                              ))}
+															))}
-                            </Disclosure.Panel>
+														</Disclosure.Panel>
-                          </>
+													</>
-                        )}
+												)}
-                      </Disclosure>
+											</Disclosure>
-                    )
+										)
-                  )}
+									)}
-                </nav>
+								</nav>
-              </div>
+							</div>
-            ) : (
+						) : (
-              extraComponent
+							extraComponent
-            )}
+						)}
-          </div>
+					</div>
-        </div>
+				</div>
-      </aside>
+			</aside>
-    </>
+		</>
-  );
+	);
 }

src/frontend/src/components/LightTooltipComponent/index.tsx

@@ -1,17 +1,17 @@
-import { styled } from '@mui/material/styles';
+import { styled } from "@mui/material/styles";
-import Tooltip, { TooltipProps, tooltipClasses } from '@mui/material/Tooltip';
+import Tooltip, { TooltipProps, tooltipClasses } from "@mui/material/Tooltip";
 
 export const LightTooltip = styled(({ className, ...props }: TooltipProps) => (
-  <Tooltip {...props} classes={{ popper: className }} />
+	<Tooltip {...props} classes={{ popper: className }} />
 ))(({ theme }) => ({
-  [`& .${tooltipClasses.tooltip}`]: {
+	[`& .${tooltipClasses.tooltip}`]: {
-    backgroundColor: theme.palette.common.white,
+		backgroundColor: theme.palette.common.white,
-    color: 'rgba(0, 0, 0, 0.87)',
+		color: "rgba(0, 0, 0, 0.87)",
-    boxShadow: theme.shadows[2],
+		boxShadow: theme.shadows[2],
-    fontSize: 14,
+		fontSize: 14,
-  },
+	},
-  [`& .${tooltipClasses.arrow}:before`]: {
+	[`& .${tooltipClasses.arrow}:before`]: {
-    color: theme.palette.common.white,
+		color: theme.palette.common.white,
-    boxShadow: theme.shadows[1],
+		boxShadow: theme.shadows[1],
-  },
+	},
-}));
+}));

src/frontend/src/components/TooltipComponent/index.tsx

@@ -2,6 +2,14 @@ import { ReactElement } from "react";
 import { LightTooltip } from "../LightTooltipComponent";
 import { TooltipComponentType } from "../../types/components";
 
-export default function Tooltip({ children, title,placement }:TooltipComponentType) {
+export default function Tooltip({
-  return <LightTooltip  placement={placement} title={title} arrow>{children}</LightTooltip>;
+	children,
+	title,
+	placement,
+}: TooltipComponentType) {
+	return (
+		<LightTooltip placement={placement} title={title} arrow>
+			{children}
+		</LightTooltip>
+	);
 }

src/frontend/src/components/chatComponent/chatMessage/index.tsx

@@ -1,9 +1,13 @@
-import { ChatBubbleLeftEllipsisIcon, ChatBubbleOvalLeftEllipsisIcon, PlusSmallIcon } from "@heroicons/react/24/outline";
+import {
+	ChatBubbleLeftEllipsisIcon,
+	ChatBubbleOvalLeftEllipsisIcon,
+	PlusSmallIcon,
+} from "@heroicons/react/24/outline";
 import { useState } from "react";
 import { ChatMessageType } from "../../../types/chat";
 import { nodeColors } from "../../../utils";
-var Convert = require('ansi-to-html');
+var Convert = require("ansi-to-html");
-var convert = new Convert({newline:true});
+var convert = new Convert({ newline: true });
 
 export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
 	const [hidden, setHidden] = useState(true);
@@ -29,13 +33,16 @@ export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
 								style={{ backgroundColor: nodeColors["thought"] }}
 								className=" text-start inline-block w-full pb-3 pt-3 px-5 cursor-pointer"
 								dangerouslySetInnerHTML={{
-									__html: convert.toHtml(chat.thought)
+									__html: convert.toHtml(chat.thought),
 								}}
 							></div>
 						)}
 						{chat.thought && chat.thought !== "" && !hidden && <br></br>}
-						<div  className="w-full rounded-b-md px-4 pb-3 pt-3 pr-8" style={{ backgroundColor: nodeColors["chat"] }}>
+						<div
-						{chat.message}
+							className="w-full rounded-b-md px-4 pb-3 pt-3 pr-8"
+							style={{ backgroundColor: nodeColors["chat"] }}
+						>
+							{chat.message}
 						</div>
 					</div>
 				</div>

src/frontend/src/components/chatComponent/chatTrigger/index.tsx

@@ -1,37 +1,45 @@
 import { Transition } from "@headlessui/react";
-import { Bars3CenterLeftIcon, ChatBubbleBottomCenterTextIcon } from "@heroicons/react/24/outline";
+import {
+	Bars3CenterLeftIcon,
+	ChatBubbleBottomCenterTextIcon,
+} from "@heroicons/react/24/outline";
 import { nodeColors } from "../../../utils";
 import { PopUpContext } from "../../../contexts/popUpContext";
 import { useContext } from "react";
 import ChatModal from "../../../modals/chatModal";
 
-export default function ChatTrigger({open, setOpen,flow}){
+export default function ChatTrigger({ open, setOpen }) {
-    const {openPopUp} = useContext(PopUpContext)
+	const { openPopUp } = useContext(PopUpContext);
-    return(<Transition
+	return (
-        show={!open}
+		<Transition
-        appear={true}
+			show={!open}
-        enter="transition ease-out duration-300"
+			appear={true}
-        enterFrom="translate-y-96"
+			enter="transition ease-out duration-300"
-        enterTo="translate-y-0"
+			enterFrom="translate-y-96"
-        leave="transition ease-in duration-300"
+			enterTo="translate-y-0"
-        leaveFrom="translate-y-0"
+			leave="transition ease-in duration-300"
-        leaveTo="translate-y-96"
+			leaveFrom="translate-y-0"
-    >
+			leaveTo="translate-y-96"
-        <div className="absolute bottom-2 right-3">
+		>
-            <div style={{backgroundColor:nodeColors['chat']}} className="border flex justify-center align-center py-1 px-3 w-12 h-12 rounded-full dark:bg-gray-800 dark:border-gray-600 dark:text-white">
+			<div className="absolute bottom-2 right-3">
-                <button
+				<div
-                    onClick={() => {
+					style={{ backgroundColor: nodeColors["chat"] }}
-                        setOpen(true);
+					className="border flex justify-center align-center py-1 px-3 w-12 h-12 rounded-full dark:bg-gray-800 dark:border-gray-600 dark:text-white"
-                    }}
+				>
-                >
+					<button
-                    <div className="flex gap-3  items-center">
+						onClick={() => {
-                        <ChatBubbleBottomCenterTextIcon
+							setOpen(true);
-                            className="h-6 w-6 mt-1"
+						}}
-                            style={{ color: "white" }}
+					>
-                        />
+						<div className="flex gap-3  items-center">
-                    </div>
+							<ChatBubbleBottomCenterTextIcon
-                </button>
+								className="h-6 w-6 mt-1"
-            </div>
+								style={{ color: "white" }}
-        </div>
+							/>
-    </Transition>)
+						</div>
-}
+					</button>
+				</div>
+			</div>
+		</Transition>
+	);
+}

src/frontend/src/components/chatComponent/index.tsx

@@ -7,26 +7,26 @@ import ChatModal from "../../modals/chatModal";
 const _ = require("lodash");
 
 export default function Chat({ flow }: ChatType) {
-  const [open, setOpen] = useState(false);
+	const [open, setOpen] = useState(false);
-  useEffect(() => {
+	useEffect(() => {
-    const handleKeyDown = (event: KeyboardEvent) => {
+		const handleKeyDown = (event: KeyboardEvent) => {
-      if (
+			if (
-        (event.key === "K" || event.key === "k") &&
+				(event.key === "K" || event.key === "k") &&
-        (event.metaKey || event.ctrlKey)
+				(event.metaKey || event.ctrlKey)
-      ) {
+			) {
-        event.preventDefault();
+				event.preventDefault();
-        setOpen((oldState) => !oldState);
+				setOpen((oldState) => !oldState);
-      }
+			}
-    };
+		};
-    document.addEventListener("keydown", handleKeyDown);
+		document.addEventListener("keydown", handleKeyDown);
-    return () => {
+		return () => {
-      document.removeEventListener("keydown", handleKeyDown);
+			document.removeEventListener("keydown", handleKeyDown);
-    };
+		};
-  }, []);
+	}, []);
-  return (
+	return (
-    <>
+		<>
-      <ChatModal key={flow.id} flow={flow} open={open} setOpen={setOpen} />
+			<ChatModal key={flow.id} flow={flow} open={open} setOpen={setOpen} />
-      <ChatTrigger open={open} setOpen={setOpen} flow={flow} />
+			<ChatTrigger open={open} setOpen={setOpen} />
-    </>
+		</>
-  );
+	);
 }

src/frontend/src/components/dropdownComponent/index.tsx

@@ -5,90 +5,90 @@ import { DropDownComponentType } from "../../types/components";
 import { classNames } from "../../utils";
 
 export default function Dropdown({
-  value,
+	value,
-  options,
+	options,
-  onSelect,
+	onSelect,
 }: DropDownComponentType) {
-  let [internalValue, setInternalValue] = useState(
+	let [internalValue, setInternalValue] = useState(
-    value === "" || !value ? "Choose an option" : value
+		value === "" || !value ? "Choose an option" : value
-  );
+	);
-  return (
+	return (
-    <>
+		<>
-      <Listbox
+			<Listbox
-        value={internalValue}
+				value={internalValue}
-        onChange={(value) => {
+				onChange={(value) => {
-          setInternalValue(value);
+					setInternalValue(value);
-          onSelect(value);
+					onSelect(value);
-        }}
+				}}
-      >
+			>
-        {({ open }) => (
+				{({ open }) => (
-          <>
+					<>
-            <div className="relative mt-1">
+						<div className="relative mt-1">
-              <Listbox.Button className="relative w-full cursor-default rounded-md border border-gray-300 bg-white dark:bg-gray-900 py-2 pl-3 pr-10 text-left shadow-sm focus:border-indigo-500 focus:outline-none focus:ring-1 focus:ring-indigo-500 sm:text-sm">
+							<Listbox.Button className="relative w-full cursor-default rounded-md border border-gray-300 bg-white dark:bg-gray-900 py-2 pl-3 pr-10 text-left shadow-sm focus:border-indigo-500 focus:outline-none focus:ring-1 focus:ring-indigo-500 sm:text-sm">
-                <span className="block w-max truncate">{internalValue}</span>
+								<span className="block w-max truncate">{internalValue}</span>
-                <span className="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-2">
+								<span className="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-2">
-                  <ChevronUpDownIcon
+									<ChevronUpDownIcon
-                    className="h-5 w-5 text-gray-400"
+										className="h-5 w-5 text-gray-400"
-                    aria-hidden="true"
+										aria-hidden="true"
-                  />
+									/>
-                </span>
+								</span>
-              </Listbox.Button>
+							</Listbox.Button>
 
-              <Transition
+							<Transition
-                show={open}
+								show={open}
-                as={Fragment}
+								as={Fragment}
-                leave="transition ease-in duration-100"
+								leave="transition ease-in duration-100"
-                leaveFrom="opacity-100"
+								leaveFrom="opacity-100"
-                leaveTo="opacity-0"
+								leaveTo="opacity-0"
-              >
+							>
-                <Listbox.Options className="absolute z-10 mt-1 max-h-60 w-full dark:bg-gray-800 overflow-auto rounded-md bg-white py-1 text-base shadow-lg ring-1 ring-black ring-opacity-5 focus:outline-none sm:text-sm">
+								<Listbox.Options className="absolute z-10 mt-1 max-h-60 w-full dark:bg-gray-800 overflow-auto rounded-md bg-white py-1 text-base shadow-lg ring-1 ring-black ring-opacity-5 focus:outline-none sm:text-sm">
-                  {options.map((option, id) => (
+									{options.map((option, id) => (
-                    <Listbox.Option
+										<Listbox.Option
-                      key={id}
+											key={id}
-                      className={({ active }) =>
+											className={({ active }) =>
-                        classNames(
+												classNames(
-                          active
+													active
-                            ? "text-white bg-indigo-600 dark:bg-indigo-500"
+														? "text-white bg-indigo-600 dark:bg-indigo-500"
-                            : "text-gray-900",
+														: "text-gray-900",
-                          "relative cursor-default select-none py-2 pl-3 pr-9 dark:text-gray-300 dark:bg-gray-800"
+													"relative cursor-default select-none py-2 pl-3 pr-9 dark:text-gray-300 dark:bg-gray-800"
-                        )
+												)
-                      }
+											}
-                      value={option}
+											value={option}
-                    >
+										>
-                      {({ selected, active }) => (
+											{({ selected, active }) => (
-                        <>
+												<>
-                          <span
+													<span
-                            className={classNames(
+														className={classNames(
-                              selected ? "font-semibold" : "font-normal",
+															selected ? "font-semibold" : "font-normal",
-                              "block truncate"
+															"block truncate"
-                            )}
+														)}
-                          >
+													>
-                            {option}
+														{option}
-                          </span>
+													</span>
 
-                          {selected ? (
+													{selected ? (
-                            <span
+														<span
-                              className={classNames(
+															className={classNames(
-                                active ? "text-white" : "text-indigo-600",
+																active ? "text-white" : "text-indigo-600",
-                                "absolute inset-y-0 right-0 flex items-center pr-4"
+																"absolute inset-y-0 right-0 flex items-center pr-4"
-                              )}
+															)}
-                            >
+														>
-                              <CheckIcon
+															<CheckIcon
-                                className="h-5 w-5"
+																className="h-5 w-5"
-                                aria-hidden="true"
+																aria-hidden="true"
-                              />
+															/>
-                            </span>
+														</span>
-                          ) : null}
+													) : null}
-                        </>
+												</>
-                      )}
+											)}
-                    </Listbox.Option>
+										</Listbox.Option>
-                  ))}
+									))}
-                </Listbox.Options>
+								</Listbox.Options>
-              </Transition>
+							</Transition>
-            </div>
+						</div>
-          </>
+					</>
-        )}
+				)}
-      </Listbox>
+			</Listbox>
-    </>
+		</>
-  );
+	);
 }

src/frontend/src/components/floatComponent/index.tsx

@@ -1,26 +1,33 @@
 import { useEffect, useState } from "react";
 import { FloatComponentType } from "../../types/components";
 
-export default function FloatComponent({value, onChange, disabled}: FloatComponentType){
+export default function FloatComponent({
-    const [myValue, setMyValue] = useState(value ?? "");
+	value,
-    useEffect(()=> {
+	onChange,
-        if(disabled){
+	disabled,
-            setMyValue("");
+}: FloatComponentType) {
-            onChange("");
+	const [myValue, setMyValue] = useState(value ?? "");
-        }
+	useEffect(() => {
-    }, [disabled, onChange])
+		if (disabled) {
-    return (
+			setMyValue("");
-        <div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
+			onChange("");
-            <input
+		}
-            type="number"
+	}, [disabled, onChange]);
-            value={myValue}
+	return (
-            className={"block w-full form-input dark:bg-gray-900 arrow-hide dark:border-gray-600 rounded-md border-gray-300 shadow-sm focus:border-indigo-500 focus:ring-indigo-500 sm:text-sm" + (disabled ? " bg-gray-200 dark:bg-gray-700" : "")}
+		<div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
-            placeholder="Type a number from zero to one"
+			<input
-            onChange={(e) => {
+				type="number"
-                setMyValue(e.target.value);
+				value={myValue}
-                onChange(e.target.value);
+				className={
-            }}
+					"block w-full form-input dark:bg-gray-900 arrow-hide dark:border-gray-600 rounded-md border-gray-300 shadow-sm focus:border-indigo-500 focus:ring-indigo-500 sm:text-sm" +
-            />
+					(disabled ? " bg-gray-200 dark:bg-gray-700" : "")
-        </div>
+				}
-    );
+				placeholder="Type a number from zero to one"
-}
+				onChange={(e) => {
+					setMyValue(e.target.value);
+					onChange(e.target.value);
+				}}
+			/>
+		</div>
+	);
+}

src/frontend/src/components/inputComponent/index.tsx

@@ -3,10 +3,10 @@ import { InputComponentType } from "../../types/components";
 import { classNames } from "../../utils";
 
 export default function InputComponent({
-  value,
+	value,
-  onChange,
+	onChange,
-  disabled,
+	disabled,
-  password,
+	password,
 }: InputComponentType) {
   const [myValue, setMyValue] = useState(value ?? "");
   const [pwdVisible, setPwdVisible] = useState(false);

src/frontend/src/components/inputFileComponent/index.tsx

@@ -9,7 +9,7 @@ export default function InputFileComponent({
 	disabled,
 	suffixes,
 	fileTypes,
-	onFileChange
+	onFileChange,
 }: FileComponentType) {
 	const [myValue, setMyValue] = useState(value);
 	const { setErrorData } = useContext(alertContext);
@@ -17,23 +17,23 @@ export default function InputFileComponent({
 		if (disabled) {
 			setMyValue("");
 			onChange("");
-			onFileChange("")
+			onFileChange("");
 		}
 	}, [disabled, onChange]);
 
 	function attachFile(fileReadEvent: ProgressEvent<FileReader>) {
 		fileReadEvent.preventDefault();
 		const file = fileReadEvent.target.result;
-		onFileChange(file as string)
+		onFileChange(file as string);
 	}
 
-	function checkFileType(fileName:string):boolean{
+	function checkFileType(fileName: string): boolean {
 		for (let index = 0; index < suffixes.length; index++) {
-			if(fileName.endsWith(suffixes[index])){
+			if (fileName.endsWith(suffixes[index])) {
-				return true
+				return true;
 			}
 		}
-		return false
+		return false;
 	}
 
 	const handleButtonClick = () => {
@@ -69,7 +69,7 @@ export default function InputFileComponent({
 		>
 			<div className="w-full flex items-center gap-3">
 				<span
-				onClick={handleButtonClick}
+					onClick={handleButtonClick}
 					className={
 						"truncate block w-full text-gray-500 px-3 py-2 rounded-md border border-gray-300 dark:border-gray-700 shadow-sm focus:border-indigo-500 focus:ring-indigo-500 sm:text-sm" +
 						(disabled ? " bg-gray-200" : "")

src/frontend/src/components/inputListComponent/index.tsx

@@ -4,51 +4,73 @@ import { InputListComponentType } from "../../types/components";
 
 var _ = require("lodash");
 
-export default function InputListComponent({ value, onChange, disabled}:InputListComponentType) {
+export default function InputListComponent({
-  const [inputList, setInputList] = useState(value ?? [""]);
+	value,
-  useEffect(()=> {
+	onChange,
-    if(disabled){
+	disabled,
-        setInputList([""]);
+}: InputListComponentType) {
-        onChange([""]);
+	const [inputList, setInputList] = useState(value ?? [""]);
-    }
+	useEffect(() => {
-}, [disabled, onChange])
+		if (disabled) {
-  return (
+			setInputList([""]);
-    <div className={(disabled ? "pointer-events-none cursor-not-allowed" : "") + "flex flex-col gap-3"}>
+			onChange([""]);
-      {inputList.map((i, idx) => (
+		}
-        <div key={idx} className="w-full flex gap-3">
+	}, [disabled, onChange]);
-          <input
+	return (
-            type="text"
+		<div
-            value={i}
+			className={
-            className={"block w-full form-input rounded-md border-gray-300 shadow-sm focus:border-indigo-500 focus:ring-indigo-500 sm:text-sm" + (disabled ? " bg-gray-200" : "")}
+				(disabled ? "pointer-events-none cursor-not-allowed" : "") +
-            placeholder="Type a text"
+				"flex flex-col gap-3"
-            onChange={(e) => {
+			}
-              setInputList((old) => {
+		>
-                let newInputList = _.cloneDeep(old);
+			{inputList.map((i, idx) => (
-                newInputList[idx] = e.target.value;
+				<div key={idx} className="w-full flex gap-3">
-                return newInputList;
+					<input
-              });
+						type="text"
-              onChange(inputList);
+						value={i}
-            }}
+						className={
-          />
+							"block w-full form-input rounded-md border-gray-300 shadow-sm focus:border-indigo-500 focus:ring-indigo-500 sm:text-sm" +
-          {idx === inputList.length - 1 ? 
+							(disabled ? " bg-gray-200" : "")
-          <button onClick={() => {setInputList((old) => {
+						}
-            let newInputList = _.cloneDeep(old);
+						placeholder="Type a text"
-            newInputList.push('');
+						onChange={(e) => {
-            return newInputList;
+							setInputList((old) => {
-          });
+								let newInputList = _.cloneDeep(old);
-          onChange(inputList);}}>
+								newInputList[idx] = e.target.value;
-            <PlusIcon className="w-4 h-4 hover:text-blue-600" />
+								return newInputList;
-          </button> 
+							});
-          : <button onClick={() => {setInputList((old) => {
+							onChange(inputList);
-            let newInputList = _.cloneDeep(old);
+						}}
-            newInputList.splice(idx, 1);
+					/>
-            return newInputList;
+					{idx === inputList.length - 1 ? (
-          });
+						<button
-          onChange(inputList);}}>
+							onClick={() => {
-            <XMarkIcon className="w-4 h-4 hover:text-red-600" />
+								setInputList((old) => {
-            </button>}
+									let newInputList = _.cloneDeep(old);
-        </div>
+									newInputList.push("");
-      ))}
+									return newInputList;
-    </div>
+								});
-  );
+								onChange(inputList);
+							}}
+						>
+							<PlusIcon className="w-4 h-4 hover:text-blue-600" />
+						</button>
+					) : (
+						<button
+							onClick={() => {
+								setInputList((old) => {
+									let newInputList = _.cloneDeep(old);
+									newInputList.splice(idx, 1);
+									return newInputList;
+								});
+								onChange(inputList);
+							}}
+						>
+							<XMarkIcon className="w-4 h-4 hover:text-red-600" />
+						</button>
+					)}
+				</div>
+			))}
+		</div>
+	);
 }

src/frontend/src/components/intComponent/index.tsx

@@ -14,12 +14,23 @@ export default function IntComponent({
 		}
 	}, [disabled, onChange]);
 	return (
-		<div className={disabled ? "pointer-events-none cursor-not-allowed w-full" : "w-full"}>
+		<div
+			className={
+				disabled ? "pointer-events-none cursor-not-allowed w-full" : "w-full"
+			}
+		>
 			<input
 				onKeyDown={(event) => {
-                    if (event.key !== 'Backspace' && event.key !== 'Enter' && event.key !== 'Delete' && event.key !== 'ArrowLeft' && event.key !== 'ArrowRight' && !/^[-]?\d*$/.test(event.key)) {
+					if (
-                        event.preventDefault();
+						event.key !== "Backspace" &&
-                    }
+						event.key !== "Enter" &&
+						event.key !== "Delete" &&
+						event.key !== "ArrowLeft" &&
+						event.key !== "ArrowRight" &&
+						!/^[-]?\d*$/.test(event.key)
+					) {
+						event.preventDefault();
+					}
 				}}
 				type="number"
 				value={myValue}

src/frontend/src/components/loadingComponent/index.tsx

@@ -1,17 +1,28 @@
-type LoadingComponentProps={
+type LoadingComponentProps = {
-    remSize:number
+	remSize: number;
+};
+
+export default function LoadingComponent({ remSize }: LoadingComponentProps) {
+	return (
+		<div role="status" className="w-min m-auto">
+			<svg
+				aria-hidden="true"
+				className={`w-${remSize} h-${remSize} mr-2 text-gray-200 animate-spin dark:text-gray-600 fill-blue-600`}
+				viewBox="0 0 100 101"
+				fill="none"
+				xmlns="http://www.w3.org/2000/svg"
+			>
+				<path
+					d="M100 50.5908C100 78.2051 77.6142 100.591 50 100.591C22.3858 100.591 0 78.2051 0 50.5908C0 22.9766 22.3858 0.59082 50 0.59082C77.6142 0.59082 100 22.9766 100 50.5908ZM9.08144 50.5908C9.08144 73.1895 27.4013 91.5094 50 91.5094C72.5987 91.5094 90.9186 73.1895 90.9186 50.5908C90.9186 27.9921 72.5987 9.67226 50 9.67226C27.4013 9.67226 9.08144 27.9921 9.08144 50.5908Z"
+					fill="currentColor"
+				/>
+				<path
+					d="M93.9676 39.0409C96.393 38.4038 97.8624 35.9116 97.0079 33.5539C95.2932 28.8227 92.871 24.3692 89.8167 20.348C85.8452 15.1192 80.8826 10.7238 75.2124 7.41289C69.5422 4.10194 63.2754 1.94025 56.7698 1.05124C51.7666 0.367541 46.6976 0.446843 41.7345 1.27873C39.2613 1.69328 37.813 4.19778 38.4501 6.62326C39.0873 9.04874 41.5694 10.4717 44.0505 10.1071C47.8511 9.54855 51.7191 9.52689 55.5402 10.0491C60.8642 10.7766 65.9928 12.5457 70.6331 15.2552C75.2735 17.9648 79.3347 21.5619 82.5849 25.841C84.9175 28.9121 86.7997 32.2913 88.1811 35.8758C89.083 38.2158 91.5421 39.6781 93.9676 39.0409Z"
+					fill="currentFill"
+				/>
+			</svg>
+			<br></br>
+			<span className="animate-pulse text-blue-600 text-lg">Loading...</span>
+		</div>
+	);
 }
-
-
-export default function LoadingComponent({remSize}:LoadingComponentProps){
-    return(
-        <div role="status" className="w-min m-auto">
-            <svg aria-hidden="true" className={`w-${remSize} h-${remSize} mr-2 text-gray-200 animate-spin dark:text-gray-600 fill-blue-600`} viewBox="0 0 100 101" fill="none" xmlns="http://www.w3.org/2000/svg">
-                <path d="M100 50.5908C100 78.2051 77.6142 100.591 50 100.591C22.3858 100.591 0 78.2051 0 50.5908C0 22.9766 22.3858 0.59082 50 0.59082C77.6142 0.59082 100 22.9766 100 50.5908ZM9.08144 50.5908C9.08144 73.1895 27.4013 91.5094 50 91.5094C72.5987 91.5094 90.9186 73.1895 90.9186 50.5908C90.9186 27.9921 72.5987 9.67226 50 9.67226C27.4013 9.67226 9.08144 27.9921 9.08144 50.5908Z" fill="currentColor"/>
-                <path d="M93.9676 39.0409C96.393 38.4038 97.8624 35.9116 97.0079 33.5539C95.2932 28.8227 92.871 24.3692 89.8167 20.348C85.8452 15.1192 80.8826 10.7238 75.2124 7.41289C69.5422 4.10194 63.2754 1.94025 56.7698 1.05124C51.7666 0.367541 46.6976 0.446843 41.7345 1.27873C39.2613 1.69328 37.813 4.19778 38.4501 6.62326C39.0873 9.04874 41.5694 10.4717 44.0505 10.1071C47.8511 9.54855 51.7191 9.52689 55.5402 10.0491C60.8642 10.7766 65.9928 12.5457 70.6331 15.2552C75.2735 17.9648 79.3347 21.5619 82.5849 25.841C84.9175 28.9121 86.7997 32.2913 88.1811 35.8758C89.083 38.2158 91.5421 39.6781 93.9676 39.0409Z" fill="currentFill"/>
-            </svg>
-            <br></br>
-            <span className="animate-pulse text-blue-600 text-lg">Loading...</span>
-        </div>
-    )
-}

src/frontend/src/components/toggleComponent/index.tsx

@@ -3,73 +3,80 @@ import { classNames } from "../../utils";
 import { useEffect } from "react";
 import { ToggleComponentType } from "../../types/components";
 
-export default function ToggleComponent({ enabled, setEnabled, disabled }:ToggleComponentType) {
+export default function ToggleComponent({
-  useEffect(()=> {
+	enabled,
-    if(disabled){
+	setEnabled,
-        setEnabled(false);
+	disabled,
-    }
+}: ToggleComponentType) {
-}, [disabled, setEnabled])
+	useEffect(() => {
-  return (
+		if (disabled) {
-    <div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
+			setEnabled(false);
-      <Switch
+		}
-        checked={enabled}
+	}, [disabled, setEnabled]);
-        onChange={(x:boolean) => {
+	return (
-          setEnabled(x);
+		<div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
-        }}
+			<Switch
-        className={classNames(
+				checked={enabled}
-          enabled ? "bg-indigo-600" : "bg-gray-200 dark:bg-gray-600",
+				onChange={(x: boolean) => {
-          "relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out "
+					setEnabled(x);
-        )}
+				}}
-      >
+				className={classNames(
-        <span className="sr-only">Use setting</span>
+					enabled ? "bg-indigo-600" : "bg-gray-200 dark:bg-gray-600",
-        <span
+					"relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out "
-          className={classNames(
+				)}
-            enabled ? "translate-x-5" : "translate-x-0",
+			>
-            "pointer-events-none relative inline-block h-5 w-5 transform rounded-full  shadow ring-0 transition duration-200 ease-in-out", disabled ? "bg-gray-200 dark:bg-gray-600" : "bg-white dark:bg-gray-800"
+				<span className="sr-only">Use setting</span>
-          )}
+				<span
-        >
+					className={classNames(
-          <span
+						enabled ? "translate-x-5" : "translate-x-0",
-            className={classNames(
+						"pointer-events-none relative inline-block h-5 w-5 transform rounded-full  shadow ring-0 transition duration-200 ease-in-out",
-              enabled
+						disabled
-                ? "opacity-0 ease-out duration-100"
+							? "bg-gray-200 dark:bg-gray-600"
-                : "opacity-100 ease-in duration-200",
+							: "bg-white dark:bg-gray-800"
-              "absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
+					)}
-            )}
+				>
-            aria-hidden="true"
+					<span
-          >
+						className={classNames(
-            <svg
+							enabled
-              className="h-3 w-3 text-gray-400"
+								? "opacity-0 ease-out duration-100"
-              fill="none"
+								: "opacity-100 ease-in duration-200",
-              viewBox="0 0 12 12"
+							"absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
-            >
+						)}
-              <path
+						aria-hidden="true"
-                d="M4 8l2-2m0 0l2-2M6 6L4 4m2 2l2 2"
+					>
-                stroke="currentColor"
+						<svg
-                strokeWidth={2}
+							className="h-3 w-3 text-gray-400"
-                strokeLinecap="round"
+							fill="none"
-                strokeLinejoin="round"
+							viewBox="0 0 12 12"
-              />
+						>
-            </svg>
+							<path
-          </span>
+								d="M4 8l2-2m0 0l2-2M6 6L4 4m2 2l2 2"
-          <span
+								stroke="currentColor"
-            className={classNames(
+								strokeWidth={2}
-              enabled
+								strokeLinecap="round"
-                ? "opacity-100 ease-in duration-200"
+								strokeLinejoin="round"
-                : "opacity-0 ease-out duration-100",
+							/>
-              "absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
+						</svg>
-            )}
+					</span>
-            aria-hidden="true"
+					<span
-          >
+						className={classNames(
-            <svg
+							enabled
-              className="h-3 w-3 text-indigo-600"
+								? "opacity-100 ease-in duration-200"
-              fill="currentColor"
+								: "opacity-0 ease-out duration-100",
-              viewBox="0 0 12 12"
+							"absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
-            >
+						)}
-              <path d="M3.707 5.293a1 1 0 00-1.414 1.414l1.414-1.414zM5 8l-.707.707a1 1 0 001.414 0L5 8zm4.707-3.293a1 1 0 00-1.414-1.414l1.414 1.414zm-7.414 2l2 2 1.414-1.414-2-2-1.414 1.414zm3.414 2l4-4-1.414-1.414-4 4 1.414 1.414z" />
+						aria-hidden="true"
-            </svg>
+					>
-          </span>
+						<svg
-        </span>
+							className="h-3 w-3 text-indigo-600"
-      </Switch>
+							fill="currentColor"
-    </div>
+							viewBox="0 0 12 12"
-  );
+						>
+							<path d="M3.707 5.293a1 1 0 00-1.414 1.414l1.414-1.414zM5 8l-.707.707a1 1 0 001.414 0L5 8zm4.707-3.293a1 1 0 00-1.414-1.414l1.414 1.414zm-7.414 2l2 2 1.414-1.414-2-2-1.414 1.414zm3.414 2l4-4-1.414-1.414-4 4 1.414 1.414z" />
+						</svg>
+					</span>
+				</span>
+			</Switch>
+		</div>
+	);
 }

src/frontend/src/contexts/alertContext.tsx

@@ -20,13 +20,13 @@ type alertContextType = {
 	notificationCenter: boolean;
 	setNotificationCenter: (newState: boolean) => void;
 	notificationList: Array<AlertItemType>;
-	pushNotificationList: (Object:AlertItemType) => void;
+	pushNotificationList: (Object: AlertItemType) => void;
 	clearNotificationList: () => void;
 	removeFromNotificationList: (index: string) => void;
 };
 
 //initial values to alertContextType
-const initialValue:alertContextType = {
+const initialValue: alertContextType = {
 	errorData: { title: "", list: [] },
 	setErrorData: () => {},
 	errorOpen: false,
@@ -49,7 +49,7 @@ const initialValue:alertContextType = {
 
 export const alertContext = createContext<alertContextType>(initialValue);
 
-export function AlertProvider({ children }:{children:ReactNode}) {
+export function AlertProvider({ children }: { children: ReactNode }) {
 	const [errorData, setErrorDataState] = useState<{
 		title: string;
 		list?: Array<string>;
@@ -73,68 +73,69 @@ export function AlertProvider({ children }:{children:ReactNode}) {
 			return newNotificationList;
 		});
 	};
-/**
+	/**
- * Sets the error data state, opens the error dialog and pushes the new error notification to the notification list
+	 * Sets the error data state, opens the error dialog and pushes the new error notification to the notification list
- * @param newState An object containing the new error data, including title and optional list of error messages
+	 * @param newState An object containing the new error data, including title and optional list of error messages
- */
+	 */
-function setErrorData(newState: { title: string; list?: Array<string> }) {
+	function setErrorData(newState: { title: string; list?: Array<string> }) {
-	setErrorDataState(newState);
+		setErrorDataState(newState);
-	setErrorOpen(true);
+		setErrorOpen(true);
-	if (newState.title) {
+		if (newState.title) {
-		setNotificationCenter(true);
+			setNotificationCenter(true);
-		pushNotificationList({
+			pushNotificationList({
-			type: "error",
+				type: "error",
-			title: newState.title,
+				title: newState.title,
-			list: newState.list,
+				list: newState.list,
-			id: _.uniqueId(),
+				id: _.uniqueId(),
-		});
+			});
+		}
 	}
-}
+	/**
-/**
+	 * Sets the state of the notice data and opens the notice modal, also adds a new notice to the notification center if the title is defined.
- * Sets the state of the notice data and opens the notice modal, also adds a new notice to the notification center if the title is defined.
+	 * @param newState An object containing the title of the notice and optionally a link.
- * @param newState An object containing the title of the notice and optionally a link.
+	 */
- */
+	function setNoticeData(newState: { title: string; link?: string }) {
-function setNoticeData(newState: { title: string; link?: string }) {
+		setNoticeDataState(newState);
-	setNoticeDataState(newState);
+		setNoticeOpen(true);
-	setNoticeOpen(true);
+		if (newState.title) {
-	if (newState.title) {
+			// Add new notice to notification center
-		// Add new notice to notification center
+			setNotificationCenter(true);
-		setNotificationCenter(true);
+			pushNotificationList({
-		pushNotificationList({
+				type: "notice",
-			type: "notice",
+				title: newState.title,
-			title: newState.title,
+				link: newState.link,
-			link: newState.link,
+				id: _.uniqueId(),
-			id: _.uniqueId(),
+			});
-		});
+		}
 	}
-}
+	/**
-/**
+	 * Update the success data state and show a success alert notification.
- * Update the success data state and show a success alert notification.
+	 * @param newState - A state object with a "title" property to set in the success data state.
- * @param newState - A state object with a "title" property to set in the success data state.
+	 */
- */
+	function setSuccessData(newState: { title: string }) {
-function setSuccessData(newState: { title: string }) {
+		setSuccessDataState(newState); // update the success data state with the provided new state
-	setSuccessDataState(newState); // update the success data state with the provided new state
+		setSuccessOpen(true); // open the success alert
-	setSuccessOpen(true); // open the success alert
 
-	// If the new state has a "title" property, add a new success notification to the list
+		// If the new state has a "title" property, add a new success notification to the list
-	if (newState.title) {
+		if (newState.title) {
-		setNotificationCenter(true); // show the notification center
+			setNotificationCenter(true); // show the notification center
-		pushNotificationList({ // add the new notification to the list
+			pushNotificationList({
-			type: "success",
+				// add the new notification to the list
-			title: newState.title,
+				type: "success",
-			id: _.uniqueId(),
+				title: newState.title,
-		});
+				id: _.uniqueId(),
+			});
+		}
 	}
-}
 	function clearNotificationList() {
 		setNotificationList([]);
 	}
 	function removeFromNotificationList(index: string) {
 		// set the notification list to a new array that filters out the alert with the matching id
 		setNotificationList((prevAlertsList) =>
-		  prevAlertsList.filter((alert) => alert.id !== index)
+			prevAlertsList.filter((alert) => alert.id !== index)
 		);
-	  }
+	}
 	return (
 		<alertContext.Provider
 			value={{

src/frontend/src/contexts/darkContext.tsx

@@ -14,13 +14,13 @@ export const darkContext = createContext<darkContextType>(initialValue);
 
 export function DarkProvider({ children }) {
 	const [dark, setDark] = useState(false);
-	useEffect(()=>{
+	useEffect(() => {
-		if(dark){
+		if (dark) {
 			document.getElementById("body").classList.add("dark");
 		} else {
 			document.getElementById("body").classList.remove("dark");
 		}
-	}, [dark])
+	}, [dark]);
 	return (
 		<darkContext.Provider
 			value={{
@@ -31,4 +31,4 @@ export function DarkProvider({ children }) {
 			{children}
 		</darkContext.Provider>
 	);
-}
+}

src/frontend/src/contexts/locationContext.tsx

@@ -32,7 +32,7 @@ type locationContextType = {
 
 //initial value for location context
 const initialValue = {
-    //actual
+	//actual
 	current: window.location.pathname.replace(/\/$/g, "").split("/"),
 	isStackedOpen:
 		window.innerWidth > 1024 && window.location.pathname.split("/")[1]
@@ -50,7 +50,7 @@ const initialValue = {
 
 export const locationContext = createContext<locationContextType>(initialValue);
 
-export function LocationProvider({ children }:{children:ReactNode}) {
+export function LocationProvider({ children }: { children: ReactNode }) {
 	const [current, setCurrent] = useState(initialValue.current);
 	const [isStackedOpen, setIsStackedOpen] = useState(
 		initialValue.isStackedOpen

src/frontend/src/contexts/popUpContext.tsx

@@ -3,31 +3,31 @@ import React, { useState } from "react";
 
 // context to set JSX element on the DOM
 export const PopUpContext = createContext({
-  openPopUp: (popUpElement: JSX.Element) => {},
+	openPopUp: (popUpElement: JSX.Element) => {},
-  closePopUp: () => {},
+	closePopUp: () => {},
 });
 
 interface PopUpProviderProps {
-  children: React.ReactNode;
+	children: React.ReactNode;
 }
 
 const PopUpProvider = ({ children }: PopUpProviderProps) => {
-  const [popUpElements, setPopUpElements] = useState<JSX.Element[]>([]);
+	const [popUpElements, setPopUpElements] = useState<JSX.Element[]>([]);
 
-  const openPopUp = (element: JSX.Element) => {
+	const openPopUp = (element: JSX.Element) => {
-    setPopUpElements(prevPopUps => [element, ...prevPopUps]);
+		setPopUpElements((prevPopUps) => [element, ...prevPopUps]);
-  };
+	};
 
-  const closePopUp = () => {
+	const closePopUp = () => {
-    setPopUpElements(prevPopUps => prevPopUps.slice(1));
+		setPopUpElements((prevPopUps) => prevPopUps.slice(1));
-  };
+	};
 
-  return (
+	return (
-    <PopUpContext.Provider value={{ openPopUp, closePopUp }}>
+		<PopUpContext.Provider value={{ openPopUp, closePopUp }}>
-      {children}
+			{children}
-      {popUpElements[0]}
+			{popUpElements[0]}
-    </PopUpContext.Provider>
+		</PopUpContext.Provider>
-  );
+	);
 };
 
 export default PopUpProvider;

src/frontend/src/contexts/tabsContext.tsx

@@ -99,7 +99,7 @@ export function TabsProvider({ children }: { children: ReactNode }) {
 		// create a link element and set its properties
 		const link = document.createElement("a");
 		link.href = jsonString;
-		link.download = `${normalCaseToSnakeCase(flows[tabIndex].name)}.json`;
+		link.download = `${flows[tabIndex].name}.json`;
 
 		// simulate a click on the link element to trigger the download
 		link.click();

src/frontend/src/contexts/typesContext.tsx

@@ -1,12 +1,12 @@
 import { createContext, ReactNode, useEffect, useState } from "react";
-import { Node} from "reactflow";
+import { Node } from "reactflow";
 import { typesContextType } from "../types/typesContext";
 import { getAll } from "../controllers/API";
 import { APIKindType } from "../types/api";
 
 //context to share types adn functions from nodes to flow
 
-const initialValue:typesContextType = {
+const initialValue: typesContextType = {
 	reactFlowInstance: null,
 	setReactFlowInstance: () => {},
 	deleteNode: () => {},
@@ -14,13 +14,13 @@ const initialValue:typesContextType = {
 	setTypes: () => {},
 	templates: {},
 	setTemplates: () => {},
-	data:{},
+	data: {},
-	setData:()=>{}
+	setData: () => {},
 };
 
 export const typesContext = createContext<typesContextType>(initialValue);
 
-export function TypesProvider({ children }:{children:ReactNode}) {
+export function TypesProvider({ children }: { children: ReactNode }) {
 	const [types, setTypes] = useState({});
 	const [reactFlowInstance, setReactFlowInstance] = useState(null);
 	const [templates, setTemplates] = useState({});
@@ -35,11 +35,11 @@ export function TypesProvider({ children }:{children:ReactNode}) {
 			setData(result.data);
 			setTemplates(
 				Object.keys(result.data).reduce((acc, curr) => {
-					Object.keys(result.data[curr]).forEach((c: keyof APIKindType)=>{
+					Object.keys(result.data[curr]).forEach((c: keyof APIKindType) => {
-						acc[c] = result.data[curr][c]
+						acc[c] = result.data[curr][c];
-					})
+					});
 					return acc;
-				},{})
+				}, {})
 			);
 			// Set the types by reducing over the keys of the result data and updating the accumulator.
 			setTypes(
@@ -59,11 +59,15 @@ export function TypesProvider({ children }:{children:ReactNode}) {
 		getTypes();
 	}, [setTypes]);
 
-	function deleteNode(idx:string) {
+	function deleteNode(idx: string) {
 		reactFlowInstance.setNodes(
-			reactFlowInstance.getNodes().filter((n:Node) => n.id !== idx)
+			reactFlowInstance.getNodes().filter((n: Node) => n.id !== idx)
+		);
+		reactFlowInstance.setEdges(
+			reactFlowInstance
+				.getEdges()
+				.filter((ns) => ns.source !== idx && ns.target !== idx)
 		);
-		reactFlowInstance.setEdges(reactFlowInstance.getEdges().filter((ns) => ns.source !== idx && ns.target !== idx));
 	}
 	return (
 		<typesContext.Provider
@@ -76,7 +80,7 @@ export function TypesProvider({ children }:{children:ReactNode}) {
 				setTemplates,
 				templates,
 				data,
-				setData
+				setData,
 			}}
 		>
 			{children}

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

@@ -4,40 +4,40 @@ import axios, { AxiosResponse } from "axios";
 import { FlowType } from "../../types/flow";
 
 export async function getAll(): Promise<AxiosResponse<APIObjectType>> {
-  return await axios.get(`/all`);
+	return await axios.get(`/all`);
 }
 
 export async function sendAll(data: sendAllProps) {
-  return await axios.post(`/predict`, data);
+	return await axios.post(`/predict`, data);
 }
 
 export async function checkCode(
-  code: string
+	code: string
 ): Promise<AxiosResponse<errorsTypeAPI>> {
-  return await axios.post("/validate/code", { code });
+	return await axios.post("/validate/code", { code });
 }
 
 export async function checkPrompt(
-  template: string
+	template: string
 ): Promise<AxiosResponse<PromptTypeAPI>> {
-  return await axios.post("/validate/prompt", { template });
+	return await axios.post("/validate/prompt", { template });
 }
 
 export async function getExamples(): Promise<FlowType[]> {
-  const url =
+	const url =
-    "https://api.github.com/repos/logspace-ai/langflow_examples/contents/examples";
+		"https://api.github.com/repos/logspace-ai/langflow_examples/contents/examples";
-  const response = await axios.get(url);
+	const response = await axios.get(url);
 
-  const jsonFiles = response.data.filter((file: any) => {
+	const jsonFiles = response.data.filter((file: any) => {
-    return file.name.endsWith(".json");
+		return file.name.endsWith(".json");
-  });
+	});
 
-  const contentsPromises = jsonFiles.map(async (file: any) => {
+	const contentsPromises = jsonFiles.map(async (file: any) => {
-    const contentResponse = await axios.get(file.download_url);
+		const contentResponse = await axios.get(file.download_url);
-    return contentResponse.data;
+		return contentResponse.data;
-  });
+	});
 
-  const contents = await Promise.all(contentsPromises);
+	const contents = await Promise.all(contentsPromises);
 
-  return contents;
+	return contents;
 }

src/frontend/src/index.css

@@ -1,15 +1,13 @@
-
-
 body {
-  margin: 0;
+	margin: 0;
-  font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen',
+	font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "Roboto", "Oxygen",
-    'Ubuntu', 'Cantarell', 'Fira Sans', 'Droid Sans', 'Helvetica Neue',
+		"Ubuntu", "Cantarell", "Fira Sans", "Droid Sans", "Helvetica Neue",
-    sans-serif;
+		sans-serif;
-  -webkit-font-smoothing: antialiased;
+	-webkit-font-smoothing: antialiased;
-  -moz-osx-font-smoothing: grayscale;
+	-moz-osx-font-smoothing: grayscale;
 }
 
 code {
-  font-family: source-code-pro, Menlo, Monaco, Consolas, 'Courier New',
+	font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
-    monospace;
+		monospace;
 }

src/frontend/src/index.tsx

@@ -11,8 +11,7 @@ const root = ReactDOM.createRoot(
 root.render(
 	<ContextWrapper>
 		<BrowserRouter>
-
+			<App />
-				<App />
 		</BrowserRouter>
 	</ContextWrapper>
 );

src/frontend/src/modals/NodeModal/index.tsx

@@ -87,37 +87,38 @@ export default function NodeModal({ data }: { data: NodeDataType }) {
 										<div className="flex h-full w-full">
 											<div className="overflow-hidden px-4 sm:p-4  w-full rounded-lg bg-white dark:bg-gray-800 shadow">
 												<div className="flex flex-col h-full gap-5">
-													{
+													{Object.keys(data.node.template)
-														Object.keys(data.node.template)
+														.filter(
-															.filter((t) => t.charAt(0) !== "_"&& data.node.template[t].advanced && data.node.template[t].show)
+															(t) =>
-															.map((t: string, idx) => {
+																t.charAt(0) !== "_" &&
-																return (
+																data.node.template[t].advanced &&
-																	<ModalField
+																data.node.template[t].show
+														)
+														.map((t: string, idx) => {
+															return (
+																<ModalField
 																	key={idx}
-																		data={data}
+																	data={data}
-																		title={
+																	title={
-																			data.node.template[t].display_name
+																		data.node.template[t].display_name
-																				? data.node.template[t].display_name
+																			? data.node.template[t].display_name
-																				: data.node.template[t].name
+																			: data.node.template[t].name
-																				? toNormalCase(
+																			? toNormalCase(data.node.template[t].name)
-																						data.node.template[t].name
+																			: toNormalCase(t)
-																				  )
+																	}
-																				: toNormalCase(t)
+																	required={data.node.template[t].required}
-																		}
+																	id={
-																		required={data.node.template[t].required}
+																		data.node.template[t].type +
-																		id={
+																		"|" +
-																			data.node.template[t].type +
+																		t +
-																			"|" +
+																		"|" +
-																			t +
+																		data.id
-																			"|" +
+																	}
-																			data.id
+																	name={t}
-																		}
+																	type={data.node.template[t].type}
-																		name={t}
+																/>
-																		type={data.node.template[t].type}
+															);
-																	/>
+														})}
-																);
-															})
-													}
 												</div>
 											</div>
 										</div>

src/frontend/src/modals/chatModal/chatInput/index.tsx

@@ -24,9 +24,9 @@ export default function ChatInput({
 				}}
 				ref={inputRef}
 				disabled={lockChat}
-				style={{resize: "none" }}
+				style={{ resize: "none" }}
 				value={lockChat ? "Thinking..." : chatValue}
-				onChange={(e) => {							
+				onChange={(e) => {
 					setChatValue(e.target.value);
 				}}
 				className={classNames(

src/frontend/src/modals/chatModal/chatMessage/codeBlock/index.tsx

@@ -0,0 +1,81 @@
+import { IconCheck, IconClipboard, IconDownload } from "@tabler/icons-react";
+import { FC, memo, useState } from "react";
+import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
+import { oneDark } from "react-syntax-highlighter/dist/cjs/styles/prism";
+import { programmingLanguages } from "../../../../utils";
+
+interface Props {
+	language: string;
+	value: string;
+}
+
+export const CodeBlock: FC<Props> = memo(({ language, value }) => {
+	const [isCopied, setIsCopied] = useState<Boolean>(false);
+
+	const copyToClipboard = () => {
+		if (!navigator.clipboard || !navigator.clipboard.writeText) {
+			return;
+		}
+
+		navigator.clipboard.writeText(value).then(() => {
+			setIsCopied(true);
+
+			setTimeout(() => {
+				setIsCopied(false);
+			}, 2000);
+		});
+	};
+	const downloadAsFile = () => {
+		const fileExtension = programmingLanguages[language] || ".file";
+		const suggestedFileName = `${"generated-code"}${fileExtension}`;
+		const fileName = window.prompt("enter file name", suggestedFileName);
+
+		if (!fileName) {
+			// user pressed cancel on prompt
+			return;
+		}
+
+		const blob = new Blob([value], { type: "text/plain" });
+		const url = URL.createObjectURL(blob);
+		const link = document.createElement("a");
+		link.download = fileName;
+		link.href = url;
+		link.style.display = "none";
+		document.body.appendChild(link);
+		link.click();
+		document.body.removeChild(link);
+		URL.revokeObjectURL(url);
+	};
+	return (
+		<div className="codeblock relative font-sans text-[16px]">
+			<div className="flex items-center justify-between py-1.5 px-4">
+				<span className="text-xs lowercase text-white">{language}</span>
+
+				<div className="flex items-center">
+					<button
+						className="flex gap-1.5 items-center rounded bg-none p-1 text-xs text-white"
+						onClick={copyToClipboard}
+					>
+						{isCopied ? <IconCheck size={18} /> : <IconClipboard size={18} />}
+						{isCopied ? "Copied!" : "Copy code"}
+					</button>
+					<button
+						className="flex items-center rounded bg-none p-1 text-xs text-white"
+						onClick={downloadAsFile}
+					>
+						<IconDownload size={18} />
+					</button>
+				</div>
+			</div>
+
+			<SyntaxHighlighter
+				language={language}
+				style={oneDark}
+				customStyle={{ margin: 0 }}
+			>
+				{value}
+			</SyntaxHighlighter>
+		</div>
+	);
+});
+CodeBlock.displayName = "CodeBlock";

src/frontend/src/modals/chatModal/chatMessage/index.tsx

@@ -1,82 +1,140 @@
 import { ChatBubbleOvalLeftEllipsisIcon } from "@heroicons/react/24/outline";
-import { useState } from "react";
+import { useEffect, useState } from "react";
 import { ChatMessageType } from "../../../types/chat";
 import { classNames } from "../../../utils";
 import AiIcon from "../../../assets/Gooey Ring-5s-271px.svg";
 import { UserIcon } from "@heroicons/react/24/solid";
 import FileCard from "../fileComponent";
+import ReactMarkdown from "react-markdown";
+import rehypeMathjax from "rehype-mathjax";
+import remarkGfm from "remark-gfm";
+import remarkMath from "remark-math";
+import { CodeBlock } from "./codeBlock";
 var Convert = require("ansi-to-html");
 var convert = new Convert({ newline: true });
 
 export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
-  const [hidden, setHidden] = useState(true);
+	const [message, setMessage] = useState("");
-  return (
+	useEffect(() => {
-    <div
+		setMessage(chat.message);
-      className={classNames(
+	}, [chat.message]);
-        "w-full py-2 pl-2 flex",
+	const [hidden, setHidden] = useState(true);
-        chat.isSend ? "bg-white dark:bg-gray-800 " : "bg-gray-200  dark:bg-gray-700"
+	return (
-      )}
+		<div
-    >
+			className={classNames(
-      <div
+				"w-full py-2 pl-2 flex",
-        className={classNames(
+				chat.isSend
-          "rounded-full w-8 h-8 flex items-center my-3 justify-center",
+					? "bg-white dark:bg-gray-800 "
-          chat.isSend ? "bg-gray-900" : "bg-gray-200"
+					: "bg-gray-200  dark:bg-gray-700"
-        )}
+			)}
-      >
+		>
-        {!chat.isSend && <img className="scale-150" src={AiIcon} />}
+			<div
-        {chat.isSend && <UserIcon className="w-6 h-6 -mb-1 text-gray-200" />}
+				className={classNames(
-      </div>
+					"rounded-full w-8 h-8 flex items-center my-3 justify-center",
-      {!chat.isSend ? (
+					chat.isSend ? "bg-gray-900" : "bg-gray-200"
-        <div className="w-full text-start flex items-center">
+				)}
-          <div className="w-full relative text-start inline-block text-gray-600 text-sm font-normal">
+			>
-            {hidden && chat.thought && chat.thought !== "" && (
+				{!chat.isSend && <img className="scale-150" src={AiIcon} />}
-              <div
+				{chat.isSend && <UserIcon className="w-6 h-6 -mb-1 text-gray-200" />}
-                onClick={() => setHidden((prev) => !prev)}
+			</div>
-                className="absolute -top-1 -left-2 cursor-pointer"
+			{!chat.isSend ? (
-              >
+				<div className="w-full text-start flex items-center">
-                <ChatBubbleOvalLeftEllipsisIcon className="w-5 h-5 animate-bounce dark:text-white" />
+					<div className="w-full relative text-start inline-block text-gray-600 text-sm font-normal">
-              </div>
+						{hidden && chat.thought && chat.thought !== "" && (
-            )}
+							<div
-            {chat.thought && chat.thought !== "" && !hidden && (
+								onClick={() => setHidden((prev) => !prev)}
-              <div
+								className="absolute -top-1 -left-2 cursor-pointer"
-                onClick={() => setHidden((prev) => !prev)}
+							>
-                className=" text-start inline-block rounded-md  h-full border border-gray-300
+								<ChatBubbleOvalLeftEllipsisIcon className="w-5 h-5 animate-bounce dark:text-white" />
+							</div>
+						)}
+						{chat.thought && chat.thought !== "" && !hidden && (
+							<div
+								onClick={() => setHidden((prev) => !prev)}
+								className=" text-start inline-block rounded-md  h-full border border-gray-300
 								bg-gray-100 w-[95%] pb-3 pt-3 px-2 ml-3 cursor-pointer scrollbar-hide overflow-scroll"
-                dangerouslySetInnerHTML={{
+								dangerouslySetInnerHTML={{
-                  __html: convert.toHtml(chat.thought),
+									__html: convert.toHtml(chat.thought),
-                }}
+								}}
-              ></div>
+							></div>
-            )}
+						)}
-            {chat.thought && chat.thought !== "" && !hidden && <br></br>}
+						{chat.thought && chat.thought !== "" && !hidden && <br></br>}
-            <div className="w-full px-4 pb-3 pt-3 pr-8">
+						<div className="w-full px-4 pb-3 pt-3 pr-8">
-              <span className="dark:text-white">
+							<div className="dark:text-white">
-                {chat.message}
+								<div>
-                {chat.files && (
+									<ReactMarkdown
-                  <div className="my-2 w-full">
+										remarkPlugins={[remarkGfm, remarkMath]}
-                    {chat.files.map((file, index) => {
+										rehypePlugins={[rehypeMathjax]}
-                      return (
+										className="markdown prose"
-                        <div key={index} className="my-2 w-full">
+										components={{
-                          <FileCard
+											code({ node, inline, className, children, ...props }) {
-                            fileName={"Generated File"}
+												if (children.length) {
-                            fileType={file.data_type}
+													if (children[0] == "▍") {
-                            content={file.data}
+														return (
-                          />
+															<span className="animate-pulse cursor-default mt-1">
-                        </div>
+																▍
-                      );
+															</span>
-                    })}
+														);
-                  </div>
+													}
-                )}
+
-              </span>
+													children[0] = (children[0] as string).replace(
-            </div>
+														"`▍`",
-          </div>
+														"▍"
-        </div>
+													);
-      ) : (
+												}
-        <div className="w-full flex items-center">
+
-          <div className="text-start inline-block px-3 text-sm text-gray-600 dark:text-white">
+												const match = /language-(\w+)/.exec(className || "");
-            {chat.message}
+
-          </div>
+												return !inline ? (
-        </div>
+													<CodeBlock
-      )}
+														key={Math.random()}
-    </div>
+														language={(match && match[1]) || ""}
-  );
+														value={String(children).replace(/\n$/, "")}
+														{...props}
+													/>
+												) : (
+													<code className={className} {...props}>
+														{children}
+													</code>
+												);
+											},
+										}}
+									>
+										{message}
+									</ReactMarkdown>
+								</div>
+								{chat.files && (
+									<div className="my-2 w-full">
+										{chat.files.map((file, index) => {
+											return (
+												<div key={index} className="my-2 w-full">
+													<FileCard
+														fileName={"Generated File"}
+														fileType={file.data_type}
+														content={file.data}
+													/>
+												</div>
+											);
+										})}
+									</div>
+								)}
+							</div>
+						</div>
+					</div>
+				</div>
+			) : (
+				<div className="w-full flex items-center">
+					<div className="text-start inline-block px-3 text-sm text-gray-600 dark:text-white">
+						<ReactMarkdown
+							remarkPlugins={[remarkGfm, remarkMath]}
+							rehypePlugins={[rehypeMathjax]}
+							className="markdown prose"
+						>
+							{message}
+						</ReactMarkdown>
+					</div>
+				</div>
+			)}
+		</div>
+	);
 }

src/frontend/src/modals/chatModal/fileComponent/index.tsx

@@ -1,19 +1,55 @@
-import {  CloudArrowDownIcon, DocumentIcon } from "@heroicons/react/24/outline";
+import { CloudArrowDownIcon, DocumentIcon } from "@heroicons/react/24/outline";
-import * as base64js from 'base64-js';
+import * as base64js from "base64-js";
+import { useState } from "react";
 
 export default function FileCard({ fileName, content, fileType }) {
-    const handleDownload = () => {
+	const handleDownload = () => {
-        const byteArray = new Uint8Array(base64js.toByteArray(content));
+		const byteArray = new Uint8Array(base64js.toByteArray(content));
-        const blob = new Blob([byteArray], { type: 'application/octet-stream' });
+		const blob = new Blob([byteArray], { type: "application/octet-stream" });
-        const url = URL.createObjectURL(blob);
+		const url = URL.createObjectURL(blob);
-        const link = document.createElement('a');
+		const link = document.createElement("a");
-        link.href = url;
+		link.href = url;
-        link.download = fileName+".png";
+		link.download = fileName + ".png";
-        document.body.appendChild(link);
+		document.body.appendChild(link);
-        link.click();
+		link.click();
-        document.body.removeChild(link);
+		document.body.removeChild(link);
-        URL.revokeObjectURL(url);
+		URL.revokeObjectURL(url);
-      };
+	};
+	const [isHovered, setIsHovered] = useState(false);
+	function handleMouseEnter() {
+		setIsHovered(true);
+	}
+	function handleMouseLeave() {
+		setIsHovered(false);
+	}
+
+	if (fileType === "image") {
+		return (
+			<div
+				className="relative w-1/4 h-1/4"
+				onMouseEnter={handleMouseEnter}
+				onMouseLeave={handleMouseLeave}
+			>
+				<img
+					src={`data:image/png;base64,${content}`}
+					alt="generated image"
+					className="rounded-lg  w-full h-full"
+				/>
+				{isHovered && (
+					<div
+						className={`absolute top-0 right-0 bg-gray-100 text-gray-700 rounded-bl-lg px-1 text-sm font-bold dark:bg-gray-700 dark:text-gray-300`}
+					>
+						<button
+							className="text-gray-500 py-1 px-2 dark:bg-gray-700 dark:text-gray-300"
+							onClick={handleDownload}
+						>
+							<CloudArrowDownIcon className="hover:scale-110 w-5 h-5 text-current"></CloudArrowDownIcon>
+						</button>
+					</div>
+				)}
+			</div>
+		);
+	}
 
 	return (
 		<button
@@ -22,12 +58,19 @@ export default function FileCard({ fileName, content, fileType }) {
 		>
 			<div className="flex gap-2 text-current items-center w-full mr-2">
 				{" "}
-				<DocumentIcon className="w-8 h-8" />
+				{fileType === "image" ? (
+					<img
+						src={`data:image/png;base64,${content}`}
+						alt=""
+						className="w-8 h-8"
+					/>
+				) : (
+					<DocumentIcon className="w-8 h-8" />
+				)}
 				<div className="flex flex-col items-start">
 					{" "}
 					<div className="truncate text-sm text-current">{fileName}</div>
-                    <div className="truncate text-xs  text-gray-500">{fileType}</div>
+					<div className="truncate text-xs  text-gray-500">{fileType}</div>
-
 				</div>
 				<CloudArrowDownIcon className="w-6 h-6 text-current ml-auto" />
 			</div>

src/frontend/src/modals/chatModal/index.tsx

@@ -1,13 +1,7 @@
 import { Dialog, Transition } from "@headlessui/react";
-import {
+import { ChatBubbleOvalLeftEllipsisIcon } from "@heroicons/react/24/outline";
-  ChatBubbleOvalLeftEllipsisIcon,
-  LockClosedIcon,
-  PaperAirplaneIcon,
-} from "@heroicons/react/24/outline";
 import { Fragment, useContext, useEffect, useRef, useState } from "react";
-import { PopUpContext } from "../../contexts/popUpContext";
 import { FlowType, NodeType } from "../../types/flow";
-import { TabsContext } from "../../contexts/tabsContext";
 import { alertContext } from "../../contexts/alertContext";
 import { toNormalCase } from "../../utils";
 import { typesContext } from "../../contexts/typesContext";
@@ -32,8 +26,15 @@ export default function ChatModal({
   const [chatHistory, setChatHistory] = useState<ChatMessageType[]>([]);
   const { reactFlowInstance } = useContext(typesContext);
   const { setErrorData, setNoticeData } = useContext(alertContext);
-  const [ws, setWs] = useState<WebSocket | null>(null);
+  const ws = useRef<WebSocket | null>(null);
   const [lockChat, setLockChat] = useState(false);
+  const isOpen = useRef(open);
+
+  useEffect(() => {
+    isOpen.current = open;
+  }, [open]);
+  var isStream = false;
+
   const addChatHistory = (
     message: string,
     isSend: boolean,
@@ -53,131 +54,172 @@ export default function ChatModal({
     });
   };
 
+  //add proper type signature for function
+
+  function updateLastMessage({
+    str,
+    thought,
+    end = false,
+    files,
+  }: {
+    str?: string;
+    thought?: string;
+    // end param default is false
+    end?: boolean;
+    files?: Array<any>;
+  }) {
+    setChatHistory((old) => {
+      let newChat = [...old];
+      if (str) {
+        if (end) {
+          newChat[newChat.length - 1].message = str;
+        } else {
+          newChat[newChat.length - 1].message =
+            newChat[newChat.length - 1].message + str;
+        }
+      }
+      if (thought) {
+        newChat[newChat.length - 1].thought = thought;
+      }
+      if (files) {
+        newChat[newChat.length - 1].files = files;
+      }
+      return newChat;
+    });
+  }
+
+  function handleOnClose(event: CloseEvent) {
+    if (isOpen.current) {
+      setErrorData({ title: event.reason });
+      setLockChat(false);
+      setTimeout(() => {
+        connectWS();
+      }, 1000);
+    }
+  }
+
+  function handleWsMessage(data: any) {
+    if (Array.isArray(data)) {
+      //set chat history
+      setChatHistory((_) => {
+        let newChatHistory: ChatMessageType[] = [];
+        data.forEach(
+          (chatItem: {
+            intermediate_steps?: "string";
+            is_bot: boolean;
+            message: string;
+            type: string;
+            files?: Array<any>;
+          }) => {
+            if (chatItem.message) {
+              newChatHistory.push(
+                chatItem.files
+                  ? {
+                    isSend: !chatItem.is_bot,
+                    message: chatItem.message,
+                    thought: chatItem.intermediate_steps,
+                    files: chatItem.files,
+                  }
+                  : {
+                    isSend: !chatItem.is_bot,
+                    message: chatItem.message,
+                    thought: chatItem.intermediate_steps,
+                  }
+              );
+            }
+          }
+        );
+        return newChatHistory;
+      });
+    }
+    if (data.type === "start") {
+      addChatHistory("", false);
+      isStream = true;
+    }
+    if (data.type === "end") {
+      if (data.intermediate_steps) {
+        updateLastMessage({
+          str: data.message,
+          thought: data.intermediate_steps,
+          end: true,
+        });
+      }
+      if (data.files) {
+        updateLastMessage({
+          end: true,
+          files: data.files,
+        });
+      }
+
+      setLockChat(false);
+      isStream = false;
+    }
+    if (data.type === "stream" && isStream) {
+      updateLastMessage({ str: data.message });
+    }
+  }
+
   function connectWS() {
-    console.log("conectou");
     try {
       const urlWs =
         process.env.NODE_ENV === "development"
           ? `ws://localhost:7860/chat/${flow.id}`
-          : `${window.location.protocol === "https:" ? "wss" : "ws"}://${
+          : `${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host
-              window.location.host
+          }/chat/${flow.id}`;
-            }/chat/${flow.id}`;
 
       const newWs = new WebSocket(urlWs);
       newWs.onopen = () => {
         console.log("WebSocket connection established!");
       };
+      console.log(flow.id);
       newWs.onmessage = (event) => {
-        try {
+        const data = JSON.parse(event.data);
-          setLockChat(false);
+        console.log("Received data:", data);
-          const data = JSON.parse(event.data);
+        handleWsMessage(data);
-          console.log("Received data:", data);
+        //get chat history
-          //get chat history
-          if (Array.isArray(data)) {
-            console.log(data);
-
-            setChatHistory((_) => {
-              let newChatHistory: ChatMessageType[] = [];
-              data.forEach(
-                (chatItem: {
-                  intermediate_steps?: "string";
-                  is_bot: boolean;
-                  message: string;
-                  type: string;
-                  files?: Array<any>;
-                }) => {
-                  if (chatItem.message) {
-                    newChatHistory.push(
-                      chatItem.files
-                        ? {
-                            isSend: !chatItem.is_bot,
-                            message: chatItem.message,
-                            thought: chatItem.intermediate_steps,
-                            files: chatItem.files,
-                          }
-                        : {
-                            isSend: !chatItem.is_bot,
-                            message: chatItem.message,
-                            thought: chatItem.intermediate_steps,
-                          }
-                    );
-                  }
-                }
-              );
-              return newChatHistory;
-            });
-          }
-          if (data.type === "end") {
-            if (data.files) {
-              addChatHistory(
-                data.message,
-                false,
-                data.intermediate_steps,
-                data.files
-              );
-            } else {
-              addChatHistory(data.message, false, data.intermediate_steps);
-            }
-          }
-          if (data.type == "file") {
-            console.log(data);
-          }
-        } catch (error) {
-          if (event.data !== "Error: 1005") {
-            setErrorData({ title: event.data });
-            newWs.close();
-            connectWS();
-          }
-        }
       };
-      newWs.onclose = (_) => {
+      newWs.onclose = (event) => {
-        if (open) {
+        handleOnClose(event);
-          setLockChat(false);
-          setTimeout(() => {
-            connectWS();
-          }, 1000);
-        }
       };
       newWs.onerror = (ev) => {
         console.log(ev, "error");
+        setErrorData({
+          title: "There was an error on web connection, please: ",
+          list: [
+            "Refresh the page",
+            "Use a new flow tab",
+            "Check if the backend is up",
+          ],
+        });
       };
-      setWs(newWs);
+      ws.current = newWs;
-
-      return newWs;
     } catch {
       setErrorData({
         title: "There was an error on web connection, please: ",
         list: [
-          "refresh the page",
+          "Refresh the page",
-          "use a new flow tab",
+          "Use a new flow tab",
-          "check if the backend is up",
+          "Check if the backend is up",
         ],
       });
     }
   }
 
   useEffect(() => {
-    if (ws && (ws.readyState === ws.CLOSED || ws.readyState === ws.CLOSING)) {
+    connectWS();
-      let newWs = connectWS();
-      return () => {
-        console.log("trigger");
-        newWs.close();
-      };
-    }
-  }, [lockChat]);
-
-  useEffect(() => {
-    let newWs = connectWS();
     return () => {
-      console.log("trigger");
+      console.log("unmount");
-      newWs.close();
+      console.log(ws);
+      if (ws) {
+        ws.current.close();
+      }
     };
   }, []);
 
   async function sendAll(data: sendAllProps) {
     try {
       if (ws) {
-        ws.send(JSON.stringify(data));
+        ws.current.send(JSON.stringify(data));
       }
     } catch (error) {
       setErrorData({
@@ -193,12 +235,6 @@ export default function ChatModal({
     if (ref.current) ref.current.scrollIntoView({ behavior: "smooth" });
   }, [chatHistory]);
 
-  useEffect(() => {
-    if (ws && ws.readyState === ws.CLOSED) {
-      setLockChat(false);
-    }
-  }, [lockChat]);
-
   function validateNode(n: NodeType): Array<string> {
     if (!n.data?.node?.template || !Object.keys(n.data.node.template)) {
       setNoticeData({
@@ -227,12 +263,11 @@ export default function ChatModal({
                   e.targetHandle.split("|")[2] === n.id
               )
             ? [
-                `${type} is missing ${
+              `${type} is missing ${template.display_name
-                  template.display_name
+                ? template.display_name
-                    ? template.display_name
+                : toNormalCase(template[t].name)
-                    : toNormalCase(template[t].name)
+              }.`,
-                }.`,
+            ]
-              ]
             : []
         ),
       [] as string[]
@@ -255,7 +290,6 @@ export default function ChatModal({
         let message = chatValue;
         setChatValue("");
         addChatHistory(message, true);
-
         sendAll({
           ...reactFlowInstance.toObject(),
           message,
@@ -278,20 +312,14 @@ export default function ChatModal({
   }
   function clearChat() {
     setChatHistory([]);
-    ws.send(JSON.stringify({ clear_history: true }));
+    ws.current.send(JSON.stringify({ clear_history: true }));
   }
 
-  const { closePopUp } = useContext(PopUpContext);
   function setModalOpen(x: boolean) {
     setOpen(x);
-    if (x === false) {
-      setTimeout(() => {
-        closePopUp();
-      }, 300);
-    }
   }
   return (
-    <Transition.Root show={open} appear={true} as={Fragment}>
+    <Transition.Root show={open} appear={open} as={Fragment}>
       <Dialog
         as="div"
         className="relative z-10"
@@ -321,11 +349,11 @@ export default function ChatModal({
               leaveFrom="opacity-100 translate-y-0 sm:scale-100"
               leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
             >
-              <Dialog.Panel className=" drop-shadow-2xl relative flex flex-col justify-between transform h-[95%] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[690px]">
+              <Dialog.Panel className=" drop-shadow-2xl relative flex flex-col justify-between transform h-[95%] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all w-[690px]">
-                <div className="relative w-full">
+                <div className="relative w-full p-4">
                   <button
                     onClick={() => clearChat()}
-                    className="absolute top-2 right-2 hover:text-red-500 dark:text-gray-300 dark:hover:text-red-500"
+                    className="absolute top-2 right-3 hover:text-red-500 dark:text-gray-300 dark:hover:text-red-500 z-30"
                   >
                     <FaEraser className="w-4 h-4" />
                   </button>
@@ -334,7 +362,7 @@ export default function ChatModal({
                   {chatHistory.length > 0 ? (
                     chatHistory.map((c, i) => <ChatMessage chat={c} key={i} />)
                   ) : (
-                    <div className="flex flex-col h-full text-center justify-center w-full items-center align-middle ">
+                    <div className="flex flex-col h-full text-center justify-center w-full items-center align-middle">
                       <span>
                         👋{" "}
                         <span className="text-gray-600 dark:text-gray-300 text-lg">

src/frontend/src/modals/codeAreaModal/index.tsx

@@ -134,7 +134,7 @@ export default function CodeAreaModal({
 																	title: "Code is ready to run",
 																});
 																setModalOpen(false);
-																setValue(code)
+																setValue(code);
 															} else {
 																if (funcErrors.length !== 0) {
 																	setErrorData({
@@ -142,7 +142,7 @@ export default function CodeAreaModal({
 																		list: funcErrors,
 																	});
 																}
-																if(importsErrors.length!==0){
+																if (importsErrors.length !== 0) {
 																	setErrorData({
 																		title: "There is an error in your imports",
 																		list: importsErrors,

src/frontend/src/modals/exportModal/index.tsx

@@ -1,9 +1,9 @@
 import { Dialog, Transition } from "@headlessui/react";
 import {
-  XMarkIcon,
+	XMarkIcon,
-  ArrowDownTrayIcon,
+	ArrowDownTrayIcon,
-  DocumentDuplicateIcon,
+	DocumentDuplicateIcon,
-  ComputerDesktopIcon,
+	ComputerDesktopIcon,
 } from "@heroicons/react/24/outline";
 import { Fragment, useContext, useRef, useState } from "react";
 import { alertContext } from "../../contexts/alertContext";
@@ -12,169 +12,169 @@ import { TabsContext } from "../../contexts/tabsContext";
 import { removeApiKeys } from "../../utils";
 
 export default function ExportModal() {
-  const [open, setOpen] = useState(true);
+	const [open, setOpen] = useState(true);
-  const { closePopUp } = useContext(PopUpContext);
+	const { closePopUp } = useContext(PopUpContext);
-  const ref = useRef();
+	const ref = useRef();
-  const { setErrorData } = useContext(alertContext);
+	const { setErrorData } = useContext(alertContext);
-  const { flows, tabIndex, updateFlow, downloadFlow } = useContext(TabsContext);
+	const { flows, tabIndex, updateFlow, downloadFlow } = useContext(TabsContext);
-  function setModalOpen(x: boolean) {
+	function setModalOpen(x: boolean) {
-    setOpen(x);
+		setOpen(x);
-    if (x === false) {
+		if (x === false) {
-      setTimeout(() => {
+			setTimeout(() => {
-        closePopUp();
+				closePopUp();
-      }, 300);
+			}, 300);
-    }
+		}
-  }
+	}
-  const [checked, setChecked] = useState(true);
+	const [checked, setChecked] = useState(true);
-  const [name, setName] = useState(flows[tabIndex].name);
+	const [name, setName] = useState(flows[tabIndex].name);
-  return (
+	return (
-    <Transition.Root show={open} appear={true} as={Fragment}>
+		<Transition.Root show={open} appear={true} as={Fragment}>
-      <Dialog
+			<Dialog
-        as="div"
+				as="div"
-        className="relative z-10"
+				className="relative z-10"
-        onClose={setModalOpen}
+				onClose={setModalOpen}
-        initialFocus={ref}
+				initialFocus={ref}
-      >
+			>
-        <Transition.Child
+				<Transition.Child
-          as={Fragment}
+					as={Fragment}
-          enter="ease-out duration-300"
+					enter="ease-out duration-300"
-          enterFrom="opacity-0"
+					enterFrom="opacity-0"
-          enterTo="opacity-100"
+					enterTo="opacity-100"
-          leave="ease-in duration-200"
+					leave="ease-in duration-200"
-          leaveFrom="opacity-100"
+					leaveFrom="opacity-100"
-          leaveTo="opacity-0"
+					leaveTo="opacity-0"
-        >
+				>
-          <div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
+					<div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
-        </Transition.Child>
+				</Transition.Child>
 
-        <div className="fixed inset-0 z-10 overflow-y-auto">
+				<div className="fixed inset-0 z-10 overflow-y-auto">
-          <div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
+					<div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
-            <Transition.Child
+						<Transition.Child
-              as={Fragment}
+							as={Fragment}
-              enter="ease-out duration-300"
+							enter="ease-out duration-300"
-              enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-              enterTo="opacity-100 translate-y-0 sm:scale-100"
+							enterTo="opacity-100 translate-y-0 sm:scale-100"
-              leave="ease-in duration-200"
+							leave="ease-in duration-200"
-              leaveFrom="opacity-100 translate-y-0 sm:scale-100"
+							leaveFrom="opacity-100 translate-y-0 sm:scale-100"
-              leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-            >
+						>
-              <Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
+							<Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
-                <div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
+								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
-                  <button
+									<button
-                    type="button"
+										type="button"
-                    className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
+										className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
-                    onClick={() => {
+										onClick={() => {
-                      setModalOpen(false);
+											setModalOpen(false);
-                    }}
+										}}
-                  >
+									>
-                    <span className="sr-only">Close</span>
+										<span className="sr-only">Close</span>
-                    <XMarkIcon className="h-6 w-6" aria-hidden="true" />
+										<XMarkIcon className="h-6 w-6" aria-hidden="true" />
-                  </button>
+									</button>
-                </div>
+								</div>
-                <div className="h-full w-full flex flex-col justify-center items-center">
+								<div className="h-full w-full flex flex-col justify-center items-center">
-                  <div className="flex w-full pb-4 z-10 justify-center shadow-sm">
+									<div className="flex w-full pb-4 z-10 justify-center shadow-sm">
-                    <div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
+										<div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
-                      <ArrowDownTrayIcon
+											<ArrowDownTrayIcon
-                        className="h-6 w-6 text-blue-600"
+												className="h-6 w-6 text-blue-600"
-                        aria-hidden="true"
+												aria-hidden="true"
-                      />
+											/>
-                    </div>
+										</div>
-                    <div className="mt-4 text-center sm:ml-4 sm:text-left">
+										<div className="mt-4 text-center sm:ml-4 sm:text-left">
-                      <Dialog.Title
+											<Dialog.Title
-                        as="h3"
+												as="h3"
-                        className="text-lg font-medium dark:text-white leading-10 text-gray-900"
+												className="text-lg font-medium dark:text-white leading-10 text-gray-900"
-                      >
+											>
-                        Export as
+												Export as
-                      </Dialog.Title>
+											</Dialog.Title>
-                    </div>
+										</div>
-                  </div>
+									</div>
-                  <div className="pt-16 flex flex-col items-start justify-start h-full w-full bg-gray-200 dark:bg-gray-900 p-4 gap-16">
+									<div className="pt-16 flex flex-col items-start justify-start h-full w-full bg-gray-200 dark:bg-gray-900 p-4 gap-16">
-                    <div className="w-full">
+										<div className="w-full">
-                      <label
+											<label
-                        htmlFor="name"
+												htmlFor="name"
-                        className="block mb-2 font-medium text-gray-700 dark:text-white"
+												className="block mb-2 font-medium text-gray-700 dark:text-white"
-                      >
+											>
-                        Name
+												Name
-                      </label>
+											</label>
-                      <input
+											<input
-                        onChange={(event) => {
+												onChange={(event) => {
-                          if (event.target.value != "") {
+													if (event.target.value != "") {
-                            let newFlow = flows[tabIndex];
+														let newFlow = flows[tabIndex];
-                            newFlow.name = event.target.value;
+														newFlow.name = event.target.value;
-                            setName(event.target.value);
+														setName(event.target.value);
-                            updateFlow(newFlow);
+														updateFlow(newFlow);
-                          } else {
+													} else {
-                            setName(event.target.value);
+														setName(event.target.value);
-                          }
+													}
-                        }}
+												}}
-                        type="text"
+												type="text"
-                        name="name"
+												name="name"
-                        value={name ?? null}
+												value={name ?? null}
-                        placeholder="File name"
+												placeholder="File name"
-                        id="name"
+												id="name"
-                        className="focus:border focus:border-blue block w-full px-3 py-2 border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-blue-500 focus:border-blue-500 dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500 text-gray-900 dark:text-gray-100"
+												className="focus:border focus:border-blue block w-full px-3 py-2 border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-blue-500 focus:border-blue-500 dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500 text-gray-900 dark:text-gray-100"
-                      />
+											/>
-                    </div>
+										</div>
-                    <div className="w-full">
+										<div className="w-full">
-                      <label
+											<label
-                        htmlFor="description"
+												htmlFor="description"
-                        className="block mb-2 font-medium text-gray-700 dark:text-white"
+												className="block mb-2 font-medium text-gray-700 dark:text-white"
-                      >
+											>
-                        Description{" "}
+												Description{" "}
-                        <span className="text-gray-400 text-sm">
+												<span className="text-gray-400 text-sm">
-                          {" "}
+													{" "}
-                          (optional)
+													(optional)
-                        </span>
+												</span>
-                      </label>
+											</label>
-                      <textarea
+											<textarea
-                        name="description"
+												name="description"
-                        id="description"
+												id="description"
-                        onChange={(event) => {
+												onChange={(event) => {
-                          let newFlow = flows[tabIndex];
+													let newFlow = flows[tabIndex];
-                          newFlow.description = event.target.value;
+													newFlow.description = event.target.value;
-                          updateFlow(newFlow);
+													updateFlow(newFlow);
-                        }}
+												}}
-                        value={flows[tabIndex].description ?? null}
+												value={flows[tabIndex].description ?? null}
-                        placeholder="Flow description"
+												placeholder="Flow description"
-                        rows={3}
+												rows={3}
-                        className=" focus:border focus:border-blue block w-full px-3 py-2 border-gray-300 text-gray-900 dark:text-gray-100 rounded-md shadow-sm focus:outline-none focus:ring-blue-500 focus:border-blue-500 dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500"
+												className=" focus:border focus:border-blue block w-full px-3 py-2 border-gray-300 text-gray-900 dark:text-gray-100 rounded-md shadow-sm focus:outline-none focus:ring-blue-500 focus:border-blue-500 dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500"
-                      ></textarea>
+											></textarea>
-                    </div>
+										</div>
 
-                    <div>
+										<div>
-                      <label htmlFor="checkbox" className="flex items-center">
+											<label htmlFor="checkbox" className="flex items-center">
-                        <input
+												<input
-                          onChange={(event) => {
+													onChange={(event) => {
-                            setChecked(event.target.checked);
+														setChecked(event.target.checked);
-                          }}
+													}}
-                          checked={checked}
+													checked={checked}
-                          id="checkbox"
+													id="checkbox"
-                          type="checkbox"
+													type="checkbox"
-                          className="h-4 w-4 text-blue-600 border-gray-300 rounded dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500"
+													className="h-4 w-4 text-blue-600 border-gray-300 rounded dark:bg-gray-800 dark:border-gray-600 dark:focus:border-blue-500 dark:focus:ring-blue-500"
-                        />
+												/>
-                        <span className="ml-2 font-medium text-gray-700 dark:text-white">
+												<span className="ml-2 font-medium text-gray-700 dark:text-white">
-                          Save with my API keys
+													Save with my API keys
-                        </span>
+												</span>
-                      </label>
+											</label>
-                    </div>
+										</div>
-                    <div className="w-full flex justify-end">
+										<div className="w-full flex justify-end">
-                      <button
+											<button
-                        onClick={() => {
+												onClick={() => {
-                          if (checked) downloadFlow(flows[tabIndex]);
+													if (checked) downloadFlow(flows[tabIndex]);
-                          else downloadFlow(removeApiKeys(flows[tabIndex]));
+													else downloadFlow(removeApiKeys(flows[tabIndex]));
-                        }}
+												}}
-                        className="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded"
+												className="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded"
-                      >
+											>
-                        Download Flow
+												Download Flow
-                      </button>
+											</button>
-                    </div>
+										</div>
-                  </div>
+									</div>
-                </div>
+								</div>
-              </Dialog.Panel>
+							</Dialog.Panel>
-            </Transition.Child>
+						</Transition.Child>
-          </div>
+					</div>
-        </div>
+				</div>
-      </Dialog>
+			</Dialog>
-    </Transition.Root>
+		</Transition.Root>
-  );
+	);
 }

src/frontend/src/modals/importModal/index.tsx

@@ -1,11 +1,11 @@
 import { Dialog, Transition } from "@headlessui/react";
 import {
-  XMarkIcon,
+	XMarkIcon,
-  ArrowDownTrayIcon,
+	ArrowDownTrayIcon,
-  DocumentDuplicateIcon,
+	DocumentDuplicateIcon,
-  ComputerDesktopIcon,
+	ComputerDesktopIcon,
-  ArrowUpTrayIcon,
+	ArrowUpTrayIcon,
-  ArrowLeftIcon,
+	ArrowLeftIcon,
 } from "@heroicons/react/24/outline";
 import { Fragment, useContext, useRef, useState } from "react";
 import { PopUpContext } from "../../contexts/popUpContext";
@@ -19,214 +19,214 @@ import { FlowType } from "../../types/flow";
 import { classNames, toNormalCase } from "../../utils";
 
 export default function ImportModal() {
-  const [open, setOpen] = useState(true);
+	const [open, setOpen] = useState(true);
-  const { setErrorData } = useContext(alertContext);
+	const { setErrorData } = useContext(alertContext);
-  const { closePopUp } = useContext(PopUpContext);
+	const { closePopUp } = useContext(PopUpContext);
-  const ref = useRef();
+	const ref = useRef();
-  const [showExamples, setShowExamples] = useState(false);
+	const [showExamples, setShowExamples] = useState(false);
-  const [loadingExamples, setLoadingExamples] = useState(false);
+	const [loadingExamples, setLoadingExamples] = useState(false);
-  const [examples, setExamples] = useState<FlowType[]>([]);
+	const [examples, setExamples] = useState<FlowType[]>([]);
-  const { uploadFlow, addFlow } = useContext(TabsContext);
+	const { uploadFlow, addFlow } = useContext(TabsContext);
-  function setModalOpen(x: boolean) {
+	function setModalOpen(x: boolean) {
-    setOpen(x);
+		setOpen(x);
-    if (x === false) {
+		if (x === false) {
-      setTimeout(() => {
+			setTimeout(() => {
-        closePopUp();
+				closePopUp();
-      }, 300);
+			}, 300);
-    }
+		}
-  }
+	}
 
-  function handleExamples() {
+	function handleExamples() {
-    setLoadingExamples(true);
+		setLoadingExamples(true);
-    getExamples()
+		getExamples()
-      .then((result) => {
+			.then((result) => {
-        setLoadingExamples(false);
+				setLoadingExamples(false);
-        setExamples(result);
+				setExamples(result);
-      })
+			})
-      .catch((error) =>
+			.catch((error) =>
-        setErrorData({
+				setErrorData({
-          title: "there was an error loading examples, please try again",
+					title: "there was an error loading examples, please try again",
-          list: [error.message],
+					list: [error.message],
-        })
+				})
-      );
+			);
-  }
+	}
 
-  return (
+	return (
-    <Transition.Root show={open} appear={true} as={Fragment}>
+		<Transition.Root show={open} appear={true} as={Fragment}>
-      <Dialog
+			<Dialog
-        as="div"
+				as="div"
-        className="relative z-10"
+				className="relative z-10"
-        onClose={setModalOpen}
+				onClose={setModalOpen}
-        initialFocus={ref}
+				initialFocus={ref}
-      >
+			>
-        <Transition.Child
+				<Transition.Child
-          as={Fragment}
+					as={Fragment}
-          enter="ease-out duration-300"
+					enter="ease-out duration-300"
-          enterFrom="opacity-0"
+					enterFrom="opacity-0"
-          enterTo="opacity-100"
+					enterTo="opacity-100"
-          leave="ease-in duration-200"
+					leave="ease-in duration-200"
-          leaveFrom="opacity-100"
+					leaveFrom="opacity-100"
-          leaveTo="opacity-0"
+					leaveTo="opacity-0"
-        >
+				>
-          <div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
+					<div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
-        </Transition.Child>
+				</Transition.Child>
 
-        <div className="fixed inset-0 z-10 overflow-y-auto">
+				<div className="fixed inset-0 z-10 overflow-y-auto">
-          <div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
+					<div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
-            <Transition.Child
+						<Transition.Child
-              as={Fragment}
+							as={Fragment}
-              enter="ease-out duration-300"
+							enter="ease-out duration-300"
-              enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-              enterTo="opacity-100 translate-y-0 sm:scale-100"
+							enterTo="opacity-100 translate-y-0 sm:scale-100"
-              leave="ease-in duration-200"
+							leave="ease-in duration-200"
-              leaveFrom="opacity-100 translate-y-0 sm:scale-100"
+							leaveFrom="opacity-100 translate-y-0 sm:scale-100"
-              leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-            >
+						>
-              <Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
+							<Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
-                <div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
+								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
-                  <button
+									<button
-                    type="button"
+										type="button"
-                    className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
+										className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
-                    onClick={() => {
+										onClick={() => {
-                      setModalOpen(false);
+											setModalOpen(false);
-                    }}
+										}}
-                  >
+									>
-                    <span className="sr-only">Close</span>
+										<span className="sr-only">Close</span>
-                    <XMarkIcon className="h-6 w-6" aria-hidden="true" />
+										<XMarkIcon className="h-6 w-6" aria-hidden="true" />
-                  </button>
+									</button>
-                </div>
+								</div>
-                {showExamples && (
+								{showExamples && (
-                  <>
+									<>
-                    <div className="z-50 absolute top-2 left-0 hidden pt-4 pl-4 sm:block">
+										<div className="z-50 absolute top-2 left-0 hidden pt-4 pl-4 sm:block">
-                      <button
+											<button
-                        type="button"
+												type="button"
-                        className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
+												className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
-                        onClick={() => {
+												onClick={() => {
-                          setShowExamples(false);
+													setShowExamples(false);
-                        }}
+												}}
-                      >
+											>
-                        <span className="sr-only">Close</span>
+												<span className="sr-only">Close</span>
-                        <ArrowLeftIcon className="h-6 w-6" aria-hidden="true" />
+												<ArrowLeftIcon className="h-6 w-6" aria-hidden="true" />
-                      </button>
+											</button>
-                    </div>
+										</div>
-                    <div className="z-50 absolute bottom-2 left-1/2 transform -translate-x-1/2 hidden pt-4 pl-2 sm:block text-center">
+										<div className="z-50 absolute bottom-2 left-1/2 transform -translate-x-1/2 hidden pt-4 pl-2 sm:block text-center">
-                      <a
+											<a
-                        href="https://github.com/logspace-ai/langflow_examples"
+												href="https://github.com/logspace-ai/langflow_examples"
-                        target="_blank"
+												target="_blank"
-                        className="flex items-center justify-center"
+												className="flex items-center justify-center"
-                        rel="noreferrer"
+												rel="noreferrer"
-                      >
+											>
-                        <svg
+												<svg
-                          width="24"
+													width="24"
-                          viewBox="0 0 98 96"
+													viewBox="0 0 98 96"
-                          xmlns="http://www.w3.org/2000/svg"
+													xmlns="http://www.w3.org/2000/svg"
-                        >
+												>
-                          <path
+													<path
-                            fill-rule="evenodd"
+														fill-rule="evenodd"
-                            clip-rule="evenodd"
+														clip-rule="evenodd"
-                            d="M48.854 0C21.839 0 0 22 0 49.217c0 21.756 13.993 40.172 33.405 46.69 2.427.49 3.316-1.059 3.316-2.362 0-1.141-.08-5.052-.08-9.127-13.59 2.934-16.42-5.867-16.42-5.867-2.184-5.704-5.42-7.17-5.42-7.17-4.448-3.015.324-3.015.324-3.015 4.934.326 7.523 5.052 7.523 5.052 4.367 7.496 11.404 5.378 14.235 4.074.404-3.178 1.699-5.378 3.074-6.6-10.839-1.141-22.243-5.378-22.243-24.283 0-5.378 1.94-9.778 5.014-13.2-.485-1.222-2.184-6.275.486-13.038 0 0 4.125-1.304 13.426 5.052a46.97 46.97 0 0 1 12.214-1.63c4.125 0 8.33.571 12.213 1.63 9.302-6.356 13.427-5.052 13.427-5.052 2.67 6.763.97 11.816.485 13.038 3.155 3.422 5.015 7.822 5.015 13.2 0 18.905-11.404 23.06-22.324 24.283 1.78 1.548 3.316 4.481 3.316 9.126 0 6.6-.08 11.897-.08 13.526 0 1.304.89 2.853 3.316 2.364 19.412-6.52 33.405-24.935 33.405-46.691C97.707 22 75.788 0 48.854 0z"
+														d="M48.854 0C21.839 0 0 22 0 49.217c0 21.756 13.993 40.172 33.405 46.69 2.427.49 3.316-1.059 3.316-2.362 0-1.141-.08-5.052-.08-9.127-13.59 2.934-16.42-5.867-16.42-5.867-2.184-5.704-5.42-7.17-5.42-7.17-4.448-3.015.324-3.015.324-3.015 4.934.326 7.523 5.052 7.523 5.052 4.367 7.496 11.404 5.378 14.235 4.074.404-3.178 1.699-5.378 3.074-6.6-10.839-1.141-22.243-5.378-22.243-24.283 0-5.378 1.94-9.778 5.014-13.2-.485-1.222-2.184-6.275.486-13.038 0 0 4.125-1.304 13.426 5.052a46.97 46.97 0 0 1 12.214-1.63c4.125 0 8.33.571 12.213 1.63 9.302-6.356 13.427-5.052 13.427-5.052 2.67 6.763.97 11.816.485 13.038 3.155 3.422 5.015 7.822 5.015 13.2 0 18.905-11.404 23.06-22.324 24.283 1.78 1.548 3.316 4.481 3.316 9.126 0 6.6-.08 11.897-.08 13.526 0 1.304.89 2.853 3.316 2.364 19.412-6.52 33.405-24.935 33.405-46.691C97.707 22 75.788 0 48.854 0z"
-                            fill="#24292f"
+														fill="#24292f"
-                          />
+													/>
-                        </svg>
+												</svg>
-                        <span className="ml-2">LangFlow Examples</span>
+												<span className="ml-2">LangFlow Examples</span>
-                      </a>
+											</a>
-                    </div>
+										</div>
-                  </>
+									</>
-                )}
+								)}
-                <div className="h-full w-full flex flex-col justify-center items-center">
+								<div className="h-full w-full flex flex-col justify-center items-center">
-                  <div className="flex w-full pb-4 z-10 justify-center shadow-sm">
+									<div className="flex w-full pb-4 z-10 justify-center shadow-sm">
-                    <div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
+										<div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
-                      <ArrowUpTrayIcon
+											<ArrowUpTrayIcon
-                        className="h-6 w-6 text-blue-600"
+												className="h-6 w-6 text-blue-600"
-                        aria-hidden="true"
+												aria-hidden="true"
-                      />
+											/>
-                    </div>
+										</div>
-                    <div className="mt-4 text-center sm:ml-4 sm:text-left">
+										<div className="mt-4 text-center sm:ml-4 sm:text-left">
-                      <Dialog.Title
+											<Dialog.Title
-                        as="h3"
+												as="h3"
-                        className="text-lg font-medium dark:text-white leading-10 text-gray-900"
+												className="text-lg font-medium dark:text-white leading-10 text-gray-900"
-                      >
+											>
-                        {showExamples ? "Select an example" : "Import from"}
+												{showExamples ? "Select an example" : "Import from"}
-                      </Dialog.Title>
+											</Dialog.Title>
-                    </div>
+										</div>
-                  </div>
+									</div>
-                  <div
+									<div
-                    className={classNames(
+										className={classNames(
-                      "h-full w-full bg-gray-200 dark:bg-gray-900 gap-4",
+											"h-full w-full bg-gray-200 dark:bg-gray-900 gap-4",
-                      showExamples && !loadingExamples
+											showExamples && !loadingExamples
-                        ? "flex flex-row start justify-start items-start p-9 flex-wrap overflow-auto"
+												? "flex flex-row start justify-start items-start p-9 flex-wrap overflow-auto"
-                        : "flex flex-row justify-center items-center p-4"
+												: "flex flex-row justify-center items-center p-4"
-                    )}
+										)}
-                  >
+									>
-                    {!showExamples && (
+										{!showExamples && (
-                      <div className="flex h-full w-full justify-evenly items-center">
+											<div className="flex h-full w-full justify-evenly items-center">
-                        <ButtonBox
+												<ButtonBox
-                          size="big"
+													size="big"
-                          bgColor="bg-emerald-500"
+													bgColor="bg-emerald-500"
-                          description="Prebuilt Examples"
+													description="Prebuilt Examples"
-                          icon={
+													icon={
-                            <DocumentDuplicateIcon className="h-10 w-10 flex-shrink-0" />
+														<DocumentDuplicateIcon className="h-10 w-10 flex-shrink-0" />
-                          }
+													}
-                          onClick={() => {
+													onClick={() => {
-                            setShowExamples(true);
+														setShowExamples(true);
-                            handleExamples();
+														handleExamples();
-                          }}
+													}}
-                          textColor="text-emerald-400"
+													textColor="text-emerald-400"
-                          title="Examples"
+													title="Examples"
-                        ></ButtonBox>
+												></ButtonBox>
-                        <ButtonBox
+												<ButtonBox
-                          size="big"
+													size="big"
-                          bgColor="bg-blue-500"
+													bgColor="bg-blue-500"
-                          description="Import from Local"
+													description="Import from Local"
-                          icon={
+													icon={
-                            <ComputerDesktopIcon className="h-10 w-10 flex-shrink-0" />
+														<ComputerDesktopIcon className="h-10 w-10 flex-shrink-0" />
-                          }
+													}
-                          onClick={() => {
+													onClick={() => {
-                            uploadFlow();
+														uploadFlow();
-                            setModalOpen(false);
+														setModalOpen(false);
-                          }}
+													}}
-                          textColor="text-blue-500"
+													textColor="text-blue-500"
-                          title="Local file"
+													title="Local file"
-                        ></ButtonBox>
+												></ButtonBox>
-                      </div>
+											</div>
-                    )}
+										)}
-                    {showExamples && loadingExamples && (
+										{showExamples && loadingExamples && (
-                      <div className="flex align-middle justify-center items-center">
+											<div className="flex align-middle justify-center items-center">
-                        <LoadingComponent remSize={30} />
+												<LoadingComponent remSize={30} />
-                      </div>
+											</div>
-                    )}
+										)}
-                    {showExamples &&
+										{showExamples &&
-                      !loadingExamples &&
+											!loadingExamples &&
-                      examples.map((example, index) => {
+											examples.map((example, index) => {
-                        return (
+												return (
-                          <div id="index">
+													<div id="index">
-                            {" "}
+														{" "}
-                            <ButtonBox
+														<ButtonBox
-                              size="small"
+															size="small"
-                              bgColor="bg-emerald-500"
+															bgColor="bg-emerald-500"
-                              description={
+															description={
-                                example.description ?? "Prebuilt Examples"
+																example.description ?? "Prebuilt Examples"
-                              }
+															}
-                              icon={
+															icon={
-                                <DocumentDuplicateIcon className="h-6 w-6 flex-shrink-0" />
+																<DocumentDuplicateIcon className="h-6 w-6 flex-shrink-0" />
-                              }
+															}
-                              onClick={() => {
+															onClick={() => {
-                                addFlow(example);
+																addFlow(example);
-                                setModalOpen(false);
+																setModalOpen(false);
-                              }}
+															}}
-                              textColor="text-emerald-400"
+															textColor="text-emerald-400"
-                              title={toNormalCase(example.name)}
+															title={toNormalCase(example.name)}
-                            ></ButtonBox>
+														></ButtonBox>
-                          </div>
+													</div>
-                        );
+												);
-                      })}
+											})}
-                  </div>
+									</div>
-                </div>
+								</div>
-              </Dialog.Panel>
+							</Dialog.Panel>
-            </Transition.Child>
+						</Transition.Child>
-          </div>
+					</div>
-        </div>
+				</div>
-      </Dialog>
+			</Dialog>
-    </Transition.Root>
+		</Transition.Root>
-  );
+	);
 }

src/frontend/src/modals/promptModal/index.tsx

@@ -135,7 +135,8 @@ export default function PromptAreaModal({
 														return setErrorData({
 															title:
 																"There is something wrong with this prompt, please review it",
-														list:[error.response.data.detail]});
+															list: [error.response.data.detail],
+														});
 													});
 											}}
 										>

src/frontend/src/modals/textAreaModal/index.tsx

@@ -1,104 +1,123 @@
 import { Dialog, Transition } from "@headlessui/react";
-import { XMarkIcon, ClipboardDocumentListIcon } from "@heroicons/react/24/outline";
+import {
+	XMarkIcon,
+	ClipboardDocumentListIcon,
+} from "@heroicons/react/24/outline";
 import { Fragment, useContext, useRef, useState } from "react";
 import { PopUpContext } from "../../contexts/popUpContext";
 
-export default function TextAreaModal({value, setValue}:{setValue:(value:string)=>void,value:string|string[]}){
+export default function TextAreaModal({
-    const [open, setOpen] = useState(true);
+	value,
-    const [myValue, setMyValue] = useState(value);
+	setValue,
-    const { closePopUp } = useContext(PopUpContext);
+}: {
-    const ref = useRef();
+	setValue: (value: string) => void;
-    function setModalOpen(x:boolean){
+	value: string | string[];
-        setOpen(x);
+}) {
-        if(x === false){
+	const [open, setOpen] = useState(true);
-            setTimeout(() => {closePopUp()}, 300);
+	const [myValue, setMyValue] = useState(value);
-        }
+	const { closePopUp } = useContext(PopUpContext);
-    }
+	const ref = useRef();
-    return (
+	function setModalOpen(x: boolean) {
-        <Transition.Root show={open} appear={true} as={Fragment}>
+		setOpen(x);
-      <Dialog
+		if (x === false) {
-        as="div"
+			setTimeout(() => {
-        className="relative z-10"
+				closePopUp();
-        onClose={setModalOpen}
+			}, 300);
-        initialFocus={ref}
+		}
-      >
+	}
-        <Transition.Child
+	return (
-          as={Fragment}
+		<Transition.Root show={open} appear={true} as={Fragment}>
-          enter="ease-out duration-300"
+			<Dialog
-          enterFrom="opacity-0"
+				as="div"
-          enterTo="opacity-100"
+				className="relative z-10"
-          leave="ease-in duration-200"
+				onClose={setModalOpen}
-          leaveFrom="opacity-100"
+				initialFocus={ref}
-          leaveTo="opacity-0"
+			>
-        >
+				<Transition.Child
-          <div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
+					as={Fragment}
-        </Transition.Child>
+					enter="ease-out duration-300"
+					enterFrom="opacity-0"
+					enterTo="opacity-100"
+					leave="ease-in duration-200"
+					leaveFrom="opacity-100"
+					leaveTo="opacity-0"
+				>
+					<div className="fixed inset-0 bg-gray-500 dark:bg-gray-600 dark:bg-opacity-75 bg-opacity-75 transition-opacity" />
+				</Transition.Child>
 
-        <div className="fixed inset-0 z-10 overflow-y-auto">
+				<div className="fixed inset-0 z-10 overflow-y-auto">
-          <div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
+					<div className="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
-            <Transition.Child
+						<Transition.Child
-              as={Fragment}
+							as={Fragment}
-              enter="ease-out duration-300"
+							enter="ease-out duration-300"
-              enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-              enterTo="opacity-100 translate-y-0 sm:scale-100"
+							enterTo="opacity-100 translate-y-0 sm:scale-100"
-              leave="ease-in duration-200"
+							leave="ease-in duration-200"
-              leaveFrom="opacity-100 translate-y-0 sm:scale-100"
+							leaveFrom="opacity-100 translate-y-0 sm:scale-100"
-              leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
-            >
+						>
-              <Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
+							<Dialog.Panel className="relative flex flex-col justify-between transform h-[600px] overflow-hidden rounded-lg bg-white dark:bg-gray-800 text-left shadow-xl transition-all sm:my-8 w-[700px]">
-                <div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
+								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
-                  <button
+									<button
-                    type="button"
+										type="button"
-                    className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
+										className="rounded-md text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2"
-                    onClick={() => {
+										onClick={() => {
-                      setModalOpen(false);
+											setModalOpen(false);
-                    }}
+										}}
-                  >
+									>
-                    <span className="sr-only">Close</span>
+										<span className="sr-only">Close</span>
-                    <XMarkIcon className="h-6 w-6" aria-hidden="true" />
+										<XMarkIcon className="h-6 w-6" aria-hidden="true" />
-                  </button>
+									</button>
-                </div>
+								</div>
-                <div className="h-full w-full flex flex-col justify-center items-center">
+								<div className="h-full w-full flex flex-col justify-center items-center">
-                  <div className="flex w-full pb-4 z-10 justify-center shadow-sm">
+									<div className="flex w-full pb-4 z-10 justify-center shadow-sm">
-                    <div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
+										<div className="mx-auto mt-4 flex h-12 w-12 flex-shrink-0 items-center justify-center rounded-full bg-blue-100 dark:bg-gray-900 sm:mx-0 sm:h-10 sm:w-10">
-                      <ClipboardDocumentListIcon
+											<ClipboardDocumentListIcon
-                        className="h-6 w-6 text-blue-600"
+												className="h-6 w-6 text-blue-600"
-                        aria-hidden="true"
+												aria-hidden="true"
-                      />
+											/>
-                    </div>
+										</div>
-                    <div className="mt-4 text-center sm:ml-4 sm:text-left">
+										<div className="mt-4 text-center sm:ml-4 sm:text-left">
-                      <Dialog.Title
+											<Dialog.Title
-                        as="h3"
+												as="h3"
-                        className="text-lg font-medium dark:text-white leading-10 text-gray-900"
+												className="text-lg font-medium dark:text-white leading-10 text-gray-900"
-                      >
+											>
-                        Edit text
+												Edit text
-                      </Dialog.Title>
+											</Dialog.Title>
-                    </div>
+										</div>
-                  </div>
+									</div>
-                  <div className="h-full w-full bg-gray-200 dark:bg-gray-900 p-4 gap-4 flex flex-row justify-center items-center">
+									<div className="h-full w-full bg-gray-200 dark:bg-gray-900 p-4 gap-4 flex flex-row justify-center items-center">
-                    <div className="flex h-full w-full">
+										<div className="flex h-full w-full">
-                      <div className="overflow-hidden px-4 py-5 sm:p-6 w-full rounded-lg bg-white dark:bg-gray-800 shadow">
+											<div className="overflow-hidden px-4 py-5 sm:p-6 w-full rounded-lg bg-white dark:bg-gray-800 shadow">
-                        <textarea ref={ref} className="form-input h-full w-full rounded-lg border-gray-300 dark:border-gray-700 dark:bg-gray-900 dark:text-white" value={myValue} onChange={(e) => {setMyValue(e.target.value); setValue(e.target.value)}}/>
+												<textarea
-                      </div>
+													ref={ref}
-                    </div>
+													className="form-input h-full w-full rounded-lg border-gray-300 dark:border-gray-700 dark:bg-gray-900 dark:text-white"
-                  </div>
+													value={myValue}
-                  <div className="bg-gray-200 dark:bg-gray-900 w-full pb-3 flex flex-row-reverse px-4">
+													onChange={(e) => {
-                    <button
+														setMyValue(e.target.value);
-                      type="button"
+														setValue(e.target.value);
-                      className="inline-flex w-full justify-center rounded-md border border-transparent bg-indigo-600 px-4 py-2 text-base font-medium text-white shadow-sm hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2 sm:ml-3 sm:w-auto sm:text-sm"
+													}}
-                      onClick={() => {
+												/>
-                        setModalOpen(false);
+											</div>
-                      }}
+										</div>
-                    >
+									</div>
-                      Finish editing
+									<div className="bg-gray-200 dark:bg-gray-900 w-full pb-3 flex flex-row-reverse px-4">
-                    </button>
+										<button
-                  </div>
+											type="button"
-                </div>
+											className="inline-flex w-full justify-center rounded-md border border-transparent bg-indigo-600 px-4 py-2 text-base font-medium text-white shadow-sm hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2 sm:ml-3 sm:w-auto sm:text-sm"
-              </Dialog.Panel>
+											onClick={() => {
-            </Transition.Child>
+												setModalOpen(false);
-          </div>
+											}}
-        </div>
+										>
-      </Dialog>
+											Finish editing
-    </Transition.Root>
+										</button>
-    )
+									</div>
-}
+								</div>
+							</Dialog.Panel>
+						</Transition.Child>
+					</div>
+				</div>
+			</Dialog>
+		</Transition.Root>
+	);
+}

src/frontend/src/pages/FlowPage/components/ConnectionLineComponent/index.tsx

@@ -1,14 +1,12 @@
-import { ConnectionLineComponentProps } from 'reactflow';
+import { ConnectionLineComponentProps } from "reactflow";
-
-
 
 const ConnectionLineComponent = ({
 	fromX,
 	fromY,
 	toX,
 	toY,
-	connectionLineStyle = {} // provide a default value for connectionLineStyle
+	connectionLineStyle = {}, // provide a default value for connectionLineStyle
-}:ConnectionLineComponentProps) => {
+}: ConnectionLineComponentProps) => {
 	return (
 		<g>
 			<path

src/frontend/src/pages/FlowPage/components/DisclosureComponent/index.tsx

@@ -1,14 +1,11 @@
-import {
+import { ChevronRightIcon } from "@heroicons/react/24/solid";
-	ChevronRightIcon,
-} from "@heroicons/react/24/solid";
 import { Disclosure } from "@headlessui/react";
 import { DisclosureComponentType } from "../../../../types/components";
 
 export default function DisclosureComponent({
 	button: { title, Icon, buttons = [] },
 	children,
-}: DisclosureComponentType
+}: DisclosureComponentType) {
-) {
 	return (
 		<Disclosure as="div" key={title}>
 			{({ open }) => (

src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx

@@ -3,13 +3,10 @@ import DisclosureComponent from "../DisclosureComponent";
 import { nodeColors, nodeIcons, nodeNames } from "../../../../utils";
 import { useContext, useEffect, useState } from "react";
 import { typesContext } from "../../../../contexts/typesContext";
-import {
+import { APIClassType, APIObjectType } from "../../../../types/api";
-	APIClassType,
-	APIObjectType,
-} from "../../../../types/api";
 
 export default function ExtraSidebar() {
-	const {data} = useContext(typesContext)
+	const { data } = useContext(typesContext);
 
 	function onDragStart(
 		event: React.DragEvent<any>,
@@ -22,45 +19,49 @@ export default function ExtraSidebar() {
 
 	return (
 		<div className="mt-1 w-full">
-			{Object.keys(data).sort().map((d: keyof APIObjectType, i) => (
+			{Object.keys(data)
-				<DisclosureComponent
+				.sort()
-					key={i}
+				.map((d: keyof APIObjectType, i) => (
-					button={{
+					<DisclosureComponent
-						title: nodeNames[d] ?? nodeNames.unknown,
+						key={i}
-						Icon: nodeIcons[d] ?? nodeIcons.unknown,
+						button={{
-					}}
+							title: nodeNames[d] ?? nodeNames.unknown,
-				>
+							Icon: nodeIcons[d] ?? nodeIcons.unknown,
-					<div className="p-2 flex flex-col gap-2">
+						}}
-						{Object.keys(data[d]).sort().map((t: string, k) => (
+					>
-							<div key={k}>
+						<div className="p-2 flex flex-col gap-2">
-								<div
+							{Object.keys(data[d])
-									draggable
+								.sort()
-									className={" cursor-grab border-l-8 rounded-l-md"}
+								.map((t: string, k) => (
-									style={{
+									<div key={k}>
-										borderLeftColor: nodeColors[d] ?? nodeColors.unknown,
+										<div
-									}}
+											draggable
-									onDragStart={(event) =>
+											className={" cursor-grab border-l-8 rounded-l-md"}
-										onDragStart(event, {
+											style={{
-											type: t,
+												borderLeftColor: nodeColors[d] ?? nodeColors.unknown,
-											node: data[d][t],
+											}}
-										})
+											onDragStart={(event) =>
-									}
+												onDragStart(event, {
-								>
+													type: t,
-									<div className="flex w-full justify-between text-sm px-3 py-1 items-center border-dashed border-gray-400 dark:border-gray-600 border-l-0 rounded-md rounded-l-none border">
+													node: data[d][t],
-										<span className="text-black dark:text-white w-36 truncate text-xs">
+												})
-											{t}
+											}
-										</span>
+										>
-										<Bars2Icon className="w-4 h-6  text-gray-400 dark:text-gray-600" />
+											<div className="flex w-full justify-between text-sm px-3 py-1 items-center border-dashed border-gray-400 dark:border-gray-600 border-l-0 rounded-md rounded-l-none border">
+												<span className="text-black dark:text-white w-36 truncate text-xs">
+													{t}
+												</span>
+												<Bars2Icon className="w-4 h-6  text-gray-400 dark:text-gray-600" />
+											</div>
+										</div>
 									</div>
-								</div>
+								))}
-							</div>
+							{Object.keys(data[d]).length === 0 && (
-						))}
+								<div className="text-gray-400 text-center">Coming soon</div>
-						{Object.keys(data[d]).length === 0 && (
+							)}
-							<div className="text-gray-400 text-center">Coming soon</div>
+						</div>
-						)}
+					</DisclosureComponent>
-					</div>
+				))}
-				</DisclosureComponent>
-			))}
 		</div>
 	);
 }

src/frontend/src/pages/FlowPage/components/tabComponent/index.tsx

@@ -5,9 +5,16 @@ import { FlowType } from "../../../../types/flow";
 
 var _ = require("lodash");
 
-export default function TabComponent({ selected, flow, onClick }:{flow:FlowType,selected:boolean,onClick:()=>void}) {
+export default function TabComponent({
-	const { removeFlow, updateFlow, flows } =
+	selected,
-		useContext(TabsContext);
+	flow,
+	onClick,
+}: {
+	flow: FlowType;
+	selected: boolean;
+	onClick: () => void;
+}) {
+	const { removeFlow, updateFlow, flows } = useContext(TabsContext);
 	const [isRename, setIsRename] = useState(false);
 	const [value, setValue] = useState("");
 	return (
@@ -19,7 +26,7 @@ export default function TabComponent({ selected, flow, onClick }:{flow:FlowType,
 						onClick={onClick}
 					>
 						<span className="w-32 truncate text-left">{flow.name}</span>
-						
+
 						<button
 							onClick={(e) => {
 								e.stopPropagation();
@@ -85,4 +92,3 @@ export default function TabComponent({ selected, flow, onClick }:{flow:FlowType,
 		</>
 	);
 }
-

src/frontend/src/pages/FlowPage/components/tabsManagerComponent/index.tsx

@@ -102,9 +102,7 @@ export default function TabsManagerComponent() {
 									>
 										<AlertDropdown />
 									</div>
-                  <div className="h-screen w-screen fixed top-0 left-0">
+									<div className="h-screen w-screen fixed top-0 left-0"></div>
-                    
-                  </div>
 								</>
 							);
 						}}

src/frontend/src/pages/FlowPage/hooks/useUndoRedo.ts

@@ -0,0 +1,109 @@
+import { useCallback, useEffect, useState } from 'react';
+import { Edge, Node, useReactFlow } from 'reactflow';
+
+type UseUndoRedoOptions = {
+  maxHistorySize: number;
+  enableShortcuts: boolean;
+};
+
+type UseUndoRedo = (options?: UseUndoRedoOptions) => {
+  undo: () => void;
+  redo: () => void;
+  takeSnapshot: () => void;
+  canUndo: boolean;
+  canRedo: boolean;
+};
+
+type HistoryItem = {
+  nodes: Node[];
+  edges: Edge[];
+};
+
+const defaultOptions: UseUndoRedoOptions = {
+  maxHistorySize: 100,
+  enableShortcuts: true,
+};
+
+// https://redux.js.org/usage/implementing-undo-history
+export const useUndoRedo: UseUndoRedo = ({
+  maxHistorySize = defaultOptions.maxHistorySize,
+  enableShortcuts = defaultOptions.enableShortcuts,
+} = defaultOptions) => {
+  // the past and future arrays store the states that we can jump to
+  const [past, setPast] = useState<HistoryItem[]>([]);
+  const [future, setFuture] = useState<HistoryItem[]>([]);
+
+  const { setNodes, setEdges, getNodes, getEdges } = useReactFlow();
+
+  const takeSnapshot = useCallback(() => {
+    // push the current graph to the past state
+    setPast((past) => [
+      ...past.slice(past.length - maxHistorySize + 1, past.length),
+      { nodes: getNodes(), edges: getEdges() },
+    ]);
+
+    // whenever we take a new snapshot, the redo operations need to be cleared to avoid state mismatches
+    setFuture([]);
+  }, [getNodes, getEdges, maxHistorySize]);
+
+  const undo = useCallback(() => {
+    // get the last state that we want to go back to
+    const pastState = past[past.length - 1];
+
+    if (pastState) {
+      // first we remove the state from the history
+      setPast((past) => past.slice(0, past.length - 1));
+      // we store the current graph for the redo operation
+      setFuture((future) => [...future, { nodes: getNodes(), edges: getEdges() }]);
+      // now we can set the graph to the past state
+      setNodes(pastState.nodes);
+      setEdges(pastState.edges);
+    }
+  }, [setNodes, setEdges, getNodes, getEdges, past]);
+
+  const redo = useCallback(() => {
+    const futureState = future[future.length - 1];
+
+    if (futureState) {
+      setFuture((future) => future.slice(0, future.length - 1));
+      setPast((past) => [...past, { nodes: getNodes(), edges: getEdges() }]);
+      setNodes(futureState.nodes);
+      setEdges(futureState.edges);
+    }
+  }, [setNodes, setEdges, getNodes, getEdges, future]);
+
+  useEffect(() => {
+    // this effect is used to attach the global event handlers
+    if (!enableShortcuts) {
+      return;
+    }
+
+    const keyDownHandler = (event: KeyboardEvent) => {
+      if (event.key === 'z' && (event.ctrlKey || event.metaKey) && event.shiftKey) {
+        redo();
+      } 
+      else if (event.key === 'y' && (event.ctrlKey || event.metaKey)) {
+        event.preventDefault(); // prevent the default action
+        redo();
+      } else if (event.key === 'z' && (event.ctrlKey || event.metaKey)) {
+        undo();
+      }
+    };
+
+    document.addEventListener('keydown', keyDownHandler);
+
+    return () => {
+      document.removeEventListener('keydown', keyDownHandler);
+    };
+  }, [undo, redo, enableShortcuts]);
+
+  return {
+    undo,
+    redo,
+    takeSnapshot,
+    canUndo: !!past.length,
+    canRedo: !!future.length,
+  };
+};
+
+export default useUndoRedo;

src/frontend/src/pages/FlowPage/index.tsx

@@ -1,15 +1,21 @@
-import { useCallback, useContext, useEffect, useRef } from "react";
+import { useCallback, useContext, useEffect, useRef, useState } from "react";
 import ReactFlow, {
-  Background,
+	Background,
-  Controls,
+	Controls,
-  addEdge,
+	addEdge,
-  useEdgesState,
+	useEdgesState,
-  useNodesState,
+	useNodesState,
-  useReactFlow,
+	useReactFlow,
-  updateEdge,
+	updateEdge,
-  EdgeChange,
+	EdgeChange,
-  Connection,
+	Connection,
-  Edge,
+	Edge,
+	useKeyPress,
+	useOnSelectionChange,
+	NodeDragHandler,
+	OnEdgesDelete,
+	OnNodesDelete,
+	SelectionDragHandler,
 } from "reactflow";
 import { locationContext } from "../../contexts/locationContext";
 import ExtraSidebar from "./components/extraSidebarComponent";
@@ -22,171 +28,275 @@ import ConnectionLineComponent from "./components/ConnectionLineComponent";
 import { FlowType, NodeType } from "../../types/flow";
 import { APIClassType } from "../../types/api";
 import { isValidConnection } from "../../utils";
+import useUndoRedo from "./hooks/useUndoRedo";
 
 const nodeTypes = {
-  genericNode: GenericNode,
+	genericNode: GenericNode,
 };
 
 var _ = require("lodash");
 
-export default function FlowPage({ flow }:{flow:FlowType}) {
+export default function FlowPage({ flow }: { flow: FlowType }) {
-	let { updateFlow, incrementNodeId} =
+	let { updateFlow, incrementNodeId } =
 		useContext(TabsContext);
 	const { types, reactFlowInstance, setReactFlowInstance, templates } =
 		useContext(typesContext);
 	const reactFlowWrapper = useRef(null);
 
-  const { setExtraComponent, setExtraNavigation } = useContext(locationContext);
+	const { undo, redo, canUndo, canRedo, takeSnapshot } = useUndoRedo();
-  const { setErrorData } = useContext(alertContext);
-  const [nodes, setNodes, onNodesChange] = useNodesState(
-    flow.data?.nodes ?? []
-  );
-  const [edges, setEdges, onEdgesChange] = useEdgesState(
-    flow.data?.edges ?? []
-  );
-  const { setViewport } = useReactFlow();
-  const edgeUpdateSuccessful = useRef(true);
 
-  useEffect(() => {
+	const onKeyDown = (event: React.KeyboardEvent<HTMLDivElement>) => {
-    if (reactFlowInstance && flow) {
+		if ((event.ctrlKey || event.metaKey) && (event.key === 'c') && lastSelection) {
-      flow.data = reactFlowInstance.toObject();
+			event.preventDefault();
-      updateFlow(flow);
+			setLastCopiedSelection(lastSelection);
-    }
+		}
-    // eslint-disable-next-line react-hooks/exhaustive-deps
+		if ((event.ctrlKey || event.metaKey) && (event.key === 'v') && lastCopiedSelection) {
-  }, [nodes, edges]);
+			event.preventDefault();
-  //update flow when tabs change
+			paste();
-  useEffect(() => {
+		}
-    setNodes(flow?.data?.nodes ?? []);
+	}
-    setEdges(flow?.data?.edges ?? []);
-    if (reactFlowInstance) {
-      setViewport(flow?.data?.viewport ?? { x: 1, y: 0, zoom: 0.5 });
-    }
-  }, [flow, reactFlowInstance, setEdges, setNodes, setViewport]);
-  //set extra sidebar
-  useEffect(() => {
-    setExtraComponent(<ExtraSidebar />);
-    setExtraNavigation({ title: "Components" });
-  }, [setExtraComponent, setExtraNavigation]);
 
-  const onEdgesChangeMod = useCallback(
+	const [lastSelection, setLastSelection] = useState(null);
-    (s: EdgeChange[]) => {
+	const [lastCopiedSelection, setLastCopiedSelection] = useState(null);
-      onEdgesChange(s);
-      setNodes((x) => {
-        let newX = _.cloneDeep(x);
-        return newX;
-      });
-    },
-    [onEdgesChange, setNodes]
-  );
 
-  const onConnect = useCallback(
+	const [position, setPosition] = useState({ x: 0, y: 0 });
-    (params: Connection) => {
-      setEdges((eds) =>
-        addEdge({ ...params, className: "animate-pulse" }, eds)
-      );
-      setNodes((x) => {
-        let newX = _.cloneDeep(x);
-        return newX;
-      });
-    },
-    [setEdges, setNodes]
-  );
 
-  const onDragOver = useCallback((event: React.DragEvent) => {
+	const handleMouseMove = (event) => {
-    event.preventDefault();
+		setPosition({ x: event.clientX, y: event.clientY });
-    event.dataTransfer.dropEffect = "move";
+	};
-  }, []);
 
-  const onDrop = useCallback(
+	useOnSelectionChange({
-    (event: React.DragEvent) => {
+		onChange: (flow) => { setLastSelection(flow); },
-      event.preventDefault();
+	})
 
-      // Helper function to generate a unique node ID
+	let paste = () => {
-      function getId() {
+		let minimumX = Infinity;
-        return `dndnode_` + incrementNodeId();
+		let minimumY = Infinity;
-      }
+		let idsMap = {};
+		lastCopiedSelection.nodes.forEach((n) => {
+			if (n.position.y < minimumY) {
+				minimumY = n.position.y
+			}
+			if (n.position.x < minimumX) {
+				minimumX = n.position.x;
+			}
+		});
 
-      // Get the current bounds of the ReactFlow wrapper element
+		const bounds = reactFlowWrapper.current.getBoundingClientRect();
-      const reactflowBounds = reactFlowWrapper.current.getBoundingClientRect();
+		const insidePosition = reactFlowInstance.project({
+			x: position.x - bounds.left,
+			y: position.y - bounds.top
+		});
 
-      // Extract the data from the drag event and parse it as a JSON object
+		lastCopiedSelection.nodes.forEach((n) => {
-      let data: { type: string; node?: APIClassType } = JSON.parse(
-        event.dataTransfer.getData("json")
-      );
 
-      // If data type is not "chatInput" or if there are no "chatInputNode" nodes present in the ReactFlow instance, create a new node
+			// Generate a unique node ID
-      if (
+			let newId = getId();
-        data.type !== "chatInput" ||
+			idsMap[n.id] = newId;
-        (data.type === "chatInput" &&
-          !reactFlowInstance.getNodes().some((n) => n.type === "chatInputNode"))
-      ) {
-        // Calculate the position where the node should be created
-        const position = reactFlowInstance.project({
-          x: event.clientX - reactflowBounds.left,
-          y: event.clientY - reactflowBounds.top,
-        });
 
-        // Generate a unique node ID
+			// Create a new node object
-        let newId = getId();
+			const newNode: NodeType = {
+				id: newId,
+				type: "genericNode",
+				position: {
+					x: insidePosition.x + n.position.x - minimumX,
+					y: insidePosition.y + n.position.y - minimumY,
+				},
+				data: {
+					...n.data,
+					id: newId,
+				},
+			};
 
-        // Create a new node object
+			// Add the new node to the list of nodes in state
-        const newNode: NodeType = {
+			setNodes((nds) => nds.map((e) => ({ ...e, selected: false })).concat({ ...newNode, selected: false }));
-          id: newId,
+		})
-          type: "genericNode",
-          position,
-          data: {
-            ...data,
-            id: newId,
-            value: null,
-          },
-        };
 
-        // Add the new node to the list of nodes in state
+		lastCopiedSelection.edges.forEach((e) => {
-        setNodes((nds) => nds.concat(newNode));
+			let source = idsMap[e.source];
-      } else {
+			let target = idsMap[e.target];
-        // If a chat input node already exists, set an error message
+			let sourceHandleSplitted = e.sourceHandle.split('|');
-        setErrorData({
+			let sourceHandle = sourceHandleSplitted[0] + '|' + source + '|' + sourceHandleSplitted.slice(2).join('|');
-          title: "Error creating node",
+			let targetHandleSplitted = e.targetHandle.split('|');
-          list: ["There can't be more than one chat input."],
+			let targetHandle = targetHandleSplitted.slice(0, -1).join('|') + '|' + target;
-        });
+			let id = "reactflow__edge-" + source + sourceHandle + "-" + target + targetHandle;
-      }
+			setEdges((eds) =>
-    },
+				addEdge({ source, target, sourceHandle, targetHandle, id, className: "animate-pulse", selected: false }, eds.map((e) => ({ ...e, selected: false })))
-    // Specify dependencies for useCallback
+			);
-    [incrementNodeId, reactFlowInstance, setErrorData, setNodes]
+		})
-  );
+	}
 
-  const onDelete = (mynodes) => {
-    setEdges(
-      edges.filter(
-        (ns) => !mynodes.some((n) => ns.source === n.id || ns.target === n.id)
-      )
-    );
-  };
 
-  const onEdgeUpdateStart = useCallback(() => {
+	const { setExtraComponent, setExtraNavigation } = useContext(locationContext);
-    edgeUpdateSuccessful.current = false;
+	const { setErrorData } = useContext(alertContext);
-  }, []);
+	const [nodes, setNodes, onNodesChange] = useNodesState(
+		flow.data?.nodes ?? []
+	);
+	const [edges, setEdges, onEdgesChange] = useEdgesState(
+		flow.data?.edges ?? []
+	);
+	const { setViewport } = useReactFlow();
+	const edgeUpdateSuccessful = useRef(true);
 
-  const onEdgeUpdate = useCallback(
+	function getId() {
-    (oldEdge: Edge, newConnection: Connection) => {
+		return `dndnode_` + incrementNodeId();
-      if (isValidConnection(newConnection, reactFlowInstance)) {
+	}
-        edgeUpdateSuccessful.current = true;
+
-        setEdges((els) => updateEdge(oldEdge, newConnection, els));
+	useEffect(() => {
-      }
+		if (reactFlowInstance && flow) {
-    },
+			flow.data = reactFlowInstance.toObject();
-    []
+			updateFlow(flow);
-  );
+		}
+		// eslint-disable-next-line react-hooks/exhaustive-deps
+	}, [nodes, edges]);
+	//update flow when tabs change
+	useEffect(() => {
+		setNodes(flow?.data?.nodes ?? []);
+		setEdges(flow?.data?.edges ?? []);
+		if (reactFlowInstance) {
+			setViewport(flow?.data?.viewport ?? { x: 1, y: 0, zoom: 0.5 });
+		}
+	}, [flow, reactFlowInstance, setEdges, setNodes, setViewport]);
+	//set extra sidebar
+	useEffect(() => {
+		setExtraComponent(<ExtraSidebar />);
+		setExtraNavigation({ title: "Components" });
+	}, [setExtraComponent, setExtraNavigation]);
+
+	const onEdgesChangeMod = useCallback(
+		(s: EdgeChange[]) => {
+			onEdgesChange(s);
+			setNodes((x) => {
+				let newX = _.cloneDeep(x);
+				return newX;
+			});
+		},
+		[onEdgesChange, setNodes]
+	);
+
+	const onConnect = useCallback(
+		(params: Connection) => {
+			takeSnapshot();
+			setEdges((eds) =>
+				addEdge({ ...params, className: "animate-pulse" }, eds)
+			);
+			setNodes((x) => {
+				let newX = _.cloneDeep(x);
+				return newX;
+			});
+		},
+		[setEdges, setNodes, takeSnapshot]
+	);
+
+	const onNodeDragStart: NodeDragHandler = useCallback(() => {
+		// 👇 make dragging a node undoable
+		takeSnapshot();
+		// 👉 you can place your event handlers here
+	}, [takeSnapshot]);
+
+	const onSelectionDragStart: SelectionDragHandler = useCallback(() => {
+		// 👇 make dragging a selection undoable
+		takeSnapshot();
+	}, [takeSnapshot]);
+
+
+	const onEdgesDelete: OnEdgesDelete = useCallback(() => {
+		// 👇 make deleting edges undoable
+		takeSnapshot();
+	}, [takeSnapshot]);
+
+	const onDragOver = useCallback((event: React.DragEvent) => {
+		event.preventDefault();
+		event.dataTransfer.dropEffect = "move";
+	}, []);
+
+	const onDrop = useCallback(
+		(event: React.DragEvent) => {
+			event.preventDefault();
+			takeSnapshot();
+
+			// Get the current bounds of the ReactFlow wrapper element
+			const reactflowBounds = reactFlowWrapper.current.getBoundingClientRect();
+
+			// Extract the data from the drag event and parse it as a JSON object
+			let data: { type: string; node?: APIClassType } = JSON.parse(
+				event.dataTransfer.getData("json")
+			);
+
+			// If data type is not "chatInput" or if there are no "chatInputNode" nodes present in the ReactFlow instance, create a new node
+			if (
+				data.type !== "chatInput" ||
+				(data.type === "chatInput" &&
+					!reactFlowInstance.getNodes().some((n) => n.type === "chatInputNode"))
+			) {
+				// Calculate the position where the node should be created
+				const position = reactFlowInstance.project({
+					x: event.clientX - reactflowBounds.left,
+					y: event.clientY - reactflowBounds.top,
+				});
+
+				// Generate a unique node ID
+				let newId = getId();
+
+				// Create a new node object
+				const newNode: NodeType = {
+					id: newId,
+					type: "genericNode",
+					position,
+					data: {
+						...data,
+						id: newId,
+						value: null,
+					},
+				};
+
+				// Add the new node to the list of nodes in state
+				setNodes((nds) => nds.concat(newNode));
+			} else {
+				// If a chat input node already exists, set an error message
+				setErrorData({
+					title: "Error creating node",
+					list: ["There can't be more than one chat input."],
+				});
+			}
+		},
+		// Specify dependencies for useCallback
+		[incrementNodeId, reactFlowInstance, setErrorData, setNodes, takeSnapshot]
+	);
+
+	const onDelete = useCallback((mynodes) => {
+		takeSnapshot();
+		setEdges(
+			edges.filter(
+				(ns) => !mynodes.some((n) => ns.source === n.id || ns.target === n.id)
+			)
+		);
+	}, [takeSnapshot, edges, setEdges]);
+
+	const onEdgeUpdateStart = useCallback(() => {
+		edgeUpdateSuccessful.current = false;
+	}, []);
+
+	const onEdgeUpdate = useCallback(
+		(oldEdge: Edge, newConnection: Connection) => {
+			if (isValidConnection(newConnection, reactFlowInstance)) {
+				edgeUpdateSuccessful.current = true;
+				setEdges((els) => updateEdge(oldEdge, newConnection, els));
+			}
+		},
+		[]
+	);
 
 	const onEdgeUpdateEnd = useCallback((_, edge) => {
 		if (!edgeUpdateSuccessful.current) {
-		  setEdges((eds) => eds.filter((e) => e.id !== edge.id));
+			setEdges((eds) => eds.filter((e) => e.id !== edge.id));
 		}
-	
+
 		edgeUpdateSuccessful.current = true;
-	  }, []);
+	}, []);
-	  
+
 	return (
-		<div className="w-full h-full" ref={reactFlowWrapper}>
+		<div className="w-full h-full" onMouseMove={handleMouseMove} ref={reactFlowWrapper}>
 			{Object.keys(templates).length > 0 && Object.keys(types).length > 0 ? (
 				<>
 					<ReactFlow
@@ -197,6 +307,7 @@ export default function FlowPage({ flow }:{flow:FlowType}) {
 						edges={edges}
 						onNodesChange={onNodesChange}
 						onEdgesChange={onEdgesChangeMod}
+						onKeyDown={(e) => onKeyDown(e)}
 						onConnect={onConnect}
 						onLoad={setReactFlowInstance}
 						onInit={setReactFlowInstance}
@@ -204,13 +315,16 @@ export default function FlowPage({ flow }:{flow:FlowType}) {
 						onEdgeUpdate={onEdgeUpdate}
 						onEdgeUpdateStart={onEdgeUpdateStart}
 						onEdgeUpdateEnd={onEdgeUpdateEnd}
+						onNodeDragStart={onNodeDragStart}
+						onSelectionDragStart={onSelectionDragStart}
+						onEdgesDelete={onEdgesDelete}
 						connectionLineComponent={ConnectionLineComponent}
 						onDragOver={onDragOver}
 						onDrop={onDrop}
 						onNodesDelete={onDelete}
-            selectNodesOnDrag={false}
+						selectNodesOnDrag={false}
 					>
-						<Background className="dark:bg-gray-900"/>
+						<Background className="dark:bg-gray-900" />
 						<Controls className="[&>button]:text-black  [&>button]:dark:bg-gray-800 hover:[&>button]:dark:bg-gray-700 [&>button]:dark:text-gray-400 [&>button]:dark:fill-gray-400 [&>button]:dark:border-gray-600">
 						</Controls>
 					</ReactFlow>

src/frontend/src/reportWebVitals.ts

@@ -1,15 +1,15 @@
 import { ReportHandler } from "web-vitals";
 
 const reportWebVitals = (onPerfEntry?: ReportHandler) => {
-  if (onPerfEntry && onPerfEntry instanceof Function) {
+	if (onPerfEntry && onPerfEntry instanceof Function) {
-    import("web-vitals").then(({ getCLS, getFID, getFCP, getLCP, getTTFB }) => {
+		import("web-vitals").then(({ getCLS, getFID, getFCP, getLCP, getTTFB }) => {
-      getCLS(onPerfEntry);
+			getCLS(onPerfEntry);
-      getFID(onPerfEntry);
+			getFID(onPerfEntry);
-      getFCP(onPerfEntry);
+			getFCP(onPerfEntry);
-      getLCP(onPerfEntry);
+			getLCP(onPerfEntry);
-      getTTFB(onPerfEntry);
+			getTTFB(onPerfEntry);
-    });
+		});
-  }
+	}
 };
 
 export default reportWebVitals;

src/frontend/src/svg.d.ts

@@ -1,4 +1,4 @@
 declare module "*.svg" {
-  const content: any;
+	const content: any;
-  export default content;
+	export default content;
 }

src/frontend/src/types/alerts/index.ts

@@ -1,29 +1,29 @@
 export type ErrorAlertType = {
-  title: string;
+	title: string;
-  list: Array<string>;
+	list: Array<string>;
-  id: string;
+	id: string;
-  removeAlert: (id: string) => void;
+	removeAlert: (id: string) => void;
 };
 export type NoticeAlertType = {
-  title: string;
+	title: string;
-  link: string;
+	link: string;
-  id: string;
+	id: string;
-  removeAlert: (id: string) => void;
+	removeAlert: (id: string) => void;
 };
 export type SuccessAlertType = {
-  title: string;
+	title: string;
-  id: string;
+	id: string;
-  removeAlert: (id: string) => void;
+	removeAlert: (id: string) => void;
 };
 export type SingleAlertComponentType = {
-  dropItem: AlertItemType;
+	dropItem: AlertItemType;
-  removeAlert: (index: string) => void;
+	removeAlert: (index: string) => void;
 };
 export type AlertDropdownType = {};
 export type AlertItemType = {
-  type: "notice" | "error" | "success";
+	type: "notice" | "error" | "success";
-  title: string;
+	title: string;
-  link?: string;
+	link?: string;
-  list?: Array<string>;
+	list?: Array<string>;
-  id: string;
+	id: string;
 };

src/frontend/src/types/api/index.ts

@@ -4,37 +4,37 @@ import { Node, Edge, Viewport } from "reactflow";
 export type APIObjectType = { kind: APIKindType; [key: string]: APIKindType };
 export type APIKindType = { class: APIClassType; [key: string]: APIClassType };
 export type APITemplateType = {
-  variable: TemplateVariableType;
+	variable: TemplateVariableType;
-  [key: string]: TemplateVariableType;
+	[key: string]: TemplateVariableType;
 };
 export type APIClassType = {
-  base_classes: Array<string>;
+	base_classes: Array<string>;
-  description: string;
+	description: string;
-  template: APITemplateType;
+	template: APITemplateType;
-  [key: string]: Array<string> | string | APITemplateType;
+	[key: string]: Array<string> | string | APITemplateType;
 };
 export type TemplateVariableType = {
-  type: string;
+	type: string;
-  required: boolean;
+	required: boolean;
-  placeholder?: string;
+	placeholder?: string;
-  list: boolean;
+	list: boolean;
-  show: boolean;
+	show: boolean;
-  multiline?: boolean;
+	multiline?: boolean;
-  value?: any;
+	value?: any;
-  [key: string]: any;
+	[key: string]: any;
 };
 export type sendAllProps = {
-  nodes: Node[];
+	nodes: Node[];
-  edges: Edge[];
+	edges: Edge[];
-  name: string;
+	name: string;
-  description: string;
+	description: string;
-  viewport: Viewport;
+	viewport: Viewport;
-  message: string;
+	message: string;
 
-  chatHistory: { message: string; isSend: boolean }[];
+	chatHistory: { message: string; isSend: boolean }[];
 };
 export type errorsTypeAPI = {
-  function: { errors: Array<string> };
+	function: { errors: Array<string> };
-  imports: { errors: Array<string> };
+	imports: { errors: Array<string> };
 };
 export type PromptTypeAPI = { input_variables: Array<string> };

src/frontend/src/types/chat/index.ts

@@ -3,8 +3,8 @@ import { FlowType } from "../flow";
 
 export type ChatType = { flow: FlowType; reactFlowInstance: ReactFlowInstance };
 export type ChatMessageType = {
-  message: string;
+	message: string;
-  isSend: boolean;
+	isSend: boolean;
-  thought?: string;
+	thought?: string;
-  files?: Array<{ data: string; type: string; data_type: string }>;
+	files?: Array<{ data: string; type: string; data_type: string }>;
 };

src/frontend/src/types/components/index.ts

@@ -1,69 +1,69 @@
 import { ForwardRefExoticComponent, ReactElement, ReactFragment, ReactNode } from "react";
 import { NodeDataType } from "../flow/index";
 export type InputComponentType = {
-  value: string;
+	value: string;
-  disabled?: boolean;
+	disabled?: boolean;
-  onChange: (value: string) => void;
+	onChange: (value: string) => void;
-  password: boolean;
+	password: boolean;
 };
 export type ToggleComponentType = {
-  enabled: boolean;
+	enabled: boolean;
-  setEnabled: (state: boolean) => void;
+	setEnabled: (state: boolean) => void;
-  disabled: boolean;
+	disabled: boolean;
 };
 export type DropDownComponentType = {
-  value: string;
+	value: string;
-  options: string[];
+	options: string[];
-  onSelect: (value: string) => void;
+	onSelect: (value: string) => void;
 };
 export type ParameterComponentType = {
-  data: NodeDataType;
+	data: NodeDataType;
-  title: string;
+	title: string;
-  id: string;
+	id: string;
-  color: string;
+	color: string;
-  left: boolean;
+	left: boolean;
-  type: string;
+	type: string;
-  required?: boolean;
+	required?: boolean;
-  name?: string;
+	name?: string;
-  tooltipTitle: string;
+	tooltipTitle: string;
 };
 export type InputListComponentType = {
-  value: string[];
+	value: string[];
-  onChange: (value: string[]) => void;
+	onChange: (value: string[]) => void;
-  disabled: boolean;
+	disabled: boolean;
 };
 
 export type TextAreaComponentType = {
-  disabled: boolean;
+	disabled: boolean;
-  onChange: (value: string[] | string) => void;
+	onChange: (value: string[] | string) => void;
-  value: string;
+	value: string;
 };
 
 export type FileComponentType = {
-  disabled: boolean;
+	disabled: boolean;
-  onChange: (value: string[] | string) => void;
+	onChange: (value: string[] | string) => void;
-  value: string;
+	value: string;
-  suffixes: Array<string>;
+	suffixes: Array<string>;
-  fileTypes: Array<string>;
+	fileTypes: Array<string>;
-  onFileChange: (value: string) => void;
+	onFileChange: (value: string) => void;
 };
 
 export type DisclosureComponentType = {
-  children: ReactNode;
+	children: ReactNode;
-  button: {
+	button: {
-    title: string;
+		title: string;
-    Icon: ForwardRefExoticComponent<React.SVGProps<SVGSVGElement>>;
+		Icon: ForwardRefExoticComponent<React.SVGProps<SVGSVGElement>>;
-    buttons?: {
+		buttons?: {
-      Icon: ReactElement;
+			Icon: ReactElement;
-      title: string;
+			title: string;
-      onClick: (event?: React.MouseEvent) => void;
+			onClick: (event?: React.MouseEvent) => void;
-    }[];
+		}[];
-  };
+	};
 };
 export type FloatComponentType = {
-  value: string;
+	value: string;
-  disabled?: boolean;
+	disabled?: boolean;
-  onChange: (value: string) => void;
+	onChange: (value: string) => void;
 };
 
 export type TooltipComponentType = {

src/frontend/src/types/entities/index.ts

@@ -1,8 +1,8 @@
 import { HomeIcon } from "@heroicons/react/24/outline";
 
 export type sidebarNavigationItemType = {
-  name: string;
+	name: string;
-  href: string;
+	href: string;
-  icon: React.ForwardRefExoticComponent<React.SVGProps<SVGSVGElement>>;
+	icon: React.ForwardRefExoticComponent<React.SVGProps<SVGSVGElement>>;
-  current: boolean;
+	current: boolean;
 };

src/frontend/src/types/flow/index.ts

@@ -3,20 +3,20 @@ import { APIClassType } from "../api/index";
 import { ReactFlowJsonObject, XYPosition } from "reactflow";
 
 export type FlowType = {
-  name: string;
+	name: string;
-  id: string;
+	id: string;
-  data: ReactFlowJsonObject;
+	data: ReactFlowJsonObject;
-  description: string;
+	description: string;
 };
 export type NodeType = {
-  id: string;
+	id: string;
-  type?: string;
+	type?: string;
-  position: XYPosition;
+	position: XYPosition;
-  data: NodeDataType;
+	data: NodeDataType;
 };
 export type NodeDataType = {
-  type: string;
+	type: string;
-  node?: APIClassType;
+	node?: APIClassType;
-  id: string;
+	id: string;
-  value: any;
+	value: any;
 };

src/frontend/src/types/tabs/index.ts

@@ -1,17 +1,22 @@
 import { FlowType } from "../flow";
 
 export type TabsContextType = {
-  save: () => void;
+	save: () => void;
-  tabIndex: number;
+	tabIndex: number;
-  setTabIndex: (index: number) => void;
+	setTabIndex: (index: number) => void;
-  flows: Array<FlowType>;
+	flows: Array<FlowType>;
-  removeFlow: (id: string) => void;
+	removeFlow: (id: string) => void;
-  addFlow: (flowData?: FlowType) => void;
+	addFlow: (flowData?: FlowType) => void;
-  updateFlow: (newFlow: FlowType) => void;
+	updateFlow: (newFlow: FlowType) => void;
-  incrementNodeId: () => number;
+	incrementNodeId: () => number;
-  downloadFlow: (flow: FlowType) => void;
+	downloadFlow: (flow: FlowType) => void;
-  uploadFlow: () => void;
+	uploadFlow: () => void;
-  hardReset: () => void;
+	hardReset: () => void;
 };
 
-export type LangFlowState={ tabIndex:number, flows:FlowType[], id:string, nodeId:number }
+export type LangFlowState = {
+	tabIndex: number;
+	flows: FlowType[];
+	id: string;
+	nodeId: number;
+};

src/frontend/src/types/templatesContext/index.ts

@@ -1,7 +1,6 @@
-
+const template: { [char: string]: string } = {};
-const template:{[char: string]: string}={}
 
 export type TemplateContextType = {
 	templates: typeof template;
 	setTemplates: (newState: {}) => void;
-};
+};

src/frontend/src/types/typesContext/index.ts

@@ -1,12 +1,11 @@
 import { ReactFlowInstance } from "reactflow";
 
-const types:{[char: string]: string}={};
+const types: { [char: string]: string } = {};
-const template:{[char: string]: string}={}
+const template: { [char: string]: string } = {};
-const data:{[char: string]: string}={}
+const data: { [char: string]: string } = {};
-
 
 export type typesContextType = {
-	reactFlowInstance: ReactFlowInstance|null;
+	reactFlowInstance: ReactFlowInstance | null;
 	setReactFlowInstance: any;
 	deleteNode: (idx: string) => void;
 	types: typeof types;

src/frontend/src/utils.ts

@@ -317,24 +317,25 @@ export function toFirstUpperCase(str: string) {
 }
 
 export function toNormalCase(str: string) {
-  let result =  str
+	let result = str
-    .split("_")
+		.split("_")
-    .map((word, index) => {
+		.map((word, index) => {
-      if (index === 0) {
+			if (index === 0) {
-        return word[0].toUpperCase() + word.slice(1).toLowerCase();
+				return word[0].toUpperCase() + word.slice(1).toLowerCase();
-      }
+			}
-      return word.toLowerCase();
+			return word.toLowerCase();
-    })
+		})
-    .join(" ");
+		.join(" ");
 
-  return result.split("-")
+	return result
-  .map((word, index) => {
+		.split("-")
-    if (index === 0) {
+		.map((word, index) => {
-      return word[0].toUpperCase() + word.slice(1).toLowerCase();
+			if (index === 0) {
-    }
+				return word[0].toUpperCase() + word.slice(1).toLowerCase();
-    return word.toLowerCase();
+			}
-  })
+			return word.toLowerCase();
-  .join(" ");
+		})
+		.join(" ");
 }
 
 export function normalCaseToSnakeCase(str: string) {
@@ -397,15 +398,15 @@ export function isValidConnection(
 }
 
 export function removeApiKeys(flow: FlowType): FlowType {
-  let cleanFLow = _.cloneDeep(flow);
+	let cleanFLow = _.cloneDeep(flow);
-  cleanFLow.data.nodes.forEach((node) => {
+	cleanFLow.data.nodes.forEach((node) => {
-    for (const key in node.data.node.template) {
+		for (const key in node.data.node.template) {
-      if (key.includes("api")) {
+			if (key.includes("api")) {
-        node.data.node.template[key].value = "";
+				node.data.node.template[key].value = "";
-      }
+			}
-    }
+		}
-  });
+	});
-  return cleanFLow;
+	return cleanFLow;
 }
 
 export function updateObject<T extends Record<string, any>>(
@@ -443,7 +444,7 @@ export function updateTemplate(
 	reference: APITemplateType,
 	objectToUpdate: APITemplateType
 ): APITemplateType {
-	let clonedObject:APITemplateType = _.cloneDeep(reference);
+	let clonedObject: APITemplateType = _.cloneDeep(reference);
 
 	// Loop through each key in the reference object
 	for (const key in clonedObject) {
@@ -454,3 +455,34 @@ export function updateTemplate(
 	}
 	return clonedObject;
 }
+
+interface languageMap {
+	[key: string]: string | undefined;
+}
+
+export const programmingLanguages: languageMap = {
+	javascript: ".js",
+	python: ".py",
+	java: ".java",
+	c: ".c",
+	cpp: ".cpp",
+	"c++": ".cpp",
+	"c#": ".cs",
+	ruby: ".rb",
+	php: ".php",
+	swift: ".swift",
+	"objective-c": ".m",
+	kotlin: ".kt",
+	typescript: ".ts",
+	go: ".go",
+	perl: ".pl",
+	rust: ".rs",
+	scala: ".scala",
+	haskell: ".hs",
+	lua: ".lua",
+	shell: ".sh",
+	sql: ".sql",
+	html: ".html",
+	css: ".css",
+	// add more file extensions here, make sure the key is same as language prop in CodeBlock.tsx component
+};

src/frontend/tailwind.config.js

@@ -73,6 +73,6 @@ module.exports = {
         }
       }
     })
-    }),require('@tailwindcss/line-clamp')
+    }),require('@tailwindcss/line-clamp'),require('@tailwindcss/typography'),
   ],
 };

tests/test_graph.py

@@ -14,7 +14,7 @@ from langflow.graph.nodes import (
     ToolNode,
     WrapperNode,
 )
-from langflow.interface.run import get_result_and_steps
+from langflow.interface.run import get_result_and_thought
 from langflow.utils.payload import get_root_node
 
 # Test cases for the graph module
@@ -335,7 +335,7 @@ def test_get_result_and_thought(basic_graph):
     # now build again and check if FakeListLLM was used
 
     # Get the result and thought
-    result, thought = get_result_and_steps(langchain_object, message)
+    result, thought = get_result_and_thought(langchain_object, message)
     # The result should be a str
     assert isinstance(result, str)
     # The thought should be a Thought