Merge branch 'logspace-ai:dev' into dev

Dockerfile

@@ -1,59 +1,14 @@
-# FROM logspace/backend_build as backend_build
+FROM python:3.10-slim
-FROM logspace/frontend_build as frontend_build
 
-# `python-base` sets up all our shared environment variables
+RUN apt-get update && apt-get install gcc g++ git make -y
-FROM python:3.10-slim as langflow_build
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
-# python
+WORKDIR $HOME/app
-ENV PYTHONUNBUFFERED=1 \
-    # prevents python creating .pyc files
-    PYTHONDONTWRITEBYTECODE=1 \
-    \
-    # pip
-    PIP_NO_CACHE_DIR=off \
-    PIP_DISABLE_PIP_VERSION_CHECK=on \
-    PIP_DEFAULT_TIMEOUT=100 \
-    \
-    # poetry
-    # https://python-poetry.org/docs/configuration/#using-environment-variables
-    POETRY_VERSION=1.4.0 \
-    # make poetry install to this location
-    POETRY_HOME="/opt/poetry" \
-    # make poetry create the virtual environment in the project's root
-    # it gets named `.venv`
-    POETRY_VIRTUALENVS_IN_PROJECT=true \
-    # do not ask any interactive question
-    POETRY_NO_INTERACTION=1 \
-    \
-    # paths
-    # this is where our requirements + virtual environment will live
-    PYSETUP_PATH="/opt/pysetup" \
-    VENV_PATH="/opt/pysetup/.venv"
 
-# prepend poetry and venv to path
+COPY --chown=user . $HOME/app
-ENV PATH="$POETRY_HOME/bin:$VENV_PATH/bin:$PATH"
 
