diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 58b56354b..db78822a4 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -327,6 +327,7 @@ export default function ParameterComponent({
                 disabled={disabled}
                 value={data.node.template[name].value ?? ""}
                 onChange={handleOnNewValue}
+                id={"textarea-" + index}
               />
             ) : (
               <InputComponent
@@ -379,6 +380,7 @@ export default function ParameterComponent({
               disabled={disabled}
               value={data.node?.template[name].value ?? ""}
               onChange={handleOnNewValue}
+              id={"code-input-" + index}
             />
           </div>
         ) : left === true && type === "file" ? (
@@ -419,6 +421,7 @@ export default function ParameterComponent({
               onChange={(e) => {
                 handleOnNewValue(e);
               }}
+              id={"prompt-input-" + index}
             />
           </div>
         ) : left === true && type === "NestedDict" ? (
diff --git a/src/frontend/src/components/codeAreaComponent/index.tsx b/src/frontend/src/components/codeAreaComponent/index.tsx
index a402fb3ed..cf7676397 100644
--- a/src/frontend/src/components/codeAreaComponent/index.tsx
+++ b/src/frontend/src/components/codeAreaComponent/index.tsx
@@ -12,6 +12,7 @@ export default function CodeAreaComponent({
   nodeClass,
   dynamic,
   setNodeClass,
+  id = "",
 }: CodeAreaComponentType) {
   const [myValue, setMyValue] = useState(
     typeof value == "string" ? value : JSON.stringify(value)
@@ -41,6 +42,7 @@ export default function CodeAreaComponent({
       >
         <div className="flex w-full items-center">
           <span
+            id={id}
             className={
               editNode
                 ? "input-edit-node input-dialog"
diff --git a/src/frontend/src/components/promptComponent/index.tsx b/src/frontend/src/components/promptComponent/index.tsx
index 5d6c43987..944138ca6 100644
--- a/src/frontend/src/components/promptComponent/index.tsx
+++ b/src/frontend/src/components/promptComponent/index.tsx
@@ -14,6 +14,7 @@ export default function PromptAreaComponent({
   onChange,
   disabled,
   editNode = false,
+  id = "",
 }: PromptAreaComponentType) {
   useEffect(() => {
     if (disabled) {
@@ -35,6 +36,7 @@ export default function PromptAreaComponent({
   return (
     <div className={disabled ? "pointer-events-none w-full " : " w-full"}>
       <GenericModal
+        id={id}
         type={TypeModal.PROMPT}
         value={value}
         buttonText="Check & Save"
@@ -47,6 +49,7 @@ export default function PromptAreaComponent({
       >
         <div className="flex w-full items-center">
           <span
+            id={id}
             className={
               editNode
                 ? "input-edit-node input-dialog"
diff --git a/src/frontend/src/components/textAreaComponent/index.tsx b/src/frontend/src/components/textAreaComponent/index.tsx
index fec59f40e..0be68a294 100644
--- a/src/frontend/src/components/textAreaComponent/index.tsx
+++ b/src/frontend/src/components/textAreaComponent/index.tsx
@@ -10,6 +10,7 @@ export default function TextAreaComponent({
   onChange,
   disabled,
   editNode = false,
+  id = "",
 }: TextAreaComponentType): JSX.Element {
   // Clear text area
   useEffect(() => {
@@ -21,6 +22,7 @@ export default function TextAreaComponent({
   return (
     <div className="flex w-full items-center">
       <Input
+        id={id}
         value={value}
         disabled={disabled}
         className={editNode ? "input-edit-node" : ""}
diff --git a/src/frontend/src/modals/EditNodeModal/index.tsx b/src/frontend/src/modals/EditNodeModal/index.tsx
index cb4d12a5c..524457ccd 100644
--- a/src/frontend/src/modals/EditNodeModal/index.tsx
+++ b/src/frontend/src/modals/EditNodeModal/index.tsx
@@ -203,6 +203,7 @@ const EditNodeModal = forwardRef(
                                       templateParam
                                     ].multiline ? (
                                     <TextAreaComponent
+                                      id={"textarea-edit-" + index}
                                       disabled={disabled}
                                       editNode={true}
                                       value={
@@ -434,6 +435,7 @@ const EditNodeModal = forwardRef(
                                     onChange={(value: string | string[]) => {
                                       handleOnNewValue(value, templateParam);
                                     }}
+                                    id={"prompt-area-edit" + index}
                                   />
                                 </div>
                               ) : myData.current.node?.template[templateParam]
@@ -458,6 +460,7 @@ const EditNodeModal = forwardRef(
                                     onChange={(value: string | string[]) => {
                                       handleOnNewValue(value, templateParam);
                                     }}
+                                    id={"code-area-edit" + index}
                                   />
                                 </div>
                               ) : myData.current.node?.template[templateParam]
diff --git a/src/frontend/src/modals/codeAreaModal/index.tsx b/src/frontend/src/modals/codeAreaModal/index.tsx
index 903fb85b5..0b3fc6179 100644
--- a/src/frontend/src/modals/codeAreaModal/index.tsx
+++ b/src/frontend/src/modals/codeAreaModal/index.tsx
@@ -8,6 +8,7 @@ import { useContext, useEffect, useState } from "react";
 import AceEditor from "react-ace";
 import IconComponent from "../../components/genericIconComponent";
 import { Button } from "../../components/ui/button";
+import { Input } from "../../components/ui/input";
 import { CODE_PROMPT_DIALOG_SUBTITLE } from "../../constants/constants";
 import { alertContext } from "../../contexts/alertContext";
 import { darkContext } from "../../contexts/darkContext";
@@ -143,6 +144,7 @@ export default function CodeAreaModal({
         />
       </BaseModal.Header>
       <BaseModal.Content>
+        <Input value={code} className="absolute left-[500%]" id="codeValue" />
         <div className="flex h-full w-full flex-col transition-all">
           <div className="h-full w-full">
             <AceEditor
@@ -180,7 +182,12 @@ export default function CodeAreaModal({
             </div>
           </div>
           <div className="flex h-fit w-full justify-end">
-            <Button className="mt-3" onClick={handleClick} type="submit">
+            <Button
+              className="mt-3"
+              onClick={handleClick}
+              type="submit"
+              id="checkAndSaveBtn"
+            >
               Check & Save
             </Button>
           </div>
diff --git a/src/frontend/src/modals/genericModal/index.tsx b/src/frontend/src/modals/genericModal/index.tsx
index f26de60fd..4598fa06b 100644
--- a/src/frontend/src/modals/genericModal/index.tsx
+++ b/src/frontend/src/modals/genericModal/index.tsx
@@ -30,6 +30,7 @@ export default function GenericModal({
   nodeClass,
   setNodeClass,
   children,
+  id = "",
 }: genericModalPropsType): JSX.Element {
   const [myButtonText] = useState(buttonText);
   const [myModalTitle] = useState(modalTitle);
@@ -210,6 +211,7 @@ export default function GenericModal({
           >
             {type === TypeModal.PROMPT && isEdit ? (
               <Textarea
+                id={"modal-" + id}
                 ref={divRefPrompt}
                 className="form-input h-full w-full rounded-lg custom-scroll focus-visible:ring-1"
                 value={inputValue}
@@ -284,7 +286,7 @@ export default function GenericModal({
                             className="m-1 max-w-[40vw] cursor-default truncate p-2.5 text-sm"
                           >
                             <div className="relative bottom-[1px]">
-                              <span>
+                              <span id={"badge" + index.toString()}>
                                 {word.replace(/[{}]/g, "").length > 59
                                   ? word.replace(/[{}]/g, "").slice(0, 56) +
                                     "..."
@@ -304,6 +306,7 @@ export default function GenericModal({
               )}
             </div>
             <Button
+              id="genericModalBtnSave"
               onClick={() => {
                 switch (myModalType) {
                   case TypeModal.TEXT:
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index 0ed800aee..ee6818fdd 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -83,6 +83,7 @@ export type TextAreaComponentType = {
   onChange: (value: string[] | string) => void;
   value: string;
   editNode?: boolean;
+  id?: string;
 };
 
 export type PromptAreaComponentType = {
@@ -93,6 +94,7 @@ export type PromptAreaComponentType = {
   onChange: (value: string[] | string) => void;
   value: string;
   editNode?: boolean;
+  id?: string;
 };
 
 export type CodeAreaComponentType = {
@@ -103,6 +105,7 @@ export type CodeAreaComponentType = {
   nodeClass?: APIClassType;
   setNodeClass?: (value: APIClassType) => void;
   dynamic?: boolean;
+  id?: string;
 };
 
 export type FileComponentType = {
@@ -489,6 +492,7 @@ export type genericModalPropsType = {
   nodeClass?: APIClassType;
   setNodeClass?: (Class: APIClassType) => void;
   children: ReactNode;
+  id?: string;
 };
 
 export type buttonBoxPropsType = {
diff --git a/src/frontend/tests/codeAreaModalComponent.spec.ts b/src/frontend/tests/codeAreaModalComponent.spec.ts
new file mode 100644
index 000000000..c063a21d6
--- /dev/null
+++ b/src/frontend/tests/codeAreaModalComponent.spec.ts
@@ -0,0 +1,144 @@
+import { expect, test } from "@playwright/test";
+
+test("CodeAreaModalComponent", async ({ page }) => {
+  await page.goto("http://localhost:3000/");
+  await page.waitForTimeout(2000);
+
+  await page.locator('//*[@id="new-project-btn"]').click();
+  await page.waitForTimeout(2000);
+
+  await page.getByPlaceholder("Search").click();
+  await page.getByPlaceholder("Search").fill("pythonfunctiontool");
+
+  await page.waitForTimeout(2000);
+
+  await page
+    .locator('//*[@id="sidePythonFunctionTool"]')
+    .dragTo(page.locator('//*[@id="react-flow-id"]'));
+  await page.mouse.up();
+  await page.mouse.down();
+
+  await page.locator('//*[@id="code-input-0"]').click();
+
+  let value = await page.locator('//*[@id="codeValue"]').inputValue();
+
+  if (
+    value !=
+    'def python_function(text: str) -> str:    """This is a default python function that returns the input text"""    return text'
+  ) {
+    expect(false).toBeTruthy();
+  }
+
+  await page.locator('//*[@id="checkAndSaveBtn"]').click();
+
+  await page
+    .locator('//*[@id="react-flow-id"]/div[1]/div[1]/div[1]/div/div[2]/div')
+    .click();
+  await page.locator('//*[@id="advancedIcon"]').click();
+  await page.locator('//*[@id="editAdvancedBtn"]').click();
+
+  await page.locator('//*[@id="showcode"]').click();
+  expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showdescription"]').click();
+  expect(
+    await page.locator('//*[@id="showdescription"]').isChecked()
+  ).toBeFalsy();
+
+  await page.locator('//*[@id="showname"]').click();
+  expect(await page.locator('//*[@id="showname"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showreturn_direct"]').click();
+  expect(
+    await page.locator('//*[@id="showreturn_direct"]').isChecked()
+  ).toBeFalsy();
+
+  await page.locator('//*[@id="showcode"]').click();
+  expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showdescription"]').click();
+  expect(
+    await page.locator('//*[@id="showdescription"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showname"]').click();
+  expect(await page.locator('//*[@id="showname"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showreturn_direct"]').click();
+  expect(
+    await page.locator('//*[@id="showreturn_direct"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showcode"]').click();
+  expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showdescription"]').click();
+  expect(
+    await page.locator('//*[@id="showdescription"]').isChecked()
+  ).toBeFalsy();
+
+  await page.locator('//*[@id="showname"]').click();
+  expect(await page.locator('//*[@id="showname"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showreturn_direct"]').click();
+  expect(
+    await page.locator('//*[@id="showreturn_direct"]').isChecked()
+  ).toBeFalsy();
+
+  await page.locator('//*[@id="showcode"]').click();
+  expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showdescription"]').click();
+  expect(
+    await page.locator('//*[@id="showdescription"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showname"]').click();
+  expect(await page.locator('//*[@id="showname"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showreturn_direct"]').click();
+  expect(
+    await page.locator('//*[@id="showreturn_direct"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showcode"]').click();
+  expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="saveChangesBtn"]').click();
+
+  const plusButtonLocator = page.locator('//*[@id="code-input-0"]');
+  const elementCount = await plusButtonLocator.count();
+  if (elementCount === 0) {
+    expect(true).toBeTruthy();
+
+    await page
+      .locator('//*[@id="react-flow-id"]/div[1]/div[1]/div[1]/div/div[2]/div')
+      .click();
+    await page.locator('//*[@id="advancedIcon"]').click();
+    await page.locator('//*[@id="editAdvancedBtn"]').click();
+
+    await page.locator('//*[@id="showcode"]').click();
+    expect(await page.locator('//*[@id="showcode"]').isChecked()).toBeTruthy();
+
+    await page
+      .locator(
+        '//*[@id="radix-:r2l:"]/div[2]/div/div[2]/div/div/div/table/tbody/tr[1]/td[2]/div/div/button/div/span'
+      )
+      .click();
+
+    let value = await page.locator('//*[@id="codeValue"]').inputValue();
+
+    if (
+      value !=
+      'def python_function(text: str) -> str:    """This is a default python function that returns the input text"""    return text'
+    ) {
+      expect(false).toBeTruthy();
+    }
+
+    await page.locator('//*[@id="checkAndSaveBtn"]').click();
+
+    await page.locator('//*[@id="saveChangesBtn"]').click();
+
+    await page.locator('//*[@id="code-input-0"]').click();
+  }
+});
diff --git a/src/frontend/tests/promptModalComponent.spec.ts b/src/frontend/tests/promptModalComponent.spec.ts
new file mode 100644
index 000000000..2ea2af4df
--- /dev/null
+++ b/src/frontend/tests/promptModalComponent.spec.ts
@@ -0,0 +1,174 @@
+import { expect, test } from "@playwright/test";
+
+test("PromptTemplateComponent", async ({ page }) => {
+  await page.goto("http://localhost:3000/");
+  await page.waitForTimeout(2000);
+
+  await page.locator('//*[@id="new-project-btn"]').click();
+  await page.waitForTimeout(2000);
+
+  await page.getByPlaceholder("Search").click();
+  await page.getByPlaceholder("Search").fill("promptTemplate");
+
+  await page.waitForTimeout(2000);
+
+  await page
+    .locator('//*[@id="sidePromptTemplate"]')
+    .dragTo(page.locator('//*[@id="react-flow-id"]'));
+  await page.mouse.up();
+  await page.mouse.down();
+
+  await page.locator('//*[@id="prompt-input-3"]').click();
+  await page
+    .locator('//*[@id="modal-prompt-input-3"]')
+    .fill("{prompt} example {prompt1}");
+
+  let value = await page
+    .locator('//*[@id="modal-prompt-input-3"]')
+    .inputValue();
+
+  if (value != "{prompt} example {prompt1}") {
+    expect(false).toBeTruthy();
+  }
+
+  let valueBadgeOne = await page.locator('//*[@id="badge0"]').innerText();
+  if (valueBadgeOne != "prompt") {
+    expect(false).toBeTruthy();
+  }
+
+  let valueBadgeTwo = await page.locator('//*[@id="badge1"]').innerText();
+  if (valueBadgeTwo != "prompt1") {
+    expect(false).toBeTruthy();
+  }
+
+  await page.locator('//*[@id="genericModalBtnSave"]').click();
+
+  await page.locator('//*[@id="textarea-6"]').click();
+  await page.locator('//*[@id="textarea-6"]').fill("prompt_value_!@#!@#");
+
+  value = await page.locator('//*[@id="textarea-6"]').inputValue();
+
+  if (value != "prompt_value_!@#!@#") {
+    expect(false).toBeTruthy();
+  }
+
+  await page.locator('//*[@id="textarea-7"]').click();
+  await page
+    .locator('//*[@id="textarea-7"]')
+    .fill("prompt_name_test_123123!@#!@#");
+
+  value = await page.locator('//*[@id="textarea-7"]').inputValue();
+
+  if (value != "prompt_name_test_123123!@#!@#") {
+    expect(false).toBeTruthy();
+  }
+
+  value = await page.locator('//*[@id="prompt-input-3"]').innerText();
+
+  if (value != "{prompt} example {prompt1}") {
+    expect(false).toBeTruthy();
+  }
+
+  await page.locator('//*[@id="editAdvancedIcon"]').click();
+
+  value = await page.locator('//*[@id="textarea-edit-1"]').inputValue();
+
+  if (value != "prompt_value_!@#!@#") {
+    expect(false).toBeTruthy();
+  }
+
+  value = await page.locator('//*[@id="textarea-edit-2"]').inputValue();
+
+  if (value != "prompt_name_test_123123!@#!@#") {
+    expect(false).toBeTruthy();
+  }
+
+  value = await page.locator('//*[@id="prompt-area-edit0"]').innerText();
+
+  if (value != "{prompt} example {prompt1}") {
+    expect(false).toBeTruthy();
+  }
+
+  await page
+    .locator('//*[@id="textarea-edit-2"]')
+    .fill("prompt_edit_test_12312312321!@#$");
+  await page
+    .locator('//*[@id="textarea-edit-1"]')
+    .fill("prompt_edit_test_44444444444!@#$");
+
+  await page.locator('//*[@id="showtemplate"]').click();
+  expect(await page.locator('//*[@id="showtemplate"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showprompt"]').click();
+  expect(await page.locator('//*[@id="showprompt"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showprompt1"]').click();
+  expect(await page.locator('//*[@id="showprompt1"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showtemplate"]').click();
+  expect(
+    await page.locator('//*[@id="showtemplate"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showprompt"]').click();
+  expect(await page.locator('//*[@id="showprompt"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showprompt1"]').click();
+  expect(await page.locator('//*[@id="showprompt1"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="showtemplate"]').click();
+  expect(await page.locator('//*[@id="showtemplate"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showprompt"]').click();
+  expect(await page.locator('//*[@id="showprompt"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showprompt1"]').click();
+  expect(await page.locator('//*[@id="showprompt1"]').isChecked()).toBeFalsy();
+
+  await page.locator('//*[@id="showtemplate"]').click();
+  expect(
+    await page.locator('//*[@id="showtemplate"]').isChecked()
+  ).toBeTruthy();
+
+  await page.locator('//*[@id="showprompt"]').click();
+  expect(await page.locator('//*[@id="showprompt"]').isChecked()).toBeTruthy();
+
+  await page.locator('//*[@id="saveChangesBtn"]').click();
+
+  const plusButtonLocator = page.locator('//*[@id="textarea-7"]');
+  const elementCount = await plusButtonLocator.count();
+  if (elementCount === 0) {
+    expect(true).toBeTruthy();
+
+    await page
+      .locator(
+        '//*[@id="react-flow-id"]/div[1]/div[1]/div[1]/div/div[2]/div/div/div[1]/div/div[1]'
+      )
+      .click();
+
+    await page.locator('//*[@id="editAdvancedIcon"]').click();
+
+    await page.locator('//*[@id="showprompt1"]').click();
+    expect(
+      await page.locator('//*[@id="showprompt1"]').isChecked()
+    ).toBeTruthy();
+
+    value = await page.locator('//*[@id="textarea-edit-1"]').inputValue();
+
+    if (value != "prompt_edit_test_44444444444!@#$") {
+      expect(false).toBeTruthy();
+    }
+
+    value = await page.locator('//*[@id="textarea-edit-2"]').inputValue();
+
+    if (value != "prompt_edit_test_12312312321!@#$") {
+      expect(false).toBeTruthy();
+    }
+
+    value = await page.locator('//*[@id="prompt-area-edit0"]').innerText();
+
+    if (value != "{prompt} example {prompt1}") {
+      expect(false).toBeTruthy();
+    }
+  }
+});