From 4063b3ab3bbabc377e5f53f554134ac407110f0e Mon Sep 17 00:00:00 2001
From: Henry <hzj94@hotmail.com>
Date: Mon, 5 Jun 2023 00:49:22 +0100
Subject: [PATCH] add embed code customization

---
 .../src/ui-component/dialog/APICodeDialog.js  | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/packages/ui/src/ui-component/dialog/APICodeDialog.js b/packages/ui/src/ui-component/dialog/APICodeDialog.js
index 94001528199..55aeca75c73 100644
--- a/packages/ui/src/ui-component/dialog/APICodeDialog.js
+++ b/packages/ui/src/ui-component/dialog/APICodeDialog.js
@@ -129,6 +129,51 @@ const embedCode = (chatflowid) => {
 </script>`
 }
 
+const embedCodeCustomization = (chatflowid) => {
+    return `<script type="module">
+    import Chatbot from "https://cdn.jsdelivr.net/npm/flowise-embed@latest/dist/web.js"
+    Chatbot.init({
+        chatflowid: "${chatflowid}",
+        apiHost: "${baseURL}",
+        theme: {
+            button: {
+                backgroundColor: "#3B81F6",
+                right: 20,
+                bottom: 20,
+                size: "medium",
+                iconColor: "white",
+                customIconSrc: "https://raw.githubusercontent.com/walkxcode/dashboard-icons/main/svg/google-messages.svg",
+            },
+            chatWindow: {
+                welcomeMessage: "Hello! This is custom welcome message",
+                backgroundColor: "#ffffff",
+                height: 700,
+                width: 400,
+                poweredByTextColor: "#303235",
+                botMessage: {
+                backgroundColor: "#f7f8ff",
+                textColor: "#303235",
+                showAvatar: true,
+                avatarSrc: "https://raw.githubusercontent.com/zahidkhawaja/langchain-chat-nextjs/main/public/parroticon.png",
+            },
+            userMessage: {
+                backgroundColor: "#3B81F6",
+                textColor: "#ffffff",
+                showAvatar: true,
+                avatarSrc:
+                  "https://raw.githubusercontent.com/zahidkhawaja/langchain-chat-nextjs/main/public/usericon.png",
+            },
+            textInput: {
+                placeholder: "Type your question",
+                backgroundColor: "#ffffff",
+                textColor: "#303235",
+                sendButtonColor: "#3B81F6",
+            },
+        },
+    })
+</script>`
+}
+
 const APICodeDialog = ({ show, dialogProps, onCancel }) => {
     const portalElement = document.getElementById('portal')
     const navigate = useNavigate()
@@ -140,6 +185,7 @@ const APICodeDialog = ({ show, dialogProps, onCancel }) => {
     const [chatflowApiKeyId, setChatflowApiKeyId] = useState('')
     const [selectedApiKey, setSelectedApiKey] = useState({})
     const [checkboxVal, setCheckbox] = useState(false)
+    const [embedChatCheckboxVal, setEmbedChatCheckbox] = useState(false)
 
     const getAllAPIKeysApi = useApi(apiKeyApi.getAllAPIKeys)
     const updateChatflowApi = useApi(chatflowsApi.updateChatflow)
@@ -152,6 +198,10 @@ const APICodeDialog = ({ show, dialogProps, onCancel }) => {
         }
     }
 
+    const onCheckBoxEmbedChatChanged = (newVal) => {
+        setEmbedChatCheckbox(newVal)
+    }
+
     const onApiKeySelected = (keyValue) => {
         if (keyValue === 'addnewkey') {
             navigate('/apikey')
@@ -588,6 +638,22 @@ query({
                                 />
                             </>
                         )}
+                        {value === 0 && (
+                            <CheckboxInput
+                                label='Show Embed Chat Config'
+                                value={embedChatCheckboxVal}
+                                onChange={onCheckBoxEmbedChatChanged}
+                            />
+                        )}
+                        {value === 0 && embedChatCheckboxVal && (
+                            <CopyBlock
+                                theme={atomOneDark}
+                                text={embedCodeCustomization(chatflowApiKeyId)}
+                                language={getLang('Embed')}
+                                showLineNumbers={false}
+                                wrapLines
+                            />
+                        )}
                     </TabPanel>
                 ))}
             </DialogContent>