-RUN apt-get update \
+RUN pip install langflow>==0.0.71 -U --user
-    && apt-get install --no-install-recommends -y \
+CMD ["langflow", "--host", "0.0.0.0", "--port", "7860"]
-    # deps for installing poetry
-    curl \
-    # deps for building python deps
-    build-essential libpq-dev git
-
-# install poetry - respects $POETRY_VERSION & $POETRY_HOME
-# RUN curl -sSL https://install.python-poetry.org | python3 -
-
-# copy project requirement files here to ensure they will be cached.
-WORKDIR /app
-COPY pyproject.toml ./
-# copy langflow
-COPY ./langflow ./langflow
-
-# Copy files from frontend
-COPY --from=frontend_build /app/build /app/src/frontend/build/
-
-RUN pip install .
-
-EXPOSE 5003
-
-CMD [ "langflow" ]

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langflow"
-version = "0.0.68"
+version = "0.0.71"
 description = "A Python package with a built-in web application"
 authors = ["Logspace <contact@logspace.ai>"]
 maintainers = [
@@ -29,7 +29,7 @@ google-search-results = "^2.4.1"
 google-api-python-client = "^2.79.0"
 typer = "^0.7.0"
 gunicorn = "^20.1.0"
-langchain = "~0.0.150"
+langchain = "^0.0.166"
 openai = "^0.27.2"
 types-pyyaml = "^6.0.12.8"
 dill = "^0.3.6"
@@ -50,6 +50,7 @@ pyarrow = "^11.0.0"
 websockets = "^11.0.2"
 tiktoken = "^0.3.3"
 wikipedia = "^1.4.0"
+gptcache = "^0.1.23"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.1.0"

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,16 +1,15 @@
 import logging
-from typing import Any, Dict
 
 from fastapi import APIRouter, HTTPException
 
-from langflow.interface.run import process_graph_cached
-from langflow.interface.types import build_langchain_types_dict
 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
 
 # build router
 router = APIRouter()

src/backend/langflow/api/schemas.py

@@ -1,4 +1,4 @@
-from typing import Any, Union, Dict, List
+from typing import Any, Dict, List, Union
 
 from pydantic import BaseModel, validator
 

src/backend/langflow/api/validate.py

@@ -1,3 +1,5 @@
+import json
+
 from fastapi import APIRouter, HTTPException
 
 from langflow.api.base import (
@@ -7,6 +9,7 @@ from langflow.api.base import (
     PromptValidationResponse,
     validate_prompt,
 )
+from langflow.graph.nodes import VectorStoreNode
 from langflow.interface.run import build_graph
 from langflow.utils.logger import logger
 from langflow.utils.validate import validate_code
@@ -44,10 +47,11 @@ def post_validate_node(node_id: str, data: dict):
         graph = build_graph(data)
         # validate node
         node = graph.get_node(node_id)
-        if node is not None:
+        if node is None:
-            _ = node.build()
+            raise ValueError(f"Node {node_id} not found")
-            return str(node.params)
+        if not isinstance(node, VectorStoreNode):
-        raise Exception(f"Node {node_id} not found")
+            node.build()
+        return json.dumps({"valid": True, "params": str(node._built_object_repr())})
     except Exception as e:
-        logger.error(e)
+        logger.exception(e)
-        raise HTTPException(status_code=500, detail=str(e)) from e
+        return json.dumps({"valid": False})

src/backend/langflow/graph/base.py

@@ -232,6 +232,9 @@ class Node:
     def __hash__(self) -> int:
         return id(self)
 
+    def _built_object_repr(self):
+        return repr(self._built_object)
+
 
 class Edge:
     def __init__(self, source: "Node", target: "Node"):

src/backend/langflow/graph/nodes.py

@@ -139,6 +139,13 @@ class DocumentLoaderNode(Node):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="documentloaders")
 
+    def _built_object_repr(self):
+        # This built_object is a list of documents. Maybe we should
+        # show how many documents are in the list?
+        if self._built_object:
+            return f"""{self.node_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
+        return f"{self.node_type}()"
+
 
 class EmbeddingNode(Node):
     def __init__(self, data: Dict):
@@ -149,6 +156,9 @@ class VectorStoreNode(Node):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="vectorstores")
 
+    def _built_object_repr(self):
+        return "Vector stores can take time to build. It will build on the first query."
+
 
 class MemoryNode(Node):
     def __init__(self, data: Dict):
@@ -158,3 +168,10 @@ class MemoryNode(Node):
 class TextSplitterNode(Node):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="textsplitters")
+
+    def _built_object_repr(self):
+        # This built_object is a list of documents. Maybe we should
+        # show how many documents are in the list?
+        if self._built_object:
+            return f"""{self.node_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
+        return f"{self.node_type}()"

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,55 +30,84 @@ 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}
 
+def convert_params_to_sets(params):
+    """Convert certain params to sets"""
+    if "allowed_special" in params:
+        params["allowed_special"] = set(params["allowed_special"])
+    if "disallowed_special" in params:
+        params["disallowed_special"] = set(params["disallowed_special"])
+    return params
+
+
+def instantiate_based_on_type(class_object, base_type, node_type, params):
     if base_type == "agents":
-        # We need to initialize it differently
+        return instantiate_agent(class_object, params)
-        return load_agent_executor(class_object, params)
     elif base_type == "prompts":
+        return instantiate_prompt(class_object, 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(class_object, node_type, params):
     if node_type == "ZeroShotPrompt":
         if "tools" not in params:
             params["tools"] = []
         return ZeroShotAgent.create_prompt(**params)
-    elif base_type == "tools":
+    return class_object(**params)
+
+
+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":
-            # 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":
+    return None  # Or some other default action
+
+
+def instantiate_toolkit(node_type, class_object, params):
     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:
+def instantiate_embedding(class_object, params):
-            params.pop("model")
+    params.pop("model", None)
-        except KeyError:
-            pass
-        # remove all params that are not in class_object.__fields__
     try:
         return class_object(**params)
     except ValidationError:
@@ -88,37 +117,33 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
             if key in class_object.__fields__
         }
         return class_object(**params)
-    elif base_type == "vectorstores":
+
+
+def instantiate_vectorstore(class_object, params):
     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":
+
+
+def instantiate_documentloader(class_object, params):
     return class_object(**params).load()
-    elif base_type == "textsplitters":
+
+
+def instantiate_textsplitter(class_object, params):
     documents = params.pop("documents")
     text_splitter = class_object(**params)
     return text_splitter.split_documents(documents)
-    elif base_type == "utilities":
+
+
+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 process_params(params):
-    """Process params"""
-    if "allowed_special" in params:
-        params["allowed_special"] = set(params["allowed_special"])
-    if "disallowed_special" in params:
-        params["disallowed_special"] = set(params["disallowed_special"])
-    return 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

@@ -1,13 +1,14 @@
 import contextlib
 import io
-from typing import Any, Dict
+from typing import Any, Dict, List, Tuple
 
 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
+from langchain.schema import AgentAction
 
 
 def load_langchain_object(data_graph, is_first_message=False):
@@ -66,40 +67,6 @@ def build_langchain_object(data_graph):
     return graph.build()
 
 
-def process_graph(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
-    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], message: str):
     """
     Process graph by extracting input variables and replacing ZeroShotPrompt
@@ -184,8 +151,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
@@ -205,17 +173,17 @@ 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 []
@@ -226,11 +194,7 @@ def get_result_and_steps(langchain_object, message: str):
             if isinstance(output, dict)
             else output
         )
-            if intermediate_steps:
+        thought = format_actions(intermediate_steps) if intermediate_steps else ""
-                thought = format_intermediate_steps(intermediate_steps)
-            else:
-                thought = output_buffer.getvalue()
-
     except NotEnoughElementsException as exc:
         raise ValueError(
             "Error: Not enough documents for ChromaDB to index. Try reducing chunk size in TextSplitter."
@@ -286,7 +250,7 @@ def get_result_and_thought(langchain_object, message: str):
                 else output
             )
             if intermediate_steps:
-                thought = format_intermediate_steps(intermediate_steps)
+                thought = format_actions(intermediate_steps)
             else:
                 thought = output_buffer.getvalue()
 
@@ -295,19 +259,17 @@ def get_result_and_thought(langchain_object, message: str):
     return result, thought
 
 
-def format_intermediate_steps(intermediate_steps):
+def format_actions(actions: List[Tuple[AgentAction, str]]) -> str:
-    formatted_chain = "> Entering new AgentExecutor chain...\n"
+    """Format a list of (AgentAction, answer) tuples into a string."""
-    for step in intermediate_steps:
+    output = []
-        action = step[0]
+    for action, answer in actions:
-        observation = step[1]
+        log = action.log
-
+        tool = action.tool
-        formatted_chain += (
+        tool_input = action.tool_input
-            f" {action.log}\nAction: {action.tool}\nAction Input: {action.tool_input}\n"
+        output.append(f"Log: {log}")
-        )
+        if "Action" not in log and "Action Input" not in log:
-        formatted_chain += f"Observation: {observation}\n"
+            output.append(f"Tool: {tool}")
-
+            output.append(f"Tool Input: {tool_input}")
-    final_answer = f"Final Answer: {observation}\n"
+        output.append(f"Answer: {answer}")
-    formatted_chain += f"Thought: I now know the final answer\n{final_answer}\n"
+        output.append("")  # Add a blank line
-    formatted_chain += "> Finished chain.\n"
+    return "\n".join(output)
-
-    return formatted_chain

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/index.html

@@ -4,12 +4,12 @@
     <meta charset="UTF-8">
     <meta http-equiv="X-UA-Compatible" content="IE=edge">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <link rel="icon" href="%PUBLIC_URL%/favicon.ico" />
+    <link rel="icon" href="/favicon.ico" />
     <title>LangFlow</title>
 </head>
 <body id='body' style="width: 100%; height:100%">
     <noscript>You need to enable JavaScript to run this app.</noscript>
-    <div style="width: 100vw; height:100vh" id='root'>
+    <div style="width: 100vw; height:100vh" id='root'></div>
-    </div>
+    <script type="module" src="/src/index.tsx"></script>
 </body>
 </html>

src/frontend/package.json

@@ -8,15 +8,9 @@
         "@headlessui/react": "^1.7.10",
         "@heroicons/react": "^2.0.15",
         "@mui/material": "^5.11.9",
+        "@tabler/icons-react": "^2.18.0",
         "@tailwindcss/forms": "^0.5.3",
         "@tailwindcss/line-clamp": "^0.4.4",
-        "@testing-library/jest-dom": "^5.16.5",
-        "@testing-library/react": "^13.4.0",
-        "@testing-library/user-event": "^13.5.0",
-        "@types/jest": "^27.5.2",
-        "@types/node": "^16.18.12",
-        "@types/react": "^18.0.27",
-        "@types/react-dom": "^18.0.10",
         "ace-builds": "^1.16.0",
         "ansi-to-html": "^0.7.2",
         "axios": "^1.3.2",
@@ -29,19 +23,21 @@
         "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-tabs": "^6.0.0",
         "reactflow": "^11.5.5",
-        "tailwindcss": "^3.2.6",
+        "rehype-mathjax": "^4.0.2",
-        "typescript": "^4.9.5",
+        "remark-gfm": "^3.0.1",
+        "remark-math": "^5.1.1",
+        "uuid": "^9.0.0",
         "web-vitals": "^2.1.4"
     },
     "scripts": {
-        "start": "react-scripts start",
+        "start": "vite",
-        "build": "react-scripts build",
+        "build": "vite build",
-        "test": "react-scripts test",
+        "serve": "vite preview"
-        "eject": "react-scripts eject"
     },
     "eslintConfig": {
         "extends": [
@@ -61,5 +57,23 @@
             "last 1 safari version"
         ]
     },
-    "proxy": "http://127.0.0.1:7860"
+    "proxy": "http://127.0.0.1:7860",
+    "devDependencies": {
+        "@tailwindcss/typography": "^0.5.9",
+        "@testing-library/jest-dom": "^5.16.5",
+        "@testing-library/react": "^13.4.0",
+        "@testing-library/user-event": "^13.5.0",
+        "@types/jest": "^27.5.2",
+        "@types/lodash": "^4.14.194",
+        "@types/node": "^16.18.12",
+        "@types/react": "^18.0.28",
+        "@types/react-dom": "^18.0.11",
+        "@types/uuid": "^9.0.1",
+        "@vitejs/plugin-react-swc": "^3.0.0",
+        "autoprefixer": "^10.4.14",
+        "postcss": "^8.4.23",
+        "tailwindcss": "^3.3.2",
+        "typescript": "^5.0.2",
+        "vite": "^4.3.5"
+    }
 }

src/frontend/postcss.config.js

@@ -0,0 +1,6 @@
+module.exports = {
+  plugins: {
+    tailwindcss: {},
+    autoprefixer: {},
+  },
+}

src/frontend/src/App.tsx

@@ -2,6 +2,8 @@ import "reactflow/dist/style.css";
 import { useState, useEffect, useContext } from "react";
 import "./App.css";
 import { useLocation } from "react-router-dom";
+import _ from "lodash";
+
 import ErrorAlert from "./alerts/error";
 import NoticeAlert from "./alerts/notice";
 import SuccessAlert from "./alerts/success";
@@ -14,7 +16,6 @@ import CrashErrorComponent from "./components/CrashErrorComponent";
 import { TabsContext } from "./contexts/tabsContext";
 
 export default function App() {
-	var _ = require("lodash");
 
 	let { setCurrent, setShowSideBar, setIsStackedOpen } =
 		useContext(locationContext);

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,13 +1,8 @@
-import { Cog6ToothIcon, TrashIcon } from "@heroicons/react/24/outline";
+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 } from "react";
+import { useContext, useState, useEffect, useRef, Fragment } from "react";
 import { NodeDataType } from "../../types/flow";
 import { alertContext } from "../../contexts/alertContext";
 import { PopUpContext } from "../../contexts/popUpContext";
@@ -15,6 +10,7 @@ import NodeModal from "../../modals/NodeModal";
 import { useCallback } from "react";
 import { TabsContext } from "../../contexts/tabsContext";
 import { debounce } from "../../utils";
+import Tooltip from "../../components/TooltipComponent";
 export default function GenericNode({
 	data,
 	selected,
@@ -27,14 +23,13 @@ export default function GenericNode({
   const { types, deleteNode } = useContext(typesContext);
   const { openPopUp } = useContext(PopUpContext);
   const Icon = nodeIcons[types[data.type]];
-  const [validationStatus, setValidationStatus] = useState("idle");
+  const [validationStatus, setValidationStatus] = useState(null);
   // State for outline color
   const [isValid, setIsValid] = useState(false);
   const { save } = useContext(TabsContext);
   const { reactFlowInstance } = useContext(typesContext);
   const [params, setParams] = useState([]);
 
-
 	useEffect(() => {
 		if (reactFlowInstance) {
 			setParams(Object.values(reactFlowInstance.toObject()));
@@ -53,10 +48,15 @@ export default function GenericNode({
 				});
 
         if (response.status === 200) {
-          setValidationStatus("success");
+          let jsonResponse = await response.json();
-        } else if (response.status === 500) {
+          let jsonResponseParsed = await JSON.parse(jsonResponse);
+          console.log(jsonResponseParsed);
+          if(jsonResponseParsed.valid){
+            setValidationStatus(jsonResponseParsed.params);
+          } else {
             setValidationStatus("error");
           }
+        }
       } catch (error) {
         // console.error("Error validating node:", error);
         setValidationStatus("error");
@@ -71,7 +71,7 @@ export default function GenericNode({
   }, [params, validateNode]);
 
   useEffect(() => {
-    if (validationStatus === "success") {
+    if (validationStatus !== "error") {
       setIsValid(true);
     } else {
       setIsValid(false);
@@ -100,14 +100,23 @@ export default function GenericNode({
       )}
     >
       <div className="w-full dark:text-white flex items-center justify-between p-4 gap-8 bg-gray-50 rounded-t-lg dark:bg-gray-800 border-b dark:border-b-gray-700 ">
-        <div className="w-full flex items-center truncate gap-4 text-lg">
+        <div className="w-full flex items-center truncate gap-2 text-lg">
           <Icon
             className="w-10 h-10 p-1 rounded"
             style={{
               color: nodeColors[types[data.type]] ?? nodeColors.unknown,
             }}
           />
-          <div className="truncate">{data.type}</div>
+          <div className="ml-2 truncate">{data.type}</div>
+          {validationStatus && validationStatus !== "error" ? 
+          <Tooltip title={
+              <div className="max-h-96 overflow-auto">
+                {validationStatus}
+              </div>}>
+            <ExclamationCircleIcon className="w-5 hover:text-gray-500 hover:dark:text-gray-300" /> 
+          </Tooltip>
+          : <></>
+}
         </div>
         <div className="flex gap-3">
           <button
@@ -134,7 +143,7 @@ export default function GenericNode({
                 )
                   ? ""
                   : "hidden",
-                "w-6 h-6  dark:text-gray-500  hover:animate-spin"
+                "w-6 h-6  dark:text-gray-300  hover:animate-spin"
               )}
             ></Cog6ToothIcon>
           </button>
@@ -143,13 +152,13 @@ export default function GenericNode({
               deleteNode(data.id);
             }}
           >
-            <TrashIcon className="w-6 h-6 hover:text-red-500 dark:text-gray-500 dark:hover:text-red-500"></TrashIcon>
+            <TrashIcon className="w-6 h-6 hover:text-red-500 dark:text-gray-300 dark:hover:text-red-500"></TrashIcon>
           </button>
         </div>
       </div>
 
       <div className="w-full h-full py-5">
-        <div className="w-full text-gray-500 px-5 pb-3 text-sm">
+        <div className="w-full text-gray-500 dark:text-gray-300 px-5 pb-3 text-sm">
           {data.node.description}
         </div>
 

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

@@ -30,21 +30,21 @@ export default function SingleAlert({
 		>
 			{type === "error" ? (
 				<div
-          className="flex bg-red-50 rounded-md p-3 mb-2 mx-2"
+					className="flex bg-red-50 dark:bg-red-900 rounded-md p-3 mb-2 mx-2"
 					key={dropItem.id}
 				>
 					<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 dark:text-red-50" aria-hidden="true" />
 					</div>
 					<div className="ml-3">
-            <h3 className="text-sm font-medium text-red-800">
+						<h3 className="text-sm break-words font-medium text-red-800 dark:text-white/80">
 							{dropItem.title}
 						</h3>
 						{dropItem.list ? (
-              <div className="mt-2 text-sm text-red-700">
+							<div className="mt-2 text-sm text-red-700 dark:text-red-50">
 								<ul className="list-disc space-y-1 pl-5">
 									{dropItem.list.map((item, idx) => (
-                    <li key={idx}>{item}</li>
+										<li className="break-words" key={idx}>{item}</li>
 									))}
 								</ul>
 							</div>
@@ -62,7 +62,7 @@ export default function SingleAlert({
 										removeAlert(dropItem.id);
 									}, 500);
 								}}
-                className="inline-flex rounded-md bg-red-50 p-1.5 text-red-500"
+								className="inline-flex rounded-md bg-red-50 dark:bg-transparent p-1.5 text-red-500 dark:text-red-50"
 							>
 								<span className="sr-only">Dismiss</span>
 								<XMarkIcon className="h-5 w-5" aria-hidden="true" />
@@ -72,22 +72,22 @@ export default function SingleAlert({
 				</div>
 			) : type === "notice" ? (
 				<div
-          className="flex rounded-md bg-blue-50 p-3 mb-2 mx-2"
+					className="flex rounded-md bg-blue-50 dark:bg-blue-900 p-3 mb-2 mx-2"
 					key={dropItem.id}
 				>
 					<div className="flex-shrink-0">
 						<InformationCircleIcon
-              className="h-5 w-5 text-blue-400"
+							className="h-5 w-5 text-blue-400 dark:text-blue-50"
 							aria-hidden="true"
 						/>
 					</div>
 					<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 dark:text-white/80">{dropItem.title}</p>
 						<p className="mt-3 text-sm md:mt-0 md:ml-6">
 							{dropItem.link ? (
 								<Link
 									to={dropItem.link}
-                  className="whitespace-nowrap font-medium text-blue-700 hover:text-blue-600"
+									className="whitespace-nowrap font-medium text-blue-700 dark:text-blue-50 dark:hover:text-blue-100 hover:text-blue-600"
 								>
 									Details
 								</Link>
@@ -106,7 +106,7 @@ export default function SingleAlert({
 										removeAlert(dropItem.id);
 									}, 500);
 								}}
-                className="inline-flex rounded-md bg-blue-50 p-1.5 text-blue-500"
+								className="inline-flex rounded-md bg-blue-50 dark:bg-transparent p-1.5 text-blue-500 dark:text-blue-50"
 							>
 								<span className="sr-only">Dismiss</span>
 								<XMarkIcon className="h-5 w-5" aria-hidden="true" />
@@ -116,17 +116,17 @@ export default function SingleAlert({
 				</div>
 			) : (
 				<div
-          className="flex bg-green-50 p-3 mb-2 mx-2 rounded-md"
+					className="flex bg-green-50 dark:bg-green-900 p-3 mb-2 mx-2 rounded-md"
 					key={dropItem.id}
 				>
 					<div className="flex-shrink-0">
 						<CheckCircleIcon
-              className="h-5 w-5 text-green-400"
+							className="h-5 w-5 text-green-400 dark:text-green-50"
 							aria-hidden="true"
 						/>
 					</div>
 					<div className="ml-3">
-            <p className="text-sm font-medium text-green-800">
+						<p className="text-sm font-medium text-green-800 dark:bg-white/80">
 							{dropItem.title}
 						</p>
 					</div>
@@ -140,7 +140,7 @@ export default function SingleAlert({
 										removeAlert(dropItem.id);
 									}, 500);
 								}}
-                className="inline-flex rounded-md bg-green-50 p-1.5 text-green-500"
+								className="inline-flex rounded-md bg-green-50 dark:bg-transparent p-1.5 text-green-500 dark:text-green-50"
 							>
 								<span className="sr-only">Dismiss</span>
 								<XMarkIcon className="h-5 w-5" aria-hidden="true" />

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

@@ -24,11 +24,11 @@ export default function AlertDropdown({}: AlertDropdownType) {
 	return (
 		<div
 			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 px-2 rounded-md bg-white dark:bg-gray-800 ring-1 ring-black ring-opacity-5 shadow-lg focus:outline-none overflow-hidden w-[400px] h-[500px] flex flex-col"
 		>
 			<div className="flex pl-3 flex-row justify-between text-md font-medium text-gray-800 dark:text-gray-200">
 				Notifications
-        <div className="flex gap-2 pr-3 ">
+				<div className="flex gap-3 pr-3 ">
 					<button
 						className="text-gray-800 hover:text-red-500 dark:text-gray-200 dark:hover:text-red-500"
 						onClick={() => {
@@ -46,7 +46,7 @@ export default function AlertDropdown({}: AlertDropdownType) {
 					</button>
 				</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-3 flex flex-col overflow-y-scroll w-full h-full scrollbar-hide text-gray-900 dark:text-gray-300">
 				{notificationList.length !== 0 ? (
 					notificationList.map((alertItem, index) => (
 						<SingleAlert

src/frontend/src/alerts/error/index.tsx

@@ -39,16 +39,16 @@ export default function ErrorAlert({
 						removeAlert(id);
 					}, 500);
 				}}
-				className="rounded-md w-96 mt-6 shadow-xl bg-red-50 p-4 cursor-pointer"
+				className="rounded-md w-96 mt-6 shadow-xl bg-red-50 dark:bg-red-900 p-4 cursor-pointer"
 			>
 				<div className="flex">
 					<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 dark:text-red-50" aria-hidden="true" />
 					</div>
 					<div className="ml-3">
-						<h3 className="text-sm font-medium text-red-800">{title}</h3>
+						<h3 className="text-sm font-medium text-red-800 dark:text-white/80">{title}</h3>
 						{list.length !== 0 ? (
-							<div className="mt-2 text-sm text-red-700">
+							<div className="mt-2 text-sm text-red-700 dark:text-red-50">
 								<ul className="list-disc space-y-1 pl-5">
 									{list.map((item, index) => (
 										<li key={index}>{item}</li>

src/frontend/src/alerts/notice/index.tsx

@@ -36,22 +36,22 @@ export default function NoticeAlert({
 					setShow(false);
 					removeAlert(id);
 				}}
-				className="rounded-md w-96 mt-6 shadow-xl bg-blue-50 p-4"
+				className="rounded-md w-96 mt-6 shadow-xl bg-blue-50 dark:bg-blue-900 p-4"
 			>
 				<div className="flex">
 					<div className="flex-shrink-0">
 						<InformationCircleIcon
-							className="h-5 w-5 text-blue-400"
+							className="h-5 w-5 text-blue-400 dark:text-blue-50"
 							aria-hidden="true"
 						/>
 					</div>
 					<div className="ml-3 flex-1 md:flex md:justify-between">
-						<p className="text-sm text-blue-700">{title}</p>
+						<p className="text-sm text-blue-700 dark:text-white/80">{title}</p>
 						<p className="mt-3 text-sm md:mt-0 md:ml-6">
 							{link !== "" ? (
 								<Link
 									to={link}
-									className="whitespace-nowrap font-medium text-blue-700 hover:text-blue-600"
+									className="whitespace-nowrap font-medium text-blue-700 dark:text-blue-50 hover:dark:text-blue-10 hover:text-blue-600"
 								>
 									Details
 								</Link>

src/frontend/src/alerts/success/index.tsx

@@ -34,17 +34,17 @@ export default function SuccessAlert({
 					setShow(false);
 					removeAlert(id);
 				}}
-				className="rounded-md w-96 mt-6 shadow-xl bg-green-50 p-4"
+				className="rounded-md w-96 mt-6 shadow-xl bg-green-50 dark:bg-green-900 p-4"
 			>
 				<div className="flex">
 					<div className="flex-shrink-0">
 						<CheckCircleIcon
-							className="h-5 w-5 text-green-400"
+							className="h-5 w-5 text-green-400 dark:text-green-50"
 							aria-hidden="true"
 						/>
 					</div>
 					<div className="ml-3">
-						<p className="text-sm font-medium text-green-800">{title}</p>
+						<p className="text-sm font-medium text-green-800 dark:text-white/80">{title}</p>
 					</div>
 				</div>
 			</div>

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/LightTooltipComponent/index.tsx

@@ -1,12 +1,12 @@
-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 }} />
 ))(({ theme }) => ({
 	[`& .${tooltipClasses.tooltip}`]: {
 		backgroundColor: theme.palette.common.white,
-    color: 'rgba(0, 0, 0, 0.87)',
+		color: "rgba(0, 0, 0, 0.87)",
 		boxShadow: theme.shadows[2],
 		fontSize: 14,
 	},

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,8 +1,12 @@
-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');
+import Convert from "ansi-to-html";
 var convert = new Convert({newline:true});
 
 export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
@@ -29,12 +33,15 @@ 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
+							className="w-full rounded-b-md px-4 pb-3 pt-3 pr-8"
+							style={{ backgroundColor: nodeColors["chat"] }}
+						>
 							{chat.message}
 						</div>
 					</div>

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

@@ -1,13 +1,17 @@
 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 (
+		<Transition
 			show={!open}
 			appear={true}
 			enter="transition ease-out duration-300"
@@ -18,7 +22,10 @@ export default function ChatTrigger({open, setOpen,flow}){
 			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
+					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"
+				>
 					<button
 						onClick={() => {
 							setOpen(true);
@@ -33,5 +40,6 @@ export default function ChatTrigger({open, setOpen,flow}){
 					</button>
 				</div>
 			</div>
-    </Transition>)
+		</Transition>
+	);
 }

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

@@ -4,7 +4,7 @@ import { ChatMessageType, ChatType } from "../../types/chat";
 import ChatTrigger from "./chatTrigger";
 import ChatModal from "../../modals/chatModal";
 
-const _ = require("lodash");
+import _ from "lodash";
 
 export default function Chat({ flow }: ChatType) {
 	const [open, setOpen] = useState(false);
@@ -26,7 +26,7 @@ export default function Chat({ flow }: ChatType) {
 	return (
 		<>
 			<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/floatComponent/index.tsx

@@ -1,25 +1,40 @@
-import { useEffect, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import { FloatComponentType } from "../../types/components";
+import { TabsContext } from "../../contexts/tabsContext";
 
-export default function FloatComponent({value, onChange, disabled}: FloatComponentType){
+export default function FloatComponent({
+	value,
+	onChange,
+	disabled,
+}: FloatComponentType) {
 	const [myValue, setMyValue] = useState(value ?? "");
 	useEffect(() => {
 		if (disabled) {
 			setMyValue("");
 			onChange("");
 		}
-    }, [disabled, onChange])
+	}, [disabled, onChange]);
+	const {setDisableCP} =  useContext(TabsContext)
 	return (
 		<div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
 			<input
 				type="number"
 				value={myValue}
-            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" : "")}
+				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" : "")
+				}
 				placeholder="Type a number from zero to one"
 				onChange={(e) => {
 					setMyValue(e.target.value);
 					onChange(e.target.value);
 				}}
+				onBlur={() => {
+					setDisableCP(false)
+				}}
+				onFocus={() => {
+					setDisableCP(true)
+				}}
 			/>
 		</div>
 	);

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

@@ -1,6 +1,7 @@
-import { useEffect, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import { InputComponentType } from "../../types/components";
 import { classNames } from "../../utils";
+import { TabsContext } from "../../contexts/tabsContext";
 
 export default function InputComponent({
 	value,
@@ -10,6 +11,7 @@ export default function InputComponent({
 }: InputComponentType) {
 	const [myValue, setMyValue] = useState(value ?? "");
 	const [pwdVisible, setPwdVisible] = useState(false);
+	const {setDisableCP} =  useContext(TabsContext)
 	useEffect(() => {
 		if (disabled) {
 			setMyValue("");
@@ -26,6 +28,12 @@ export default function InputComponent({
 		>
 			<input
 				value={myValue}
+				onBlur={() => {
+					setDisableCP(false)
+				}}
+				onFocus={() => {
+					setDisableCP(true)
+				}}
 				className={classNames(
 					"block w-full pr-12 form-input dark:bg-gray-900 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" : "",

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 {
 		for (let index = 0; index < suffixes.length; index++) {
 			if (fileName.endsWith(suffixes[index])) {
-				return true
+				return true;
 			}
 		}
-		return false
+		return false;
 	}
 
 	const handleButtonClick = () => {
@@ -71,7 +71,7 @@ export default function InputFileComponent({
 				<span
 					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" +
+						"truncate block w-full text-gray-500 dark:text-gray-300 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

@@ -1,25 +1,39 @@
 import { PlusIcon, XMarkIcon } from "@heroicons/react/24/outline";
-import { useEffect, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import { InputListComponentType } from "../../types/components";
+import { TabsContext } from "../../contexts/tabsContext";
 
-var _ = require("lodash");
+import _ from "lodash";
 
-export default function InputListComponent({ value, onChange, disabled}:InputListComponentType) {
+export default function InputListComponent({
+	value,
+	onChange,
+	disabled,
+}: InputListComponentType) {
 	const [inputList, setInputList] = useState(value ?? [""]);
 	useEffect(() => {
 		if (disabled) {
 			setInputList([""]);
 			onChange([""]);
 		}
-}, [disabled, onChange])
+	}, [disabled, onChange]);
+	const {setDisableCP} = useContext(TabsContext)
 	return (
-    <div className={(disabled ? "pointer-events-none cursor-not-allowed" : "") + "flex flex-col gap-3"}>
+		<div
+			className={
+				(disabled ? "pointer-events-none cursor-not-allowed" : "") +
+				"flex flex-col gap-3"
+			}
+		>
 			{inputList.map((i, idx) => (
 				<div key={idx} className="w-full flex gap-3">
 					<input
 						type="text"
 						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" + (disabled ? " bg-gray-200" : "")}
+						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" : "")
+						}
 						placeholder="Type a text"
 						onChange={(e) => {
 							setInputList((old) => {
@@ -29,24 +43,40 @@ export default function InputListComponent({ value, onChange, disabled}:InputLis
 							});
 							onChange(inputList);
 						}}
+						onBlur={() => {
+							setDisableCP(false)
+						}}
+						onFocus={() => {
+							setDisableCP(true)
+						}}
 					/>
-          {idx === inputList.length - 1 ? 
+					{idx === inputList.length - 1 ? (
-          <button onClick={() => {setInputList((old) => {
+						<button
+							onClick={() => {
+								setInputList((old) => {
 									let newInputList = _.cloneDeep(old);
-            newInputList.push('');
+									newInputList.push("");
 									return newInputList;
 								});
-          onChange(inputList);}}>
+								onChange(inputList);
+							}}
+						>
 							<PlusIcon className="w-4 h-4 hover:text-blue-600" />
 						</button>
-          : <button onClick={() => {setInputList((old) => {
+					) : (
+						<button
+							onClick={() => {
+								setInputList((old) => {
 									let newInputList = _.cloneDeep(old);
 									newInputList.splice(idx, 1);
 									return newInputList;
 								});
-          onChange(inputList);}}>
+								onChange(inputList);
+							}}
+						>
 							<XMarkIcon className="w-4 h-4 hover:text-red-600" />
-            </button>}
+						</button>
+					)}
 				</div>
 			))}
 		</div>

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

@@ -1,5 +1,6 @@
-import { useEffect, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import { FloatComponentType } from "../../types/components";
+import { TabsContext } from "../../contexts/tabsContext";
 
 export default function IntComponent({
 	value,
@@ -13,11 +14,23 @@ export default function IntComponent({
 			onChange("");
 		}
 	}, [disabled, onChange]);
+	const {setDisableCP} =useContext(TabsContext)
 	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.key !== "Backspace" &&
+						event.key !== "Enter" &&
+						event.key !== "Delete" &&
+						event.key !== "ArrowLeft" &&
+						event.key !== "ArrowRight" &&
+						!/^[-]?\d*$/.test(event.key)
+					) {
 						event.preventDefault();
 					}
 				}}
@@ -32,7 +45,14 @@ export default function IntComponent({
 					setMyValue(e.target.value);
 					onChange(e.target.value);
 				}}
+				onBlur={() => {
+					setDisableCP(false)
+				}}
+				onFocus={() => {
+					setDisableCP(true)
+				}}
 			/>
+			
 		</div>
 	);
 }

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

@@ -1,17 +1,28 @@
 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">
+			<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"/>
+				aria-hidden="true"
-                <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"/>
+				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/textAreaComponent/index.tsx

@@ -18,7 +18,7 @@ export default function TextAreaComponent({ value, onChange, disabled }:TextArea
       <div className="w-full flex items-center gap-3">
         <span onClick={()=>{openPopUp(<TextAreaModal value={myValue} setValue={(t:string) => {setMyValue(t); onChange(t);}}/>)}}
           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" +
+            "truncate block w-full text-gray-500 dark:text-gray-100 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/toggleComponent/index.tsx

@@ -3,12 +3,16 @@ 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({
+	enabled,
+	setEnabled,
+	disabled,
+}: ToggleComponentType) {
 	useEffect(() => {
 		if (disabled) {
 			setEnabled(false);
 		}
-}, [disabled, setEnabled])
+	}, [disabled, setEnabled]);
 	return (
 		<div className={disabled ? "pointer-events-none cursor-not-allowed" : ""}>
 			<Switch
@@ -25,7 +29,10 @@ export default function ToggleComponent({ enabled, setEnabled, disabled }:Toggle
 				<span
 					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"
+						"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

src/frontend/src/contexts/alertContext.tsx

@@ -1,7 +1,7 @@
 import { createContext, ReactNode, useState } from "react";
 import { AlertItemType } from "../types/alerts";
 
-var _ = require("lodash");
+import _ from "lodash";
 
 //types for alertContextType
 type alertContextType = {
@@ -119,7 +119,8 @@ function setSuccessData(newState: { title: string }) {
 		// If the new state has a "title" property, add a new success notification to the list
 		if (newState.title) {
 			setNotificationCenter(true); // show the notification center
-		pushNotificationList({ // add the new notification to the list
+			pushNotificationList({
+				// add the new notification to the list
 				type: "success",
 				title: newState.title,
 				id: _.uniqueId(),

src/frontend/src/contexts/darkContext.tsx

@@ -20,7 +20,7 @@ export function DarkProvider({ children }) {
 		} else {
 			document.getElementById("body").classList.remove("dark");
 		}
-	}, [dark])
+	}, [dark]);
 	return (
 		<darkContext.Provider
 			value={{

src/frontend/src/contexts/popUpContext.tsx

@@ -15,11 +15,11 @@ const PopUpProvider = ({ children }: PopUpProviderProps) => {
 	const [popUpElements, setPopUpElements] = useState<JSX.Element[]>([]);
 
 	const openPopUp = (element: JSX.Element) => {
-    setPopUpElements(prevPopUps => [element, ...prevPopUps]);
+		setPopUpElements((prevPopUps) => [element, ...prevPopUps]);
 	};
 
 	const closePopUp = () => {
-    setPopUpElements(prevPopUps => prevPopUps.slice(1));
+		setPopUpElements((prevPopUps) => prevPopUps.slice(1));
 	};
 
 	return (

src/frontend/src/contexts/tabsContext.tsx

@@ -12,7 +12,7 @@ import { normalCaseToSnakeCase, updateObject, updateTemplate } from "../utils";
 import { alertContext } from "./alertContext";
 import { typesContext } from "./typesContext";
 import { APITemplateType, TemplateVariableType } from "../types/api";
-const { v4: uuidv4 } = require("uuid");
+import { v4 as uuidv4 } from "uuid";
 
 const TabsContextInitialValue: TabsContextType = {
 	save: () => {},
@@ -26,6 +26,8 @@ const TabsContextInitialValue: TabsContextType = {
 	downloadFlow: (flow: FlowType) => {},
 	uploadFlow: () => {},
 	hardReset: () => {},
+	disableCP:false,
+	setDisableCP:(state:boolean)=>{},
 };
 
 export const TabsContext = createContext<TabsContextType>(
@@ -212,10 +214,13 @@ export function TabsProvider({ children }: { children: ReactNode }) {
 			return newFlows;
 		});
 	}
+	const [disableCP, setDisableCP] = useState(false);
 
 	return (
 		<TabsContext.Provider
 			value={{
+				disableCP,
+				setDisableCP,
 				save,
 				hardReset,
 				tabIndex,

src/frontend/src/contexts/typesContext.tsx

@@ -15,7 +15,7 @@ const initialValue:typesContextType = {
 	templates: {},
 	setTemplates: () => {},
 	data: {},
-	setData:()=>{}
+	setData: () => {},
 };
 
 export const typesContext = createContext<typesContextType>(initialValue);
@@ -36,8 +36,8 @@ export function TypesProvider({ children }:{children:ReactNode}) {
 			setTemplates(
 				Object.keys(result.data).reduce((acc, curr) => {
 					Object.keys(result.data[curr]).forEach((c: keyof APIKindType) => {
-						acc[c] = result.data[curr][c]
+						acc[c] = result.data[curr][c];
-					})
+					});
 					return acc;
 				}, {})
 			);
@@ -63,7 +63,11 @@ export function TypesProvider({ children }:{children:ReactNode}) {
 		reactFlowInstance.setNodes(
 			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/index.css

@@ -1,15 +1,17 @@
-
+@tailwind base;
+@tailwind components;
+@tailwind utilities;
 
 body {
 	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;
 	-webkit-font-smoothing: antialiased;
 	-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;
 }

src/frontend/src/index.tsx

@@ -5,13 +5,14 @@ import reportWebVitals from "./reportWebVitals";
 import { BrowserRouter } from "react-router-dom";
 import ContextWrapper from "./contexts";
 
+import './index.css';
+
 const root = ReactDOM.createRoot(
 	document.getElementById("root") as HTMLElement
 );
 root.render(
 	<ContextWrapper>
 		<BrowserRouter>
-
 			<App />
 		</BrowserRouter>
 	</ContextWrapper>

src/frontend/src/modals/ApiModal/index.tsx

@@ -0,0 +1,186 @@
+import { Dialog, Transition } from "@headlessui/react";
+import { IconCheck, IconClipboard, IconDownload } from '@tabler/icons-react';
+import { XMarkIcon, CommandLineIcon, CodeBracketSquareIcon } from "@heroicons/react/24/outline";
+import { Fragment, useContext, useRef, useState } from "react";
+import { PopUpContext } from "../../contexts/popUpContext";
+import "ace-builds/src-noconflict/mode-python";
+import "ace-builds/src-noconflict/theme-github";
+import "ace-builds/src-noconflict/theme-twilight";
+import "ace-builds/src-noconflict/ext-language_tools";
+// import "ace-builds/webpack-resolver";
+import { darkContext } from "../../contexts/darkContext";
+import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
+import { oneDark } from 'react-syntax-highlighter/dist/cjs/styles/prism';
+
+
+
+export default function ApiModal({ flowName }) {
+	const [open, setOpen] = useState(true);
+	const { dark } = useContext(darkContext);
+	const { closePopUp } = useContext(PopUpContext);
+	const [activeTab, setActiveTab] = useState(0);
+	const ref = useRef();
+	const [isCopied, setIsCopied] = useState<Boolean>(false);
+
+	const copyToClipboard = () => {
+		if (!navigator.clipboard || !navigator.clipboard.writeText) {
+			return;
+		}
+
+		navigator.clipboard.writeText(tabs[activeTab].code).then(() => {
+			setIsCopied(true);
+
+			setTimeout(() => {
+				setIsCopied(false);
+			}, 2000);
+		});
+	};
+	function setModalOpen(x: boolean) {
+		setOpen(x);
+		if (x === false) {
+			setTimeout(() => {
+				closePopUp();
+			}, 300);
+		}
+	}
+
+	const pythonApiCode = `import requests
+
+API_URL = "${window.location.protocol}//${window.location.host}/predict"
+
+def predict(message):
+    with open("${flowName}.json", "r") as f:
+        json_data = json.load(f)
+    payload = {'exported_flow': json_data, 'message': message}
+    response = requests.post(API_URL, json=payload)
+    return response.json() # JSON {"result": "Response"}
+
+print(predict("Your message"))`;
+
+const pythonCode = `from langflow import load_flow_from_json
+
+flow = load_flow_from_json("${flowName}.json")
+# Now you can use it like any chain
+flow("Hey, have you heard of LangFlow?")`;
+
+	const tabs = [
+		{
+			name: "Python API",
+			mode: "python",
+			image: "https://images.squarespace-cdn.com/content/v1/5df3d8c5d2be5962e4f87890/1628015119369-OY4TV3XJJ53ECO0W2OLQ/Python+API+Training+Logo.png?format=1000w",
+			code: pythonApiCode,
+		},
+		{
+			name: "Python Code",
+			mode: "python",
+			image: "https://cdn-icons-png.flaticon.com/512/5968/5968350.png",
+			code: pythonCode,
+		},
+
+	]
+	return (
+		<Transition.Root show={open} appear={true} as={Fragment}>
+			<Dialog
+				as="div"
+				className="relative z-10"
+				onClose={setModalOpen}
+				initialFocus={ref}
+			>
+				<Transition.Child
+					as={Fragment}
+					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="flex h-full items-end justify-center p-4 text-center sm:items-center sm:p-0">
+						<Transition.Child
+							as={Fragment}
+							enter="ease-out duration-300"
+							enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
+							enterTo="opacity-100 translate-y-0 sm:scale-100"
+							leave="ease-in duration-200"
+							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="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">
+									<button
+										type="button"
+										className="rounded-md text-gray-400 hover:text-gray-500"
+										onClick={() => {
+											setModalOpen(false);
+										}}
+									>
+										<span className="sr-only">Close</span>
+										<XMarkIcon className="h-6 w-6" aria-hidden="true" />
+									</button>
+								</div>
+								<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="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">
+											<CodeBracketSquareIcon
+												className="h-6 w-6 text-blue-600"
+												aria-hidden="true"
+											/>
+										</div>
+										<div className="mt-4 text-center sm:ml-4 sm:text-left">
+											<Dialog.Title
+												as="h3"
+												className="text-lg font-medium dark:text-white leading-10 text-gray-900"
+											>
+												Code
+											</Dialog.Title>
+										</div>
+									</div>
+									<div className="h-full w-full bg-gray-200 overflow-auto dark:bg-gray-900 p-4 gap-4 flex flex-row justify-center items-center">
+										<div className="flex flex-col h-full w-full ">
+											<div className="flex px-5 z-10">
+												{tabs.map((tab, index) => (
+													<button onClick={() => {
+														setActiveTab(index);
+													}} className={"p-2 rounded-t-lg w-44 border border-b-0 border-gray-300 dark:border-gray-700 dark:text-gray-300 -mr-px flex justify-center items-center gap-4 " + (activeTab === index ? " bg-white dark:bg-gray-800" : "bg-gray-100 dark:bg-gray-900")}>
+														{tab.name}
+														<img src={tab.image} className="w-6" />
+													</button>
+												))}
+											</div>
+											<div className="overflow-hidden px-4 py-5 sm:p-6 w-full h-full rounded-lg shadow bg-white dark:bg-gray-800">
+												<div className="w-full flex items-center justify-between mb-2">
+													<span className="text-sm text-gray-500 dark:text-gray-300">
+														Export your flow to use it with this code.
+													</span>
+													<button
+														className="flex gap-1.5 items-center rounded bg-none p-1 text-xs text-gray-500 dark:text-gray-300"
+														onClick={copyToClipboard}
+													>
+														{isCopied ? <IconCheck size={18} /> : <IconClipboard size={18} />}
+														{isCopied ? 'Copied!' : 'Copy code'}
+													</button>
+												</div>
+												<SyntaxHighlighter
+												className="h-[370px]"
+													language={tabs[activeTab].mode}
+													style={oneDark}
+													customStyle={{ margin: 0 }}
+												>
+													{tabs[activeTab].code}
+												</SyntaxHighlighter>
+											</div>
+										</div>
+									</div>
+								</div>
+							</Dialog.Panel>
+						</Transition.Child>
+					</div>
+				</div>
+			</Dialog>
+		</Transition.Root>
+	);
+}

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

@@ -35,7 +35,7 @@ export default function ModalField({ data, title, required, id, name, type }) {
 		>
 			{display && (
 				<div>
-					<span className="mx-2">{title}</span>
+					<span className="mx-2 dark:text-gray-300">{title}</span>
 					<span className="text-red-600">{required ? " *" : ""}</span>
 				</div>
 			)}

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

@@ -56,7 +56,7 @@ export default function NodeModal({ data }: { data: NodeDataType }) {
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 								<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -87,9 +87,13 @@ 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) =>
+																t.charAt(0) !== "_" &&
+																data.node.template[t].advanced &&
+																data.node.template[t].show
+														)
 														.map((t: string, idx) => {
 															return (
 																<ModalField
@@ -99,9 +103,7 @@ export default function NodeModal({ data }: { data: NodeDataType }) {
 																		data.node.template[t].display_name
 																			? data.node.template[t].display_name
 																			: data.node.template[t].name
-																				? toNormalCase(
+																			? toNormalCase(data.node.template[t].name)
-																						data.node.template[t].name
-																				  )
 																			: toNormalCase(t)
 																	}
 																	required={data.node.template[t].required}
@@ -116,8 +118,7 @@ export default function NodeModal({ data }: { data: NodeDataType }) {
 																	type={data.node.template[t].type}
 																/>
 															);
-															})
+														})}
-													}
 												</div>
 											</div>
 										</div>

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

@@ -30,8 +30,8 @@ export default function ChatInput({
 					setChatValue(e.target.value);
 				}}
 				className={classNames(
-					lockChat ? "bg-gray-500 text-white" : "dark:bg-gray-700",
+					lockChat ? "bg-gray-300 text-black dark:bg-gray-700 dark:text-gray-300" : "bg-gray-200 text-black dark:bg-gray-900 dark:text-gray-300",
-					"form-input block w-full  custom-scroll h-10 rounded-md border-gray-300 dark:border-gray-600  dark:text-white pr-10 sm:text-sm"
+					"form-input block w-full  custom-scroll h-10 rounded-md border-gray-300 dark:border-gray-600  pr-10 sm:text-sm"
 				)}
 				placeholder={"Send a message..."}
 			/>
@@ -39,12 +39,12 @@ export default function ChatInput({
 				<button disabled={lockChat} onClick={() => sendMessage()}>
 					{lockChat ? (
 						<LockClosedIcon
-							className="h-5 w-5 text-gray-400  dark:hover:text-gray-300 animate-pulse"
+							className="h-5 w-5 text-gray-500  dark:hover:text-gray-300 animate-pulse"
 							aria-hidden="true"
 						/>
 					) : (
 						<PaperAirplaneIcon
-							className="h-5 w-5 text-gray-400 hover:text-gray-600 dark:hover:text-gray-300"
+							className="h-5 w-5 text-gray-500 hover:text-gray-600 dark:hover:text-gray-300"
 							aria-hidden="true"
 						/>
 					)}

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

@@ -0,0 +1,82 @@
+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 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
+			className=" w-[570px]"
+				language={language}
+				style={oneDark}
+				customStyle={{ margin: 0 }}
+			>
+				{value}
+			</SyntaxHighlighter>
+		</div>
+	);
+});
+CodeBlock.displayName = "CodeBlock";

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

@@ -1,34 +1,48 @@
 import { ChatBubbleOvalLeftEllipsisIcon } from "@heroicons/react/24/outline";
-import { useState } from "react";
+import { useEffect, useRef, useState } from "react";
 import { ChatMessageType } from "../../../types/chat";
 import { classNames } from "../../../utils";
 import AiIcon from "../../../assets/Gooey Ring-5s-271px.svg";
+import AiIconStill from "../../../assets/froze-flow.png"
 import { UserIcon } from "@heroicons/react/24/solid";
 import FileCard from "../fileComponent";
-var Convert = require("ansi-to-html");
+import ReactMarkdown from "react-markdown";
-var convert = new Convert({ newline: true });
+import rehypeMathjax from "rehype-mathjax";
+import remarkGfm from "remark-gfm";
+import remarkMath from "remark-math";
+import { CodeBlock } from "./codeBlock";
+import Convert from "ansi-to-html";
 
-export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
+export default function ChatMessage({ chat, lockChat }: { chat: ChatMessageType, lockChat: boolean }) {
+	const [message, setMessage] = useState("");
+	const imgRef = useRef(null);
+	useEffect(() => {
+		setMessage(chat.message);
+	}, [chat.message]);
 	const [hidden, setHidden] = useState(true);
 	return (
 		<div
 			className={classNames(
 				"w-full py-2 pl-2 flex",
-        chat.isSend ? "bg-white dark:bg-gray-800 " : "bg-gray-200  dark:bg-gray-700"
+				chat.isSend
+					? "bg-white dark:bg-gray-900 "
+					: "bg-gray-200  dark:bg-gray-800"
 			)}
 		>
 			<div
 				className={classNames(
-          "rounded-full w-8 h-8 flex items-center my-3 justify-center",
+					"rounded-full overflow-hidden w-8 h-8 flex items-center my-3 justify-center"
-          chat.isSend ? "bg-gray-900" : "bg-gray-200"
 				)}
 			>
-        {!chat.isSend && <img className="scale-150" src={AiIcon} />}
+				{!chat.isSend && <div className="relative w-8 h-8">
-        {chat.isSend && <UserIcon className="w-6 h-6 -mb-1 text-gray-200" />}
+				<img className={"absolute transition-opacity duration-500 scale-150 " + (lockChat ? "opacity-100" : "opacity-0")} src={AiIcon} />
+				<img className={"absolute transition-opacity duration-500 scale-150 " + (lockChat ? "opacity-0" : "opacity-100")} src={AiIconStill} />
+					</div>}
+				{chat.isSend && <UserIcon className="w-6 h-6 -mb-1 text-gray-800 dark:text-gray-200" />}
 			</div>
 			{!chat.isSend ? (
 				<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">
+					<div className="w-full relative text-start inline-block text-gray-600 dark:text-gray-300 text-sm font-normal">
 						{hidden && chat.thought && chat.thought !== "" && (
 							<div
 								onClick={() => setHidden((prev) => !prev)}
@@ -40,8 +54,8 @@ export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
 						{chat.thought && chat.thought !== "" && !hidden && (
 							<div
 								onClick={() => setHidden((prev) => !prev)}
-                className=" text-start inline-block rounded-md  h-full border border-gray-300
+								className=" text-start inline-block rounded-md text-gray-600 dark:text-gray-200 h-full border border-gray-300 dark:border-gray-500
-								bg-gray-100 w-[95%] pb-3 pt-3 px-2 ml-3 cursor-pointer scrollbar-hide overflow-scroll"
+								bg-gray-100 dark:bg-gray-800 w-[95%] pb-3 pt-3 px-2 ml-3 cursor-pointer scrollbar-hide overflow-scroll"
 								dangerouslySetInnerHTML={{
 									__html: convert.toHtml(chat.thought),
 								}}
@@ -49,8 +63,49 @@ export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
 						)}
 						{chat.thought && chat.thought !== "" && !hidden && <br></br>}
 						<div className="w-full px-4 pb-3 pt-3 pr-8">
-              <span className="dark:text-white">
+							<div className="dark:text-white w-full">
-                {chat.message}
+								<div className="w-full">
+									<ReactMarkdown
+										remarkPlugins={[remarkGfm, remarkMath]}
+										rehypePlugins={[rehypeMathjax]}
+										className="markdown prose dark:prose-invert text-gray-600 dark:text-gray-200"
+										components={{
+											code({ node, inline, className, children, ...props }) {
+												if (children.length) {
+													if (children[0] == "▍") {
+														return (
+															<span className="animate-pulse cursor-default mt-1">
+																▍
+															</span>
+														);
+													}
+
+													children[0] = (children[0] as string).replace(
+														"`▍`",
+														"▍"
+													);
+												}
+
+												const match = /language-(\w+)/.exec(className || "");
+
+												return !inline ? (
+													<CodeBlock
+														key={Math.random()}
+														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) => {
@@ -66,14 +121,20 @@ export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
 										})}
 									</div>
 								)}
-              </span>
+							</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">
-            {chat.message}
+						<ReactMarkdown
+							remarkPlugins={[remarkGfm, remarkMath]}
+							rehypePlugins={[rehypeMathjax]}
+							className="markdown prose dark:prose-invert text-gray-600 dark:text-gray-200"
+						>
+							{message}
+						</ReactMarkdown>
 					</div>
 				</div>
 			)}

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

@@ -1,12 +1,13 @@
 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 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 link = document.createElement('a');
+		const link = document.createElement("a");
 		link.href = url;
 		link.download = fileName + ".png";
 		document.body.appendChild(link);
@@ -14,6 +15,41 @@ export default function FileCard({ fileName, content, fileType }) {
 		document.body.removeChild(link);
 		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">
 				{" "}
+				{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>
 				<CloudArrowDownIcon className="w-6 h-6 text-current ml-auto" />
 			</div>

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

@@ -1,23 +1,18 @@
 import { Dialog, Transition } from "@headlessui/react";
-import {
+import { ChatBubbleOvalLeftEllipsisIcon, XMarkIcon } 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";
 import ChatMessage from "./chatMessage";
 import { FaEraser } from "react-icons/fa";
+import { HiX } from "react-icons/hi";
 import { sendAllProps } from "../../types/api";
 import { ChatMessageType, ChatType } from "../../types/chat";
 import ChatInput from "./chatInput";
 
-const _ = require("lodash");
+import _ from "lodash";
 
 export default function ChatModal({
   flow,
@@ -32,8 +27,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,29 +55,53 @@ export default function ChatModal({
     });
   };
 
-  function connectWS() {
+  //add proper type signature for function
-    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.host
-            }/chat/${flow.id}`;
 
-      const newWs = new WebSocket(urlWs);
+  function updateLastMessage({
-      newWs.onopen = () => {
+    str,
-        console.log("WebSocket connection established!");
+    thought,
-      };
+    end = false,
-      newWs.onmessage = (event) => {
+    files,
-        try {
+  }: {
+    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);
-          const data = JSON.parse(event.data);
+      setTimeout(() => {
-          console.log("Received data:", data);
+        connectWS();
-          //get chat history
+      }, 1000);
-          if (Array.isArray(data)) {
+    }
-            console.log(data);
+  }
 
+  function handleWsMessage(data: any) {
+    if (Array.isArray(data)) {
+      //set chat history
       setChatHistory((_) => {
         let newChatHistory: ChatMessageType[] = [];
         data.forEach(
@@ -107,81 +133,98 @@ export default function ChatModal({
         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) {
-              addChatHistory(
+        updateLastMessage({
-                data.message,
+          end: true,
-                false,
+          files: data.files,
-                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 = (_) => {
-        if (open) {
       setLockChat(false);
-          setTimeout(() => {
+      isStream = false;
-            connectWS();
-          }, 1000);
     }
+    if (data.type === "stream" && isStream) {
+      updateLastMessage({ str: data.message });
+    }
+  }
+
+  function connectWS() {
+    try {
+      const urlWs =
+        process.env.NODE_ENV === "development"
+          ? `ws://localhost:7860/chat/${flow.id}`
+          : `${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host
+          }/chat/${flow.id}`;
+
+      const newWs = new WebSocket(urlWs);
+      newWs.onopen = () => {
+        console.log("WebSocket connection established!");
+      };
+      console.log(flow.id);
+      newWs.onmessage = (event) => {
+        const data = JSON.parse(event.data);
+        console.log("Received data:", data);
+        handleWsMessage(data);
+        //get chat history
+      };
+      newWs.onclose = (event) => {
+        handleOnClose(event);
       };
       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");
+      console.log("unmount");
-        newWs.close();
+      console.log(ws);
-      };
+      if (ws) {
+        ws.current.close();
       }
-  }, [lockChat]);
-
-  useEffect(() => {
-    let newWs = connectWS();
-    return () => {
-      console.log("trigger");
-      newWs.close();
     };
   }, []);
 
   async function sendAll(data: sendAllProps) {
     try {
       if (ws) {
-        ws.send(JSON.stringify(data));
+        ws.current.send(JSON.stringify(data));
       }
     } catch (error) {
       setErrorData({
-        title: "There was an erro sending the message",
+        title: "There was an error sending the message",
         list: [error.message],
       });
       setChatValue(data.message);
@@ -193,12 +236,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,8 +264,7 @@ 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
                 : toNormalCase(template[t].name)
               }.`,
@@ -255,7 +291,6 @@ export default function ChatModal({
         let message = chatValue;
         setChatValue("");
         addChatHistory(message, true);
-
         sendAll({
           ...reactFlowInstance.toObject(),
           message,
@@ -278,20 +313,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,28 +350,34 @@ 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"
+                    className="absolute top-2 right-10 hover:text-red-500 text-gray-600 dark:text-gray-300 dark:hover:text-red-500 z-30"
                   >
                     <FaEraser className="w-4 h-4" />
                   </button>
+                  <button
+                    onClick={() => setModalOpen(false)}
+                    className="absolute top-1.5 right-2 hover:text-red-500 text-gray-600 dark:text-gray-300 dark:hover:text-red-500 z-30"
+                  >
+                    <HiX className="w-5 h-5" />
+                  </button>
                 </div>
                 <div className="w-full h-full bg-white dark:bg-gray-800 border-t dark:border-t-gray-600 flex-col flex items-center overflow-scroll scrollbar-hide">
                   {chatHistory.length > 0 ? (
-                    chatHistory.map((c, i) => <ChatMessage chat={c} key={i} />)
+                    chatHistory.map((c, i) => <ChatMessage lockChat={lockChat} chat={c} key={i} />)
                   ) : (
                     <div className="flex flex-col h-full text-center justify-center w-full items-center align-middle">
                       <span>
                         👋{" "}
-                        <span className="text-gray-600 text-lg">
+                        <span className="text-gray-600 dark:text-gray-300 text-lg">
                           LangFlow Chat
                         </span>
                       </span>
                       <br />
-                      <div className="bg-gray-100 rounded-md w-2/4 px-6 py-8 border border-gray-200">
+                      <div className="bg-gray-100 dark:bg-gray-900 rounded-md w-2/4 px-6 py-8 border border-gray-200 dark:border-gray-700">
                         <span className="text-base text-gray-500">
                           Start a conversation and click the agent’s thoughts{" "}
                           <span>

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

@@ -7,7 +7,7 @@ import "ace-builds/src-noconflict/mode-python";
 import "ace-builds/src-noconflict/theme-github";
 import "ace-builds/src-noconflict/theme-twilight";
 import "ace-builds/src-noconflict/ext-language_tools";
-import "ace-builds/webpack-resolver";
+// import "ace-builds/webpack-resolver";
 import { darkContext } from "../../contexts/darkContext";
 import { checkCode } from "../../controllers/API";
 import { alertContext } from "../../contexts/alertContext";
@@ -67,7 +67,7 @@ export default function CodeAreaModal({
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 								<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -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({

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

@@ -16,7 +16,7 @@ export default function ExportModal() {
 	const { closePopUp } = useContext(PopUpContext);
 	const ref = useRef();
 	const { setErrorData } = useContext(alertContext);
-  const { flows, tabIndex, updateFlow, downloadFlow } = useContext(TabsContext);
+	const { flows, tabIndex, updateFlow, downloadFlow,setDisableCP } = useContext(TabsContext);
 	function setModalOpen(x: boolean) {
 		setOpen(x);
 		if (x === false) {
@@ -62,7 +62,7 @@ export default function ExportModal() {
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 								<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -113,6 +113,12 @@ export default function ExportModal() {
 												placeholder="File 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"
+												onBlur={() => {
+													setDisableCP(false);
+												}}
+												onFocus={() => {
+													setDisableCP(true);
+												}}
 											/>
 										</div>
 										<div className="w-full">
@@ -127,6 +133,12 @@ export default function ExportModal() {
 												</span>
 											</label>
 											<textarea
+												onBlur={() => {
+													setDisableCP(false);
+												}}
+												onFocus={() => {
+													setDisableCP(true);
+												}}
 												name="description"
 												id="description"
 												onChange={(event) => {
@@ -151,6 +163,12 @@ export default function ExportModal() {
 													id="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"
+													onBlur={() => {
+														setDisableCP(false);
+													}}
+													onFocus={() => {
+														setDisableCP(true);
+													}}
 												/>
 												<span className="ml-2 font-medium text-gray-700 dark:text-white">
 													Save with my API keys

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

@@ -29,93 +29,81 @@ export default function ButtonBox({
 	let padding: string;
 	let marginTop: string;
 	let height: string;
-	let widht: string;
+	let width: string;
 	switch (size) {
 		case "small":
 			bigCircle = "h-12 w-12";
 			smallCircle = "h-8 w-8";
 			titleFontSize = "text-sm";
 			descriptionFontSize = "text-xs";
-			padding = "p-2";
+			padding = "p-2 py-3";
 			marginTop = "mt-2";
 			height = "h-36";
-			widht = "w-32";
+			width = "w-32";
 			break;
 		case "medium":
 			bigCircle = "h-16 w-16";
 			smallCircle = "h-12 w-12";
 			titleFontSize = "text-base";
 			descriptionFontSize = "text-sm";
-			padding = "p-4";
+			padding = "p-4 py-5";
 			marginTop = "mt-3";
 			height = "h-44";
-			widht = "w-36";
+			width = "w-36";
 			break;
 		case "big":
 			bigCircle = "h-20 w-20";
 			smallCircle = "h-16 w-16";
 			titleFontSize = "text-lg";
 			descriptionFontSize = "text-sm";
-			padding = "p-8";
+			padding = "p-8 py-10";
 			marginTop = "mt-6";
 			height = "h-56";
-			widht = "w-44";
+			width = "w-44";
 			break;
 		default:
 			bigCircle = "h-20 w-20";
 			smallCircle = "h-16 w-16";
 			titleFontSize = "text-lg";
 			descriptionFontSize = "text-sm";
-			padding = "p-8";
+			padding = "p-8 py-10";
 			marginTop = "mt-6";
 			height = "h-56";
-			widht = "w-44";
+			width = "w-44";
 			break;
 	}
 	return (
 		<button disabled={deactivate} onClick={onClick}>
-			<Tooltip title={description} placement="bottom">
 				<div
 					className={classNames(
-						"col-span-1 flex flex-col  divide-y divide-gray-200 rounded-lg text-center shadow border border-gray-300 hover:shadow-lg transform hover:scale-105",
+						"flex flex-col justify-center items-center rounded-lg text-center shadow border border-gray-300 dark:border-gray-800 hover:shadow-lg transform hover:scale-105",
 						bgColor,
 						height,
-						widht
+						width,
+						padding
 					)}
 				>
-					<div className={`flex flex-1 flex-col ${padding}`}>
 						<div
-							className={`mx-auto flex items-center justify-center ${bigCircle} bg-white/30 rounded-full`}
+							className={`flex items-center justify-center ${bigCircle} bg-white/30 dark:bg-white/30 rounded-full`}
 						>
 							<div
-								className={`mx-auto flex items-center justify-center ${smallCircle} bg-white rounded-full`}
+								className={`flex items-center justify-center ${smallCircle} bg-white dark:bg-white/80 rounded-full`}
 							>
 								<div className={textColor}>{icon}</div>
 							</div>
 						</div>
+						<div className="mt-auto mb-auto">
 						<h3
 							className={classNames(
-								"font-semibold text-white",
+								"font-semibold text-white dark:text-white/80",
 								titleFontSize,
 								marginTop
 							)}
 						>
 							{title}
 						</h3>
-						<div className="mt-1 flex flex-grow flex-col justify-between">
-							<dt className="sr-only">{title}</dt>
-							{/* <dd
-								className={classNames(
-									"text-gray-100 line-clamp-2",
-									descriptionFontSize
-								)}
-							>
-								{deactivate ? "Coming soon" : description}
-							</dd> */}
 						</div>
 				</div>
-				</div>
-			</Tooltip>
 		</button>
 	);
 }

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

@@ -16,7 +16,7 @@ import { error } from "console";
 import { alertContext } from "../../contexts/alertContext";
 import LoadingComponent from "../../components/loadingComponent";
 import { FlowType } from "../../types/flow";
-import { classNames, toNormalCase } from "../../utils";
+import { classNames, snakeToSpaces, toNormalCase } from "../../utils";
 
 export default function ImportModal() {
 	const [open, setOpen] = useState(true);
@@ -82,11 +82,11 @@ export default function ImportModal() {
 							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="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-[776px]">
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 								<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -109,28 +109,6 @@ export default function ImportModal() {
 												<ArrowLeftIcon className="h-6 w-6" aria-hidden="true" />
 											</button>
 										</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">
-                      <a
-                        href="https://github.com/logspace-ai/langflow_examples"
-                        target="_blank"
-                        className="flex items-center justify-center"
-                        rel="noreferrer"
-                      >
-                        <svg
-                          width="24"
-                          viewBox="0 0 98 96"
-                          xmlns="http://www.w3.org/2000/svg"
-                        >
-                          <path
-                            fill-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"
-                            fill="#24292f"
-                          />
-                        </svg>
-                        <span className="ml-2">LangFlow Examples</span>
-                      </a>
-                    </div>
 									</>
 								)}
 								<div className="h-full w-full flex flex-col justify-center items-center">
@@ -152,7 +130,7 @@ export default function ImportModal() {
 									</div>
 									<div
 										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 overflow-y-auto scrollbar-hide",
 											showExamples && !loadingExamples
 												? "flex flex-row start justify-start items-start p-9 flex-wrap overflow-auto"
 												: "flex flex-row justify-center items-center p-4"
@@ -162,7 +140,7 @@ export default function ImportModal() {
 											<div className="flex h-full w-full justify-evenly items-center">
 												<ButtonBox
 													size="big"
-                          bgColor="bg-emerald-500"
+													bgColor="bg-emerald-500 dark:bg-emerald-500/75"
 													description="Prebuilt Examples"
 													icon={
 														<DocumentDuplicateIcon className="h-10 w-10 flex-shrink-0" />
@@ -171,12 +149,12 @@ export default function ImportModal() {
 														setShowExamples(true);
 														handleExamples();
 													}}
-                          textColor="text-emerald-400"
+													textColor="text-emerald-500 dark:text-emerald-500/75"
 													title="Examples"
 												></ButtonBox>
 												<ButtonBox
 													size="big"
-                          bgColor="bg-blue-500"
+													bgColor="bg-blue-500 dark:bg-blue-500/75"
 													description="Import from Local"
 													icon={
 														<ComputerDesktopIcon className="h-10 w-10 flex-shrink-0" />
@@ -185,8 +163,8 @@ export default function ImportModal() {
 														uploadFlow();
 														setModalOpen(false);
 													}}
-                          textColor="text-blue-500"
+													textColor="text-blue-500 dark:text-blue-500/75"
-                          title="Local file"
+													title="Local File"
 												></ButtonBox>
 											</div>
 										)}
@@ -203,7 +181,7 @@ export default function ImportModal() {
 														{" "}
 														<ButtonBox
 															size="small"
-                              bgColor="bg-emerald-500"
+															bgColor="bg-emerald-500 dark:bg-emerald-500/75"
 															description={
 																example.description ?? "Prebuilt Examples"
 															}
@@ -214,13 +192,35 @@ export default function ImportModal() {
 																addFlow(example);
 																setModalOpen(false);
 															}}
-                              textColor="text-emerald-400"
+															textColor="text-emerald-500 dark:text-emerald-500/75"
-                              title={toNormalCase(example.name)}
+															title={example.name}
 														></ButtonBox>
 													</div>
 												);
 											})}
 									</div>
+									<div className="bg-white dark:bg-gray-800 w-full h-20 flex items-center justify-center">
+											<a
+												href="https://github.com/logspace-ai/langflow_examples"
+												target="_blank"
+												className="flex items-center justify-center text-gray-600 dark:text-gray-300"
+												rel="noreferrer"
+											>
+												<svg
+													width="24"
+													viewBox="0 0 98 96"
+													xmlns="http://www.w3.org/2000/svg"
+												>
+													<path
+														fill-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"
+														fill="currentColor"
+													/>
+												</svg>
+												<span className="ml-2 ">LangFlow Examples</span>
+											</a>
+										</div>
 								</div>
 							</Dialog.Panel>
 						</Transition.Child>

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

@@ -61,7 +61,7 @@ export default function PromptAreaModal({
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 								<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -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,9 +1,18 @@
 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({
+	value,
+	setValue,
+}: {
+	setValue: (value: string) => void;
+	value: string | string[];
+}) {
 	const [open, setOpen] = useState(true);
 	const [myValue, setMyValue] = useState(value);
 	const { closePopUp } = useContext(PopUpContext);
@@ -11,7 +20,9 @@ export default function TextAreaModal({value, setValue}:{setValue:(value:string)
 	function setModalOpen(x: boolean) {
 		setOpen(x);
 		if (x === false) {
-            setTimeout(() => {closePopUp()}, 300);
+			setTimeout(() => {
+				closePopUp();
+			}, 300);
 		}
 	}
 	return (
@@ -49,7 +60,7 @@ export default function TextAreaModal({value, setValue}:{setValue:(value:string)
 								<div className=" z-50 absolute top-0 right-0 hidden pt-4 pr-4 sm:block">
 									<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"
 										onClick={() => {
 											setModalOpen(false);
 										}}
@@ -78,7 +89,15 @@ export default function TextAreaModal({value, setValue}:{setValue:(value:string)
 									<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="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
+													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);
+													}}
+												/>
 											</div>
 										</div>
 									</div>
@@ -100,5 +119,5 @@ export default function TextAreaModal({value, setValue}:{setValue:(value:string)
 				</div>
 			</Dialog>
 		</Transition.Root>
-    )
+	);
 }

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

@@ -1,13 +1,11 @@
-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) => {
 	return (
 		<g>

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

@@ -1,23 +1,20 @@
-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 }) => (
 				<>
 					<div>
-						<Disclosure.Button className="select-none bg-gray-50 dark:bg-gray-700 dark:border-y-gray-600 w-full flex justify-between items-center -mt-px px-3 py-2 border-y border-y-gray-200">
+						<Disclosure.Button className="select-none bg-gray-50 dark:bg-gray-700/60 dark:border-y-gray-600 w-full flex justify-between items-center -mt-px px-3 py-2 border-y border-y-gray-200">
 							<div className="flex gap-4">
-								<Icon className="w-6 text-gray-800 dark:text-white" />
+								<Icon className="w-6 text-gray-800 dark:text-white/80" />
-								<span className="flex items-center text-sm text-gray-800 dark:text-white font-medium">
+								<span className="flex items-center text-sm text-gray-800 dark:text-white/80 font-medium">
 									{title}
 								</span>
 							</div>

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,7 +19,9 @@ export default function ExtraSidebar() {
 
 	return (
 		<div className="mt-1 w-full">
-			{Object.keys(data).sort().map((d: keyof APIObjectType, i) => (
+			{Object.keys(data)
+				.sort()
+				.map((d: keyof APIObjectType, i) => (
 					<DisclosureComponent
 						key={i}
 						button={{
@@ -31,7 +30,9 @@ export default function ExtraSidebar() {
 						}}
 					>
 						<div className="p-2 flex flex-col gap-2">
-						{Object.keys(data[d]).sort().map((t: string, k) => (
+							{Object.keys(data[d])
+								.sort()
+								.map((t: string, k) => (
 									<div key={k}>
 										<div
 											draggable

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

@@ -3,11 +3,18 @@ import { useContext, useState } from "react";
 import { TabsContext } from "../../../../contexts/tabsContext";
 import { FlowType } from "../../../../types/flow";
 
-var _ = require("lodash");
+import _ from "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, setDisableCP } = useContext(TabsContext);
 	const [isRename, setIsRename] = useState(false);
 	const [value, setValue] = useState("");
 	return (
@@ -30,14 +37,18 @@ export default function TabComponent({ selected, flow, onClick }:{flow:FlowType,
 						</button>
 					</div>
 				) : (
-					<div className="bg-white dark:text-white dark:bg-gray-700 flex select-none justify-between w-44 items-center border border-b-0 border-gray-300 dark:border-gray-600 px-4 py-1 rounded-t-xl -ml-px">
+					<div className="bg-white dark:text-white dark:bg-gray-700/60 flex select-none justify-between w-44 items-center border border-b-0 border-gray-300 dark:border-gray-600 px-4 py-1 rounded-t-xl -ml-px">
 						{isRename ? (
 							<input
+								onFocus={() => {
+									setDisableCP(true);
+								}}
 								autoFocus
 								className="bg-transparent focus:border-none active:outline hover:outline focus:outline outline-gray-300 rounded-md  w-28"
 								onBlur={() => {
 									setIsRename(false);
 									if (value !== "") {
+										setDisableCP(false);
 										let newFlow = _.cloneDeep(flow);
 										newFlow.name = value;
 										updateFlow(newFlow);
@@ -85,4 +96,3 @@ export default function TabComponent({ selected, flow, onClick }:{flow:FlowType,
 		</>
 	);
 }
-

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

@@ -8,6 +8,7 @@ import {
 	ArrowDownTrayIcon,
 	ArrowUpTrayIcon,
 	BellIcon,
+	CodeBracketSquareIcon,
 	MoonIcon,
 	SunIcon,
 } from "@heroicons/react/24/outline";
@@ -17,13 +18,14 @@ import { alertContext } from "../../../../contexts/alertContext";
 import ImportModal from "../../../../modals/importModal";
 import ExportModal from "../../../../modals/exportModal";
 import { typesContext } from "../../../../contexts/typesContext";
+import ApiModal from "../../../../modals/ApiModal";
 
 export default function TabsManagerComponent() {
 	const { flows, addFlow, tabIndex, setTabIndex, uploadFlow, downloadFlow } =
 		useContext(TabsContext);
 	const { openPopUp } = useContext(PopUpContext);
 	const { templates } = useContext(typesContext);
-	const AlertWidth = 256;
+	const AlertWidth = 384;
 	const { dark, setDark } = useContext(darkContext);
 	const { notificationCenter, setNotificationCenter } =
 		useContext(alertContext);
@@ -55,20 +57,26 @@ export default function TabsManagerComponent() {
 					flow={null}
 				/>
 				<div className="ml-auto mr-2 flex gap-3">
+					<button
+						onClick={() => openPopUp(<ApiModal flowName={flows[tabIndex].name} />)}
+						className="flex items-center gap-1 pr-2 border-gray-400 border-r text-sm text-gray-600 hover:text-gray-500 dark:text-gray-300 dark:hover:text-gray-200"
+					>
+						Code <CodeBracketSquareIcon className="w-5 h-5" />
+					</button>
 					<button
 						onClick={() => openPopUp(<ImportModal />)}
-						className="flex items-center gap-1 pr-2 border-gray-400 border-r text-sm text-gray-600 hover:text-gray-500"
+						className="flex items-center gap-1 pr-2 border-gray-400 border-r text-sm text-gray-600 hover:text-gray-500 dark:text-gray-300 dark:hover:text-gray-200"
 					>
 						Import <ArrowUpTrayIcon className="w-5 h-5" />
 					</button>
 					<button
 						onClick={() => openPopUp(<ExportModal />)}
-						className="flex items-center gap-1 mr-2 text-sm text-gray-600 hover:text-gray-500"
+						className="flex items-center gap-1 pr-2 text-sm text-gray-600  border-gray-400 border-r hover:text-gray-500 dark:text-gray-300 dark:hover:text-gray-200"
 					>
 						Export <ArrowDownTrayIcon className="h-5 w-5" />
 					</button>
 					<button
-						className="text-gray-600 hover:text-gray-500 "
+						className="text-gray-600 hover:text-gray-500 dark:text-gray-300 dark:hover:text-gray-200"
 						onClick={() => {
 							setDark(!dark);
 						}}
@@ -80,7 +88,7 @@ export default function TabsManagerComponent() {
 						)}
 					</button>
 					<button
-						className="text-gray-600 hover:text-gray-500 relative"
+						className="text-gray-600 hover:text-gray-500 dark:text-gray-300 dark:hover:text-gray-200 relative"
 						onClick={(event: React.MouseEvent<HTMLElement>) => {
 							setNotificationCenter(false);
 							const top = (event.target as Element).getBoundingClientRect().top;
@@ -90,13 +98,11 @@ export default function TabsManagerComponent() {
 								<>
 									<div
 										className="z-10 absolute"
-										style={{ top: top + 20, left: left - AlertWidth }}
+										style={{ top: top + 34, left: left - AlertWidth }}
 									>
 										<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/index.tsx

@@ -17,6 +17,7 @@ import ReactFlow, {
 	OnNodesDelete,
 	SelectionDragHandler,
 } from "reactflow";
+import _ from "lodash";
 import { locationContext } from "../../contexts/locationContext";
 import ExtraSidebar from "./components/extraSidebarComponent";
 import Chat from "../../components/chatComponent";
@@ -34,10 +35,9 @@ const nodeTypes = {
 	genericNode: GenericNode,
 };
 
-var _ = require("lodash");
 
 export default function FlowPage({ flow }: { flow: FlowType }) {
-	let { updateFlow, incrementNodeId} =
+	let { updateFlow, incrementNodeId, disableCP} =
 		useContext(TabsContext);
 	const { types, reactFlowInstance, setReactFlowInstance, templates } =
 		useContext(typesContext);
@@ -46,11 +46,11 @@ export default function FlowPage({ flow }:{flow:FlowType}) {
 	const { undo, redo, canUndo, canRedo, takeSnapshot } = useUndoRedo();
 
 	const onKeyDown = (event: React.KeyboardEvent<HTMLDivElement>) => {
-    if ((event.ctrlKey || event.metaKey) && (event.key === 'c') && lastSelection){
+		if ((event.ctrlKey || event.metaKey) && (event.key === 'c') && lastSelection && !disableCP) {
 			event.preventDefault();
 			setLastCopiedSelection(lastSelection);
 		}
-    if ((event.ctrlKey || event.metaKey) && (event.key === 'v') && lastCopiedSelection){
+		if ((event.ctrlKey || event.metaKey) && (event.key === 'v') && lastCopiedSelection && !disableCP) {
 			event.preventDefault();
 			paste();
 		}

src/frontend/src/png.d.ts

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

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

@@ -1,4 +1,4 @@
-import { ForwardRefExoticComponent, ReactElement, ReactNode } from "react";
+import { ForwardRefExoticComponent, ReactElement, ReactFragment, ReactNode } from "react";
 import { NodeDataType } from "../flow/index";
 export type InputComponentType = {
 	value: string;
@@ -68,7 +68,7 @@ export type FloatComponentType = {
 
 export type TooltipComponentType = {
   children: ReactElement;
-  title: string;
+  title: string | ReactElement;
   placement?:
     | "bottom-end"
     | "bottom-start"

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

@@ -12,6 +12,14 @@ export type TabsContextType = {
 	downloadFlow: (flow: FlowType) => void;
 	uploadFlow: () => void;
 	hardReset: () => void;
+	//disable CopyPaste
+	disableCP: boolean;
+	setDisableCP: (value: boolean) => 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,5 +1,4 @@
-
+const template: { [char: string]: string } = {};
-const template:{[char: string]: string}={}
 
 export type TemplateContextType = {
 	templates: typeof template;

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

@@ -1,9 +1,8 @@
 import { ReactFlowInstance } from "reactflow";
 
 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;

src/frontend/src/utils.ts

@@ -19,7 +19,7 @@ import {
 import { Connection, Edge, Node, ReactFlowInstance } from "reactflow";
 import { FlowType } from "./types/flow";
 import { APITemplateType, TemplateVariableType } from "./types/api";
-var _ = require("lodash");
+import _ from "lodash";
 
 export function classNames(...classes: Array<string>) {
 	return classes.filter(Boolean).join(" ");
@@ -316,6 +316,14 @@ export function toFirstUpperCase(str: string) {
 		.join("");
 }
 
+export function snakeToSpaces(str: string) {
+	let result = str
+		.split("_")
+		.join(" ");
+
+	return result
+}
+
 export function toNormalCase(str: string) {
 	let result = str
 		.split("_")
@@ -327,7 +335,8 @@ export function toNormalCase(str: string) {
 		})
 		.join(" ");
 
-  return result.split("-")
+	return result
+		.split("-")
 		.map((word, index) => {
 			if (index === 0) {
 				return word[0].toUpperCase() + word.slice(1).toLowerCase();
@@ -454,3 +463,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

@@ -1,7 +1,10 @@
 /** @type {import('tailwindcss').Config} */
-const plugin = require("tailwindcss/plugin");
+import plugin from "tailwindcss/plugin";
 module.exports = {
-  content: ["./src/**/*.{js,ts,tsx,jsx}"],
+  content: [
+    "./index.html",
+    "./src/**/*.{js,ts,tsx,jsx}",
+  ],
   darkMode: "class",
   important: true,
   theme: {
@@ -57,6 +60,11 @@ module.exports = {
           "font-family": "text-security-disc"
 
         },
+        '.stop': {
+          '-webkit-animation-play-state': 'paused',
+          '-moz-animation-play-state': 'paused',
+          'animation-play-state': 'paused',
+        },
         '.custom-scroll':{
           '&::-webkit-scrollbar': {
             'width': '8px',
@@ -73,6 +81,6 @@ module.exports = {
         }
       }
     })
-    }),require('@tailwindcss/line-clamp')
+    }),require('@tailwindcss/line-clamp'),require('@tailwindcss/typography'),
   ],
 };

src/frontend/vite.config.ts

@@ -0,0 +1,34 @@
+import { defineConfig } from 'vite';
+import react from '@vitejs/plugin-react-swc';
+
+const apiRoutes = [
+    '/all',
+    '/predict',
+    '^/validate/*',
+    '^/chat/*',
+];
+
+const proxyTargets = apiRoutes.reduce((proxyObj, route) => {
+    proxyObj[route] = {
+        target: 'http://127.0.0.1:7860',
+        changeOrigin: true,
+        secure: false,
+        ws: true,
+    };
+    return proxyObj;
+}, {});
+
+export default defineConfig(() => {
+    return {
+        build: {
+            outDir: 'build',
+        },
+        plugins: [react()],
+        server: {
+            port: 3000,
+            proxy: {
+                ...proxyTargets
+            }
+        },
+    };
+});

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