diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 13a9cce89..28db741e6 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -304,7 +304,7 @@ export default function ParameterComponent({
       ref={ref}
       className={
         "relative mt-1 flex w-full flex-wrap items-center justify-between bg-muted px-5 py-2" +
-        (name.includes("code") && type==="code" ? " hidden " : "")
+        ((name==="code" && type==="code") || (name.includes("code") && proxy) ? " hidden " : "")
       }
     >
       <>
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 776bc779e..003d58ed1 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -141,9 +141,8 @@ export default function GenericNode({
     const iconColor = nodeColors[types[data.type]];
     const iconName =
       iconElement || (data.node?.flow ? "group_components" : name);
-    const iconClassName = `generic-node-icon ${
-      !showNode ? " absolute inset-x-6 h-12 w-12 " : ""
-    }`;
+    const iconClassName = `generic-node-icon ${!showNode ? " absolute inset-x-6 h-12 w-12 " : ""
+      }`;
     if (iconElement && isEmoji) {
       return nodeIconFragment(iconElement);
     } else {
@@ -271,7 +270,7 @@ export default function GenericNode({
           numberOfHandles={handles}
           showNode={showNode}
           openAdvancedModal={false}
-          onCloseAdvancedModal={() => {}}
+          onCloseAdvancedModal={() => { }}
           selected={selected}
         ></NodeToolbarComponent>
       </NodeToolbar>
@@ -393,13 +392,28 @@ export default function GenericNode({
                             })}
                             data={data}
                             color={
-                              nodeColors[
-                                types[data.node?.template[templateField].type!]
-                              ] ??
-                              nodeColors[
+                              data.node?.template[templateField].input_types &&
+                                data.node?.template[templateField].input_types!
+                                  .length > 0
+                                ? nodeColors[
+                                data.node?.template[templateField]
+                                  .input_types![data.node?.template[templateField]
+                                    .input_types!.length - 1]
+                                ] ??
+                                nodeColors[
+                                types[
+                                data.node?.template[templateField]
+                                  .input_types![data.node?.template[templateField]
+                                    .input_types!.length - 1]
+                                ]
+                                ]
+                                : nodeColors[
                                 data.node?.template[templateField].type!
-                              ] ??
-                              nodeColors.unknown
+                                ] ??
+                                nodeColors[
+                                types[data.node?.template[templateField].type!]
+                                ] ??
+                                nodeColors.unknown
                             }
                             title={getFieldTitle(
                               data.node?.template!,
@@ -443,7 +457,7 @@ export default function GenericNode({
                     color={nodeColors[types[data.type]] ?? nodeColors.unknown}
                     title={
                       data.node?.output_types &&
-                      data.node.output_types.length > 0
+                        data.node.output_types.length > 0
                         ? data.node.output_types.join(" | ")
                         : data.type
                     }
@@ -464,12 +478,6 @@ export default function GenericNode({
               <Button
                 variant="secondary"
                 className={"group h-9 px-1.5"}
-                onClick={() => {
-                  if (buildStatus === BuildStatus.BUILDING || isBuilding)
-                    return;
-                  setValidationStatus(null);
-                  buildFlow({ nodeId: data.id });
-                }}
               >
                 <div>
                   <ShadTooltip
@@ -482,17 +490,22 @@ export default function GenericNode({
                         <div className="max-h-96 overflow-auto">
                           {typeof validationStatus.params === "string"
                             ? `${durationString}\n${validationStatus.params}`
-                                .split("\n")
-                                .map((line, index) => (
-                                  <div key={index}>{line}</div>
-                                ))
+                              .split("\n")
+                              .map((line, index) => (
+                                <div key={index}>{line}</div>
+                              ))
                             : durationString}
                         </div>
                       )
                     }
                     side="bottom"
                   >
-                    <div className="generic-node-status-position flex items-center justify-center">
+                    <div onClick={() => {
+                      if (buildStatus === BuildStatus.BUILDING || isBuilding)
+                        return;
+                      setValidationStatus(null);
+                      buildFlow({ nodeId: data.id });
+                    }} className="generic-node-status-position flex items-center justify-center">
                       {renderIconStatus(buildStatus, validationStatus)}
                     </div>
                   </ShadTooltip>
@@ -572,7 +585,7 @@ export default function GenericNode({
                   }}
                 >
                   {(data.node?.description === "" || !data.node?.description) &&
-                  nameEditable
+                    nameEditable
                     ? "Double Click to Edit Description"
                     : data.node?.description}
                 </div>
@@ -593,7 +606,7 @@ export default function GenericNode({
                 .map((templateField: string, idx) => (
                   <div key={idx}>
                     {data.node!.template[templateField].show &&
-                    !data.node!.template[templateField].advanced ? (
+                      !data.node!.template[templateField].advanced ? (
                       <ParameterComponent
                         index={idx.toString()}
                         key={scapedJSONStringfy({
@@ -607,25 +620,27 @@ export default function GenericNode({
                         data={data}
                         color={
                           data.node?.template[templateField].input_types &&
-                          data.node?.template[templateField].input_types!
-                            .length > 0
+                            data.node?.template[templateField].input_types!
+                              .length > 0
                             ? nodeColors[
-                                data.node?.template[templateField]
-                                  .input_types![0]
-                              ] ??
-                              nodeColors[
-                                types[
-                                  data.node?.template[templateField]
-                                    .input_types![0]
-                                ]
-                              ]
+                            data.node?.template[templateField]
+                              .input_types![data.node?.template[templateField]
+                                .input_types!.length - 1]
+                            ] ??
+                            nodeColors[
+                            types[
+                            data.node?.template[templateField]
+                              .input_types![data.node?.template[templateField]
+                                .input_types!.length - 1]
+                            ]
+                            ]
                             : nodeColors[
-                                data.node?.template[templateField].type!
-                              ] ??
-                              nodeColors[
-                                types[data.node?.template[templateField].type!]
-                              ] ??
-                              nodeColors.unknown
+                            data.node?.template[templateField].type!
+                            ] ??
+                            nodeColors[
+                            types[data.node?.template[templateField].type!]
+                            ] ??
+                            nodeColors.unknown
                         }
                         title={getFieldTitle(
                           data.node?.template!,
@@ -677,9 +692,9 @@ export default function GenericNode({
                   data={data}
                   color={
                     (data.node?.output_types &&
-                    data.node.output_types.length > 0
+                      data.node.output_types.length > 0
                       ? nodeColors[data.node.output_types[0]] ??
-                        nodeColors[types[data.node.output_types[0]]]
+                      nodeColors[types[data.node.output_types[0]]]
                       : nodeColors[types[data.type]]) ?? nodeColors.unknown
                   }
                   title={
diff --git a/src/frontend/src/components/IOview/index.tsx b/src/frontend/src/components/IOview/index.tsx
index f60479129..56dee33d6 100644
--- a/src/frontend/src/components/IOview/index.tsx
+++ b/src/frontend/src/components/IOview/index.tsx
@@ -18,6 +18,7 @@ import NewChatView from "../newChatView";
 import { Badge } from "../ui/badge";
 import { Button } from "../ui/button";
 import { Tabs, TabsContent, TabsList, TabsTrigger } from "../ui/tabs";
+import { NodeType } from "../../types/flow";
 
 export default function IOView({ children, open, setOpen }): JSX.Element {
   const inputs = useFlowStore((state) => state.inputs).filter(
@@ -51,6 +52,7 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
   const [chatValue, setChatValue] = useState("");
   const isBuilding = useFlowStore((state) => state.isBuilding);
   const currentFlow = useFlowsManagerStore((state) => state.currentFlow);
+  const setNode = useFlowStore((state) => state.setNode);
 
   async function updateVertices() {
     return updateVerticesOrder(currentFlow!.id, null);
@@ -74,6 +76,14 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
       });
     }
     setLockChat(false);
+    if(chatInput) {
+      setNode(chatInput.id, (node:NodeType)=>{
+        
+        const newNode = {...node}
+        newNode.data.node!.template["input_value"].value = chatValue;
+        return newNode;
+      });
+    }
   }
 
   useEffect(() => {
diff --git a/src/frontend/src/modals/EditNodeModal/index.tsx b/src/frontend/src/modals/EditNodeModal/index.tsx
index 93a81724c..e789e1ab8 100644
--- a/src/frontend/src/modals/EditNodeModal/index.tsx
+++ b/src/frontend/src/modals/EditNodeModal/index.tsx
@@ -165,7 +165,8 @@ const EditNodeModal = forwardRef(
                                 )
                             ) ?? false;
                           return (
-                            <TableRow key={index} className="h-10">
+                            <TableRow key={index} className={"h-10 " + ((templateParam==="code" && myData.node?.template[templateParam].type==="code") || (templateParam.includes("code") && myData.node?.template[templateParam].proxy) ? " hidden " : "")
+                          }>
                               <TableCell className="truncate p-0 text-center text-sm text-foreground sm:px-3">
                                 <ShadTooltip
                                   content={