diff --git a/apps/dashboard-app/app/(dashboard)/automations/_components/Workflow.tsx b/apps/dashboard-app/app/(dashboard)/automations/_components/Workflow.tsx
index 3c33ffa..ef001e3 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/_components/Workflow.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/_components/Workflow.tsx
@@ -12,7 +12,7 @@ const Workflow = ({workflow}:any) => {
     <Card className=''>
         <CardHeader className=''>
             <CardTitle className='flex items-start justify-between leading-3 '>
-                <div className='flex items-center gap-2 text-button leading-normal'>
+                <div className='flex justify-start items-center flex-wrap gap-2 text-button leading-normal'>
                     {workflow.name}
                     {workflow.publish && <CircleCheckIcon className='w-4 h-4 text-green-400'/>}
                     {!workflow.publish && <CirclePauseIcon className='w-4 h-4 text-red-400'/>}
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/Nodes.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/Nodes.tsx
index 960a0ac..16204b0 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/Nodes.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/Nodes.tsx
@@ -113,10 +113,11 @@ const Nodes = () => {
                 </Label>
                 <Switch id='airplane-mode' onClick={onToggle} defaultChecked={editor.publish!} />
             </div>
-            <div className='flex items-center justify-between gap-2'>
+            <div className='flex items-center gap-2 w-full justify-center'>
                 {showNameEdit ? (
                     <Input
-                        placeholder={editor.name}
+                        className='w-full'
+                        placeholder={name}
                         onChange={(e: any) => setName(e.target.value)}
                         onBlur={handleEditName} // Exit edit mode when clicking outside the input
                         autoFocus // Automatically focus the input when entering edit mode
@@ -128,7 +129,7 @@ const Nodes = () => {
                     </div>
                 )}
             </div>
-            <div className='flex items-center justify-center gap-2 text-sm text-description w-[40%] '>
+            <div className='flex items-center justify-center gap-2 text-sm text-description w-full '>
                 {showDescriptionEdit ? (
                     <Textarea
                         placeholder={editor.description}
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionChildrenComponent.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionChildrenComponent.tsx
new file mode 100644
index 0000000..6fc39f7
--- /dev/null
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionChildrenComponent.tsx
@@ -0,0 +1,64 @@
+import { Button } from '@repo/ui/atoms/shadcn/Button';
+import { Input } from '@repo/ui/atoms/shadcn/Input';
+import ConfirmDialog from '@repo/ui/molecules/custom/ConfirmDialog';
+import { Select, SelectContent, SelectItem, SelectTrigger } from '@repo/ui/molecules/shadcn/Select';
+import { Trash2Icon } from 'lucide-react';
+import React, { useState } from 'react'
+
+const NotionChildrenComponent = ({dbId,access_token,modifyChildrenBody}:any) => {
+   let types = ["embed","callout","table_of_contents","heading_1","heading_2","heading_3","paragraph",
+    "bulleted_list_item","to_do", "numbered_list_item"]
+   const [selectedChildren, setSelectedChildren] = useState<any>([
+    {type: "", value: ""}
+   ]);
+
+   const modifyChild = (index:any, key:any, value:any) => {
+    const pastChildren = [...selectedChildren]
+    pastChildren[index][key] = value
+    setSelectedChildren(pastChildren)
+    modifyChildrenBody(pastChildren)
+   }
+
+   const addChildren = () => {
+    setSelectedChildren([...selectedChildren,{type: "", value: ""}])
+   }
+
+    const removeChildren = (index:any) => {
+     setSelectedChildren(selectedChildren.filter((_:any, i:any) => i !== index))
+    }
+   
+  return (
+    <div className='rounded-md border-border/30 border-2 p-4 flex flex-col gap-4 '>
+        {selectedChildren.map((selectedChild:any,index:any) => (
+            <div key={index} className="grid grid-cols-10 gap-2  items-center">
+                <div>Type {index+1}</div>
+                <Select onValueChange={(value) => modifyChild(index,"type",value)}>
+                    <SelectTrigger className='col-span-4'> 
+                        {selectedChildren[index].type}
+                    </SelectTrigger>
+                    <SelectContent>
+                    {types.map((type) => (
+                        <SelectItem key={type} value={type}>{type}</SelectItem>
+                    ))}
+                    </SelectContent>
+                </Select>
+                <Input 
+                    placeholder='Enter Value'  
+                    className='col-span-4' 
+                    onChange={(e) => modifyChild(index,"value",e.target.value)}
+                />
+                <ConfirmDialog
+                    alertActionFunction={() => removeChildren(index)} 
+                    alertTitle='Delete Children' 
+                    alertDescription='Are you sure you want to delete this children?'
+                    buttonDiv={<Trash2Icon className='w-5 h-5 cursor-pointer col-span-1 text-foreground/50' />}
+                    alertActionText='Delete'
+                />
+            </div>
+        ))}
+        <Button onClick={addChildren} variant="outline" className='mt-4'>Add Children</Button>
+    </div>
+  )
+}
+
+export default NotionChildrenComponent
\ No newline at end of file
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionCode.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionCode.tsx
index 17e9923..6e361d8 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionCode.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionCode.tsx
@@ -11,6 +11,7 @@ import SearchableSelect from '@repo/ui/molecules/custom/SearchableSelect';
 import NotionFilterComponent from './NotionFilterComponent';
 import { Tabs, TabsContent, TabsList, TabsTrigger } from '@repo/ui/molecules/shadcn/Tabs';
 import NotionPropertiesComponent from './NotionPropertiesComponent';
+import NotionChildrenComponent from './NotionChildrenComponent';
 
 
 const NotionCode = () => {
@@ -21,6 +22,7 @@ const NotionCode = () => {
     const [databases, setDatabases] = useState<any>([])
     const [selectedDatabase, setSelectedDatabase] = useState('')
     const [filterBody, setFilterBody] = useState<any>({})
+    const [children, setChildren] = useState<any>({})
     const [properties, setProperties] = useState<any>({})
 
     const [sampleCode, setSampleCode] = useState('');
@@ -32,7 +34,7 @@ const NotionCode = () => {
 
     const fetchSampleQueryDatabaseCode = async () => {
         try {
-            const response = await fetch('/samplePythonCodes/queryNotionDatabase.txt'); // Assuming the file is in the public folder
+            const response = await fetch('/samplePythonCodes/notion/queryNotionDatabase.txt'); // Assuming the file is in the public folder
             let text = await response.text();
             text = text.replaceAll("{{token}}", selectedNotionAccount);
             text = text.replaceAll("{{db_id}}", JSON.parse(selectedDatabase).id.replaceAll("-",""));
@@ -47,7 +49,7 @@ const NotionCode = () => {
 
     const fetchSampleCreatePageCode = async () => {
         try {
-            const response = await fetch('/samplePythonCodes/createNotionPage.txt'); // Assuming the file is in the public folder
+            const response = await fetch('/samplePythonCodes/notion/createNotionPage.txt'); // Assuming the file is in the public folder
             let text = await response.text();
             text = text.replaceAll("{{token}}", selectedNotionAccount);
             text = text.replaceAll("{{db_id}}", JSON.parse(selectedDatabase).id.replaceAll("-",""));
@@ -72,7 +74,7 @@ const NotionCode = () => {
 
     const fetchSampleUpdatePageCode = async () => {
         try {
-            const response = await fetch('/samplePythonCodes/updateNotionPage.txt'); // Assuming the file is in the public folder
+            const response = await fetch('/samplePythonCodes/notion/updateNotionPage.txt'); // Assuming the file is in the public folder
             let text = await response.text();
             text = text.replaceAll("{{token}}", selectedNotionAccount);
 
@@ -96,6 +98,32 @@ const NotionCode = () => {
         }
     };
 
+    const fetchAppendBlockChildren = async () => {
+        try {
+            const response = await fetch('/samplePythonCodes/notion/appendBlockChildren.txt'); // Assuming the file is in the public folder
+            let text = await response.text();
+            text = text.replaceAll("{{token}}", selectedNotionAccount);
+
+            
+            let lines = text.split('\n')
+            let childrenIndex= -1
+            // Find the index of the line where `children = []` is located
+            lines.forEach((line:any, index:any) => {
+                if (line.includes('children = []')) {
+                    childrenIndex = index;
+                }
+                
+            });
+            lines.splice(childrenIndex+1,0,children.join('\n'))
+            setSampleCode(lines.join('\n'));
+
+
+        } catch (error) {
+            console.error('Error fetching sample query:', error);
+            setSampleCode('// Error fetching the sample query.');
+        }
+    };
+
     const modifyFilterBody = (groupCondition:any,filterGroup:any) => {
         let filterBody:any = {}
         filterBody['filter'] = {}
@@ -151,7 +179,7 @@ const NotionCode = () => {
             }
             else if (property.type === "checkbox"){
                 lines.push(
-                    `properties["${property.name}"] = {"checkbox": "${property.value}"}`
+                    `properties["${property.name}"] = {"checkbox": ${property.value}}`
                 )
             }
             else if (property.type === "select"){
@@ -178,6 +206,28 @@ const NotionCode = () => {
         setProperties(lines)
     }
 
+    const modifyChildrenBody = (children:any) => {
+        let lines:any = []
+        children.forEach((child:any) => {
+            if (child.type === "table_of_contents"){
+                lines.push(
+                    `children.append({"type": "${child.type}"`
+                )
+            }
+            else if(child.type === "embed"){
+                lines.push(
+                    `children.append({"type": "${child.type}", "${child.type}": {"url": "${child.value}"}})`
+                )
+            }
+            else {
+                lines.push(
+                    `children.append({"type": "${child.type}", "${child.type}": {"rich_text": [{"type": "text", "text": {"content": "${child.value}"}}]}})`
+                )
+            }
+        })
+        setChildren(lines)
+    }
+
     useEffect(() => {
         const fetchNotionDetails = async () => {
             if (!userId) return;
@@ -225,6 +275,9 @@ const NotionCode = () => {
                         <TabsTrigger key="Properties" value="Properties" className='flex gap-1 border-b-2 shadow-md shadow-border/10 hover:bg-accent ' >
                           <div>Properties</div>
                         </TabsTrigger>
+                        <TabsTrigger key="Children" value="Children" className='flex gap-1 border-b-2 shadow-md shadow-border/10 hover:bg-accent ' >
+                          <div>Children</div>
+                        </TabsTrigger>
                     </TabsList>
                     <TabsContent value='Filters'>
                         <NotionFilterComponent dbId={JSON.parse(selectedDatabase).id.replaceAll("-","")}
@@ -236,6 +289,11 @@ const NotionCode = () => {
                             access_token={selectedNotionAccount}
                             modifyProperties={modifyProperties}/>
                     </TabsContent>
+                    <TabsContent value='Children'>
+                        <NotionChildrenComponent dbId={JSON.parse(selectedDatabase).id.replaceAll("-","")}
+                            access_token={selectedNotionAccount}
+                            modifyChildrenBody={modifyChildrenBody}/>
+                    </TabsContent>
 
                 </Tabs>
             </div>}
@@ -260,6 +318,10 @@ const NotionCode = () => {
                 <Button size="sm" variant="outline"  onClick={fetchSampleUpdatePageCode}>
                     Update Notion Page Sample Code
                 </Button>}
+            {selectedDatabase && 
+                <Button size="sm" variant="outline"  onClick={fetchAppendBlockChildren}>
+                    Append Children Block Sample Code
+                </Button>}
         </div>
         {variable && <input className='p-2 mt-4 w-full' type="text" value={variable} placeholder='Variable Value' />}
         {sampleCode && <textarea className='p-2 mt-4 w-full h-96' value={sampleCode} placeholder='Sample Code' />}
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionFilterComponent.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionFilterComponent.tsx
index 9b58481..b0bcad0 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionFilterComponent.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/NotionFilterComponent.tsx
@@ -1,10 +1,8 @@
 import React, { useEffect, useState } from 'react';
 import { Button } from '@repo/ui/atoms/shadcn/Button';
 import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '@repo/ui/molecules/shadcn/Select';
-import { Checkbox } from '@repo/ui/atoms/shadcn/Checkbox';
 import { Input } from '@repo/ui/atoms/shadcn/Input';
 import { queryNotionDatabaseProperties } from '../../../../../../../actions/notion/notion';
-import { Switch } from '@repo/ui/molecules/shadcn/Switch';
 import { CircleXIcon, RecycleIcon, Trash2Icon } from 'lucide-react';
 import ConfirmDialog from '@repo/ui/molecules/custom/ConfirmDialog';
 
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/OpenAICode.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/OpenAICode.tsx
index f3e6e3e..f424608 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/OpenAICode.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/OpenAICode.tsx
@@ -6,29 +6,72 @@ import { useSession } from 'next-auth/react';
 import { Label } from '@repo/ui/atoms/shadcn/Label';
 import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '@repo/ui/molecules/shadcn/Select';
 import { getOpenAIConnection } from '../../../../../../../actions/connections/openai-connections';
+import { getAllModels } from '../../../../../../../actions/openai/openai';
+import { Textarea } from '@repo/ui/atoms/shadcn/Textarea';
+import { CopyIcon } from 'lucide-react';
+import { useToast } from '../../../../../../../../hooks/useToast';
 
 const OpenAICode = () => {
     
     //OpenAI Variables
     const [openAIAccounts, setOpenAIAccounts] = useState<any>([])
     const [selectedOpenAIAccount, setSelectedOpenAIAccount] = useState('')
+    const [messages, setMessages] = useState<any>('')
+    let responseFormats:any = ["json_object","text"]
+    const [models, setModels] = useState<any>([])
+    const [selectedModel, setSelectedModel] = useState('gpt-4o')
+    const [selectedResponseFormat, setSelectedResponseFormat] = useState('text')
+    const [selectedSystemInstructions, setSelectedSystemInstructions] = useState('You are a helpful assistant')
+    const [sampleCode, setSampleCode] = useState(''); 
 
     const [variable, setVariable] = useState<any>('');
     const session = useSession();
     const userId = session?.data?.user?.id;
+    const { toast } = useToast();
 
     useEffect(() => {
         const fetchOpenAIDetails = async () => {
             if (!userId) return;
             const res:any = await getOpenAIConnection(userId);
+            console.log(res[0].apiKey)
             setOpenAIAccounts(res);
-            console.log('openaiAccounts',res)
+            const models:any = await getAllModels(res[0].apiKey);
+            console.log(models)
+            setModels(models?.data)
         }
         fetchOpenAIDetails()
     },[userId,selectedOpenAIAccount])
+
+    const fetchChatCompletionCode = async () => {
+        try {
+            const response = await fetch('/samplePythonCodes/openai/chatCompletion.txt'); // Assuming the file is in the public folder
+            let text = await response.text();
+            text = text.replaceAll("{{token}}", selectedOpenAIAccount);
+            text = text.replaceAll("{{model}}", selectedModel);
+            text = text.replaceAll("{{responseFormat}}", selectedResponseFormat);
+            text = text.replaceAll("{{systemInstructions}}", selectedSystemInstructions);
+            text = text.replaceAll("{{messages}}", messages);
+            setSampleCode(text);
+
+        } catch (error) {
+            console.error('Error fetching sample query:', error);
+            setSampleCode('// Error fetching the sample query.');
+        }
+    };
+
+    const copySampleCode = () => {
+        navigator.clipboard.writeText(sampleCode).then(() => {
+            toast({title: "Copied", description: "Copied Sample Code to clipboard!", variant: 'default'})
+        }).catch(err => {
+            toast({title: "Error", description: "Failed to copy code.", variant: 'destructive'})
+            console.error("Error copying code:", err);
+        });
+    };
+
+
   return (
     <div>
-        <div className='flex flex-col gap-4  mt-4 ml-2 w-[80%]'>
+        <div className='flex flex-col gap-4  mt-8 ml-2 w-[80%] text-paragraph'>
             <Label className='ml-2'>OpenAI Account</Label>
             <Select onValueChange={(value) => setSelectedOpenAIAccount(value)} defaultValue={selectedOpenAIAccount}>
                 <SelectTrigger>
@@ -40,6 +83,46 @@ const OpenAICode = () => {
                     ))}
                 </SelectContent>
             </Select>
+            {selectedOpenAIAccount && 
+                <div className='flex flex-col gap-4 mt-2'>
+                    <Label className='ml-2'>Models</Label>
+                    <Select onValueChange={(value) => setSelectedModel(value)} defaultValue={selectedModel}>
+                        <SelectTrigger>
+                            <SelectValue placeholder='Select Model'/>  
+                        </SelectTrigger>
+                        <SelectContent>
+                            {models?.map((model:any) => (
+                                <SelectItem key={model.id} value={model.id}>{model.id}</SelectItem>
+                            ))}
+                        </SelectContent>
+                    </Select>
+                </div>
+            }
+            {selectedOpenAIAccount && 
+                <div className='flex flex-col gap-4  mt-2'>
+                    <Label className='ml-2'>Response Format</Label>
+                    <Select onValueChange={(value) => setSelectedResponseFormat(value)} defaultValue={selectedResponseFormat}>
+                        <SelectTrigger>
+                            <SelectValue placeholder='Select Response Format'/>  
+                        </SelectTrigger>
+                        <SelectContent>
+                            {responseFormats?.map((responseFormat:any) => (
+                                <SelectItem key={responseFormat} value={responseFormat}>{responseFormat}</SelectItem>
+                            ))}
+                        </SelectContent>
+                    </Select>
+                </div>
+            }
+            {selectedOpenAIAccount && 
+                <div className='flex flex-col gap-4  mt-2'>
+                    <Label className='ml-2'>System Instructions</Label>
+                    <Textarea className='p-2' value={selectedSystemInstructions} onChange={(e) => setSelectedSystemInstructions(e.target.value)} placeholder='System Instructions' />
+                </div>}
+            {selectedOpenAIAccount && 
+                <div className='flex flex-col gap-4  mt-2'>
+                    <Label className='ml-2'>Messages</Label>
+                    <Textarea className='p-2' value={messages} onChange={(e) => setMessages(e.target.value)} placeholder='Enter the Messages' />
+                </div>}
         </div>
         
         <div className='flex gap-2 mt-4 ml-2 items-center'>
@@ -47,11 +130,23 @@ const OpenAICode = () => {
                 <Button size="sm" variant="outline"  onClick={() => setVariable(selectedOpenAIAccount)}>
                     Get Access Token
                 </Button> }
+            {selectedOpenAIAccount && 
+                <Button size="sm" variant="outline"  onClick={fetchChatCompletionCode}>
+                    Get Chat Completion Code
+                </Button> }
         </div>
-            <div className='flex flex-col gap-2 '>
 
-        </div>
-        <input className='p-2 mt-4 w-full' type="text" value={variable} placeholder='Variable Value' />
+        {variable && 
+            <input className='p-2 mt-4 w-full' type="text" value={variable} placeholder='Variable Value' />
+        }
+        {sampleCode && 
+            <div className='flex flex-col relative'>
+                <div className="absolute top-8 right-8 flex gap-2 ">
+                    <CopyIcon onClick={copySampleCode} className='w-5 h-5 cursor-pointer ' />
+                </div>
+                <textarea className='p-2 mt-4 w-full h-96' value={sampleCode} placeholder='Sample Code' />
+            </div>
+        }
     </div>
   )
 }
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/PythonCode.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/PythonCode.tsx
index ebe73a8..a325940 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/PythonCode.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/PythonCode.tsx
@@ -17,6 +17,7 @@ import { ArrowDownFromLineIcon, ForwardIcon, PlayIcon, SquarePlusIcon, TrashIcon
 import ConfirmDialog from '@repo/ui/molecules/custom/ConfirmDialog';
 import CodeConstruction from './CodeConstruction';
 import { add, set } from 'date-fns';
+import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@repo/ui/atoms/shadcn/Tooltip';
 
 
 export const PythonCode = ({funcType,nodeType,type,subType,node}: any) => {
@@ -156,58 +157,102 @@ export const PythonCode = ({funcType,nodeType,type,subType,node}: any) => {
         );
      };
 
+     // Function to copy all code blocks to clipboard
+    const copyAllCode = () => {
+        const allCode = codeBlocks.map((block: any) => block.code).join('\n\n');
+        navigator.clipboard.writeText(allCode).then(() => {
+            toast({title: "Copied", description: "All code blocks copied to clipboard!", variant: 'default'})
+        }).catch(err => {
+            toast({title: "Error", description: "Failed to copy code.", variant: 'destructive'})
+            console.error("Error copying code:", err);
+        });
+    };
 
 
     return (
-        <div className='flex flex-col gap-4'>
+        <div className='relative flex flex-col gap-4'>
+
+              {/* Overlay for loading */}
+            {loading && (
+                <div className="fixed inset-0 bg-gray-700 bg-opacity-50 flex items-center justify-center z-[5000]">
+                    <div className="text-white text-lg">Currently Running Code...</div>
+                </div>
+            )}
             <CodeConstruction/>
-            <div className='flex w-full justify-center gap-4'>
+            <div className='flex flex-wrap w-full justify-center gap-4'>
                 <Button size="lg" variant="default" type="submit" onClick={onSubmit} > Add / Edit Action</Button>
                 <Button size="lg" onClick={runAllCode}>Run all Code Blocks</Button>
+                <Button size="lg" onClick={copyAllCode}>Copy All Code</Button>
                 <Button size="lg" onClick={()=> addCodeBlock(-1)}>Add code block </Button>
                 
             </div>
 
-            {loading && (
-                <Alert>
-                <AlertDescription>
-                    <pre className="whitespace-pre-wrap break-words overflow-auto">
-                        Currently Running Code ......
-                    </pre>
-                </AlertDescription>
-                </Alert>
-            )}
-
             {codeBlocks.map((block: any) => (
                 <div
                 key={block.id}
                 className='relative  gap-4 border rounded p-2 hover:shadow-lg group'
                 >
-                <CodeMirror
-                    value={block.code}
-                    onChange={(value) => modifyCode(block.id, value)}
-                    theme="dark"
-                    height={''}
-                    extensions={[
-                        python(), // Syntax highlighting for Python
-                        autocompletion() // Enable autocompletion
-                      ]}
-                    className="border rounded overflow-auto"
-                />
-
-                {/* Icon buttons, only visible on hover */}
-                <div className="absolute top-3 right-8 flex gap-2 opacity-0 group-hover:opacity-100 transition-opacity">
-                    <PlayIcon size={18} className='cursor-pointer' onClick={() => runCode(block.code)}/>
-                    <ForwardIcon size={18} className='cursor-pointer' onClick={() => runTillCurrentCode(block.id)}/>
-                    <ConfirmDialog
-                        alertActionFunction={() => removeCodeBlock(block.id)} 
-                        alertTitle='Delete Code Block' 
-                        alertDescription='Are you sure you want to delete this codeblock permanently?'
-                        buttonDiv={<TrashIcon size={18} className='cursor-pointer'/>}
-                        alertActionText='Delete'
+                    <CodeMirror
+                        value={block.code}
+                        onChange={(value) => modifyCode(block.id, value)}
+                        theme="dark"
+                        height={''}
+                        extensions={[
+                            python(), // Syntax highlighting for Python
+                            autocompletion() // Enable autocompletion
+                        ]}
+                        className="border rounded overflow-auto"
                     />
-                    <SquarePlusIcon className='cursor-pointer' size={18}  onClick={()=>addCodeBlock(block.id)}/>
-                </div>
+
+                    {/* Icon buttons, only visible on hover */}
+                    <div className="absolute top-3 right-8 flex gap-2 opacity-0 group-hover:opacity-100 transition-opacity">
+                        <TooltipProvider>
+                            <Tooltip>
+                                <TooltipTrigger>
+                                    <PlayIcon size={18} className='cursor-pointer' onClick={() => runCode(block.code)}/>
+                                </TooltipTrigger>
+                                <TooltipContent>
+                                    Run this code block
+                                </TooltipContent>
+                            </Tooltip>
+                        </TooltipProvider>
+                        <TooltipProvider>
+                            <Tooltip>
+                                <TooltipTrigger>
+                                    <ForwardIcon size={18} className='cursor-pointer' onClick={() => runTillCurrentCode(block.id)}/>
+                                </TooltipTrigger>
+                                <TooltipContent>
+                                    Run this + all previous code blocks
+                                </TooltipContent>
+                            </Tooltip>
+                        </TooltipProvider>
+                        <TooltipProvider>
+                            <Tooltip>
+                                <TooltipTrigger>
+                                <ConfirmDialog
+                                    alertActionFunction={() => removeCodeBlock(block.id)} 
+                                    alertTitle='Delete Code Block' 
+                                    alertDescription='Are you sure you want to delete this codeblock permanently?'
+                                    buttonDiv={<TrashIcon size={18} className='cursor-pointer'/>}
+                                    alertActionText='Delete'
+                                />
+                                </TooltipTrigger>
+                                <TooltipContent>
+                                    Delete this code block
+                                </TooltipContent>
+                            </Tooltip>
+                        </TooltipProvider>
+                        <TooltipProvider>
+                            <Tooltip>
+                                <TooltipTrigger>
+                                    <SquarePlusIcon className='cursor-pointer' size={18}  onClick={()=>addCodeBlock(block.id)}/>
+                                </TooltipTrigger>
+                                <TooltipContent>
+                                    Add a empty code block after this
+                                </TooltipContent>
+                            </Tooltip>
+                        </TooltipProvider>
+                    </div>
                 </div>
             ))}
 
diff --git a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/YoutubeCode.tsx b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/YoutubeCode.tsx
index 3e55d17..af536d9 100644
--- a/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/YoutubeCode.tsx
+++ b/apps/dashboard-app/app/(dashboard)/automations/editor/[editorId]/_components/action-forms/code/YoutubeCode.tsx
@@ -52,7 +52,35 @@ const YoutubeCode = () => {
             let text = await response.text();
             text = text.replaceAll("{{client_id}}", youtubeClientId);
             text = text.replaceAll("{{client_secret}}", youtubeClientSecret);
-            text = text.replaceAll("{{access_token}}", selectedYoutubeAccount);
+            text = text.replaceAll("{{refresh_token}}", selectedYoutubeAccount);
+            setSampleCode(text);
+
+        } catch (error) {
+            console.error('Error fetching sample query:', error);
+            setSampleCode('// Error fetching the sample query.');
+        }
+    }
+
+
+    const fetchLikedVideosCode = async () => {
+        try {
+            const response = await fetch('/samplePythonCodes/youtube/getLikedVideos.txt'); // Assuming the file is in the public folder
+            let text = await response.text();
+            text = text.replaceAll("{{client_id}}", youtubeClientId);
+            text = text.replaceAll("{{client_secret}}", youtubeClientSecret);
+            text = text.replaceAll("{{refresh_token}}", selectedYoutubeAccount);
+            setSampleCode(text);
+
+        } catch (error) {
+            console.error('Error fetching sample query:', error);
+            setSampleCode('// Error fetching the sample query.');
+        }
+    }
+
+    const fetchYoutubeTranscriptCode = async () => {
+        try {
+            const response = await fetch('/samplePythonCodes/youtube/getYoutubeTranscript.txt'); // Assuming the file is in the public folder
+            let text = await response.text();
             setSampleCode(text);
 
         } catch (error) {
@@ -88,12 +116,18 @@ const YoutubeCode = () => {
             <Button size="sm" variant="outline"  onClick={() => setVariable(youtubeClientSecret)}>
                 Get Youtube Client Secret
             </Button>
+            {<Button size="sm" variant="outline"  onClick={fetchYoutubeTranscriptCode}>
+                Get Youtube Transcription
+            </Button>}
             {selectedYoutubeAccount && <Button size="sm" variant="outline"  onClick={fetchAccessTokenCode}>
                 Get Access Token
             </Button>}
             {selectedYoutubeAccount && <Button size="sm" variant="outline"  onClick={fetchSubscribedChannelsDataCode}>
                 Get Subscribed Channels
             </Button>}
+            {selectedYoutubeAccount && <Button size="sm" variant="outline"  onClick={fetchLikedVideosCode}>
+                Get Liked Videos
+            </Button>}
         </div>
 
         {variable && <input className='p-2 mt-4 w-full' type="text" value={variable} placeholder='Variable Value' />}
diff --git a/apps/dashboard-app/app/(dashboard)/connections/_components/Connections.tsx b/apps/dashboard-app/app/(dashboard)/connections/_components/Connections.tsx
index 3da629c..6a3fb34 100644
--- a/apps/dashboard-app/app/(dashboard)/connections/_components/Connections.tsx
+++ b/apps/dashboard-app/app/(dashboard)/connections/_components/Connections.tsx
@@ -15,7 +15,7 @@ const Connections = () => {
         Connect all your apps directly from here. You may need to connect these apps regularly to refresh verfication.
       </div>
       <div className='mt-10 grid grid-cols-1 md:grid-cols-2 xl:grid-cols-3 gap-4'>
-        {CONNECTIONS.map((connection) => (
+        {CONNECTIONS.map((connection:any) => (
               <ConnectionCard
                 key={connection.title}
                 connection={connection}
diff --git a/apps/dashboard-app/app/(dashboard)/connections/page.tsx b/apps/dashboard-app/app/(dashboard)/connections/page.tsx
index a86a0f9..aaa5bf5 100644
--- a/apps/dashboard-app/app/(dashboard)/connections/page.tsx
+++ b/apps/dashboard-app/app/(dashboard)/connections/page.tsx
@@ -13,6 +13,7 @@ import { getUserInfo } from '../../actions/connections/user-connections';
 import { onNotionConnection } from '../../actions/connections/notion-connections';
 import { onYoutubeConnection } from '../../actions/connections/youtube-connections';
 import { useToast } from '../../../hooks/useToast';
+import { onDriveConnection } from '../../actions/connections/drive-connections';
 
 const PlannerPage = () => {
   const isMobile = useMedia("(max-width: 1324px)", false);
@@ -51,6 +52,9 @@ const PlannerPage = () => {
         if (type === 'Youtube'){
           response = await onYoutubeConnection({access_token,refresh_token,scopes,userId})
         }
+        if (type === 'Drive'){
+          response = await onDriveConnection({access_token,refresh_token,scopes,userId})
+        }
         if (response?.success){
           toast({
             title: "Success",
diff --git a/apps/dashboard-app/app/actions/connections/common-connection.tsx b/apps/dashboard-app/app/actions/connections/common-connection.tsx
index 0d7c717..3ffc5d0 100644
--- a/apps/dashboard-app/app/actions/connections/common-connection.tsx
+++ b/apps/dashboard-app/app/actions/connections/common-connection.tsx
@@ -11,7 +11,7 @@ export const getConnectionsAction = async(userId:string) => {
 
     const newConnections = await Promise.all(
         userInfo.connections.map(async (connection: any) => {
-            const cons = CONNECTIONS.find((con) => con.title === connection.type);
+            const cons = CONNECTIONS.find((con:any) => con.title === connection.type);
             if (cons) {
                 const newConnection = { ...cons, ...connection };
                 return newConnection;
diff --git a/apps/dashboard-app/app/actions/connections/drive-connections.tsx b/apps/dashboard-app/app/actions/connections/drive-connections.tsx
new file mode 100644
index 0000000..bcdd85f
--- /dev/null
+++ b/apps/dashboard-app/app/actions/connections/drive-connections.tsx
@@ -0,0 +1,43 @@
+'use server'
+
+import { createConnection, getConnectionByAccessTokenAndUserId } from '@repo/prisma-db/repo/connection';
+import { google } from 'googleapis';
+
+
+export const onDriveConnection = async ({access_token, refresh_token, scopes, userId}: any) => {
+    if(access_token && userId){
+        const connected = await getConnectionByAccessTokenAndUserId(access_token,userId)
+        if (connected) return {success: "Drive Connection already exists"}
+        const drive:any = await createConnection({type:'Drive', accessToken: access_token, refreshToken:refresh_token, scopes, userId})
+        if(!drive) return {error: "Drive Connection not created successfully"}
+        return {success: "Drive Connection created successfully", result: drive}
+    
+    }
+    return null
+}
+
+export const createDriveOAuthUrl = async () => {
+    const oauth2Client = new google.auth.OAuth2(
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_ID,
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_SECRET,
+        process.env.NEXT_PUBLIC_DRIVE_REDIRECT_URI
+    );
+
+    // generate a url that asks permissions for Blogger and Google Calendar scopes
+    const scopes = [
+        'https://www.googleapis.com/auth/drive',                 // Full access to Google Drive
+        'https://www.googleapis.com/auth/drive.file',            // Access to files created or opened by the app
+        'https://www.googleapis.com/auth/drive.metadata.readonly', // Read metadata of files in Google Drive
+        'https://www.googleapis.com/auth/drive.readonly',        // Read-only access to Google Drive
+        'https://www.googleapis.com/auth/drive.appdata'          // Access to app-specific data in Google Drive
+    ];
+
+
+    const url = oauth2Client.generateAuthUrl({
+    // 'online' (default) or 'offline' (gets refresh_token)
+        access_type: 'offline',
+        // If you only need one scope you can pass it as a string
+        scope: scopes
+    });
+    return url;
+}
diff --git a/apps/dashboard-app/app/actions/openai/openai.ts b/apps/dashboard-app/app/actions/openai/openai.ts
new file mode 100644
index 0000000..47a63d6
--- /dev/null
+++ b/apps/dashboard-app/app/actions/openai/openai.ts
@@ -0,0 +1,15 @@
+export const getAllModels = async (token:any) => {
+    try{
+        const response = await fetch('https://api.openai.com/v1/models', {
+            method: 'GET',
+            headers: {
+                'Authorization': `Bearer ${token}`,
+            },
+        });
+        const data = await response.json();
+        return data
+    }
+    catch(err){
+        console.log(err)
+    }
+}
\ No newline at end of file
diff --git a/apps/dashboard-app/app/api/callback/drive/route.ts b/apps/dashboard-app/app/api/callback/drive/route.ts
new file mode 100644
index 0000000..7dd6759
--- /dev/null
+++ b/apps/dashboard-app/app/api/callback/drive/route.ts
@@ -0,0 +1,23 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { google } from 'googleapis';
+
+export async function GET(req: NextRequest) {   
+
+    const oauth2Client = new google.auth.OAuth2(
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_ID,
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_SECRET,
+        process.env.NEXT_PUBLIC_DRIVE_REDIRECT_URI
+    );
+    
+    const code = req.nextUrl.searchParams.get('code');
+    const response = await oauth2Client.getToken(code as string);
+    const type = 'Drive';
+    if (response){
+        return NextResponse.redirect(
+            `${process.env.NEXT_PUBLIC_URL}/connections?access_token=${response.tokens.access_token}&refresh_token=${response.tokens.refresh_token}&scopes=${response.tokens.scope}&type=${type}`
+          );
+    }
+
+
+    return NextResponse.redirect(`${process.env.NEXT_PUBLIC_URL}/connections`);
+}
diff --git a/apps/dashboard-app/app/api/oauth/drive/route.ts b/apps/dashboard-app/app/api/oauth/drive/route.ts
new file mode 100644
index 0000000..79314d8
--- /dev/null
+++ b/apps/dashboard-app/app/api/oauth/drive/route.ts
@@ -0,0 +1,30 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { google } from 'googleapis';
+
+
+export async function GET(req: NextRequest) {
+
+    const oauth2Client = new google.auth.OAuth2(
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_ID,
+        process.env.NEXT_PUBLIC_DRIVE_CLIENT_SECRET,
+        process.env.NEXT_PUBLIC_DRIVE_REDIRECT_URI
+    );
+
+    // generate a url that asks permissions for Blogger and Google Calendar scopes
+    const scopes = [
+        'https://www.googleapis.com/auth/drive',                 // Full access to Google Drive
+        'https://www.googleapis.com/auth/drive.file',            // Access to files created or opened by the app
+        'https://www.googleapis.com/auth/drive.metadata.readonly', // Read metadata of files in Google Drive
+        'https://www.googleapis.com/auth/drive.readonly',        // Read-only access to Google Drive
+        'https://www.googleapis.com/auth/drive.appdata'          // Access to app-specific data in Google Drive
+    ];
+
+
+    const url = oauth2Client.generateAuthUrl({
+    // 'online' (default) or 'offline' (gets refresh_token)
+        access_type: 'offline',
+        // If you only need one scope you can pass it as a string
+        scope: scopes
+    });
+    return NextResponse.redirect(url);
+}
\ No newline at end of file
diff --git a/apps/dashboard-app/components/ConnectionCard.tsx b/apps/dashboard-app/components/ConnectionCard.tsx
index 1e66913..5cf79d8 100644
--- a/apps/dashboard-app/components/ConnectionCard.tsx
+++ b/apps/dashboard-app/components/ConnectionCard.tsx
@@ -12,6 +12,7 @@ import { createOpenAIConnection } from '../app/actions/connections/openai-connec
 import { useRouter } from 'next/navigation'
 import { getEnvironmentVariables } from '../app/actions/env/env'
 import { createYoutubeOAuthUrl } from '../app/actions/connections/youtube-connections'
+import { createDriveOAuthUrl } from '../app/actions/connections/drive-connections'
 
 
 const ConnectionCard = ({connection}:any) => {
@@ -44,6 +45,10 @@ const ConnectionCard = ({connection}:any) => {
         setCallbackUrl(res["NEXT_PUBLIC_URL"]+'/api/callback/youtube')
         setOauthUrl(res["NEXT_PUBLIC_YOUTUBE_OAUTH_URL"])
       }
+      else if (connection.title === 'Drive'){
+        setCallbackUrl(res["NEXT_PUBLIC_URL"]+'/api/callback/drive')
+        setOauthUrl(res["NEXT_PUBLIC_DRIVE_OAUTH_URL"])
+      }
     }
     updateEnvironmentVariables()
   },[connection.title])
@@ -55,7 +60,12 @@ const ConnectionCard = ({connection}:any) => {
     }
     else if (connection.title === "Notion"){
       window.location.href = oauthUrl
-    } else {
+    } 
+    else if (connection.title === "Drive"){
+      const url = await createDriveOAuthUrl()
+      window.location.href = url
+    }
+    else {
       console.error('No oauth url found')
     }
   }
diff --git a/apps/dashboard-app/lib/constant.ts b/apps/dashboard-app/lib/constant.ts
index 3ee50fc..755e68d 100644
--- a/apps/dashboard-app/lib/constant.ts
+++ b/apps/dashboard-app/lib/constant.ts
@@ -255,7 +255,7 @@ export const personalInfoItems = [
   {title: 'settings',icon: Settings, }
 ]
 
-export const CONNECTIONS: Connection[] = [
+export const CONNECTIONS: any = [
     {
       title: 'Notion',
       description: 'Create entries in your notion dashboard and automate tasks.',
@@ -320,12 +320,12 @@ export const CONNECTIONS: Connection[] = [
       ]   
     },
     {
-      title: 'Google Drive',
+      title: 'Drive',
       description: 'Connect to your Google Drive',
       image: '/googleDrive.png',
       accessTokenKey: 'accessToken',
       showModal: false,
-      published: false
+      published: true
     },
     {
       title: 'Youtube',
diff --git a/apps/dashboard-app/public/samplePythonCodes/notion/appendBlockChildren.txt b/apps/dashboard-app/public/samplePythonCodes/notion/appendBlockChildren.txt
new file mode 100644
index 0000000..1adf503
--- /dev/null
+++ b/apps/dashboard-app/public/samplePythonCodes/notion/appendBlockChildren.txt
@@ -0,0 +1,16 @@
+#Add a new Notion Block 
+
+notion_url = "https://api.notion.com/v1"
+token = "{{token}}"
+page_id = "{{page_id}}"
+headers = {
+    "Authorization": f"Bearer {token}",
+    "Notion-Version":"2022-02-22",
+    "Content-Type":"application/json"
+}
+notion_page_url = os.path.join(notion_url,'blocks')
+notion_page_id_url = os.path.join(notion_page_url,page_id,'children')
+body = {}
+children = []
+body['children'] = children
+response = requests.request("PATCH",notion_page_id_url,headers=headers,data=json.dumps(body))
\ No newline at end of file
diff --git a/apps/dashboard-app/public/samplePythonCodes/createNotionPage.txt b/apps/dashboard-app/public/samplePythonCodes/notion/createNotionPage.txt
similarity index 100%
rename from apps/dashboard-app/public/samplePythonCodes/createNotionPage.txt
rename to apps/dashboard-app/public/samplePythonCodes/notion/createNotionPage.txt
diff --git a/apps/dashboard-app/public/samplePythonCodes/queryNotionDatabase.txt b/apps/dashboard-app/public/samplePythonCodes/notion/queryNotionDatabase.txt
similarity index 100%
rename from apps/dashboard-app/public/samplePythonCodes/queryNotionDatabase.txt
rename to apps/dashboard-app/public/samplePythonCodes/notion/queryNotionDatabase.txt
diff --git a/apps/dashboard-app/public/samplePythonCodes/updateNotionPage.txt b/apps/dashboard-app/public/samplePythonCodes/notion/updateNotionPage.txt
similarity index 100%
rename from apps/dashboard-app/public/samplePythonCodes/updateNotionPage.txt
rename to apps/dashboard-app/public/samplePythonCodes/notion/updateNotionPage.txt
diff --git a/apps/dashboard-app/public/samplePythonCodes/openai/chatCompletion.txt b/apps/dashboard-app/public/samplePythonCodes/openai/chatCompletion.txt
new file mode 100644
index 0000000..381bc07
--- /dev/null
+++ b/apps/dashboard-app/public/samplePythonCodes/openai/chatCompletion.txt
@@ -0,0 +1,24 @@
+#Chat Completion 
+
+openai_url = "https://api.openai.com/v1/chat/completions"
+token = "{{token}}"
+headers = {
+    "Authorization": f"Bearer {token}",
+    "Content-Type":"application/json"
+}
+model = "{{model}}"
+system_instructions = """{{systemInstructions}}"""
+response_format = "{{responseFormat}}"
+messages = """{{messages}}"""
+
+body = {}
+body["model"] = model
+body["response_format"] = {}
+body["response_format"]["type"] = response_format
+body["messages"] = []
+body["messages"].append({"role": "system", "content": system_instructions})
+body["messages"].append({"role": "user", "content": messages})
+
+response = requests.request("POST",openai_url ,headers=headers,data=json.dumps(body))
+result = response.json()
+content = json.loads(result['choices'][0]['message']['content'])
\ No newline at end of file
diff --git a/apps/dashboard-app/public/samplePythonCodes/openai/listModels.txt b/apps/dashboard-app/public/samplePythonCodes/openai/listModels.txt
new file mode 100644
index 0000000..5b917a5
--- /dev/null
+++ b/apps/dashboard-app/public/samplePythonCodes/openai/listModels.txt
@@ -0,0 +1,8 @@
+#Chat Completion 
+
+openai_url = "https://api.openai.com/v1/models"
+token = "{{token}}"
+headers = {
+    "Authorization": f"Bearer {token}",
+}
+response = requests.request("GET",openai_url ,headers=headers)
diff --git a/apps/dashboard-app/public/samplePythonCodes/youtube/getLikedVideos.txt b/apps/dashboard-app/public/samplePythonCodes/youtube/getLikedVideos.txt
new file mode 100644
index 0000000..d0d950e
--- /dev/null
+++ b/apps/dashboard-app/public/samplePythonCodes/youtube/getLikedVideos.txt
@@ -0,0 +1,61 @@
+#Get Access Token from RefreshToken
+
+url = "https://oauth2.googleapis.com/token"
+client_id="{{client_id}}"
+client_secret="{{client_secret}}"
+refresh_token="{{refresh_token}}"
+grant_type="refresh_token"
+payload = f"client_id={client_id}&client_secret={client_secret}&refresh_token={refresh_token}&grant_type={grant_type}"
+headers = {
+  'Content-Type': 'application/x-www-form-urlencoded'
+}
+response = requests.request("POST", url, headers=headers, data=payload)
+access_token = response.json()['access_token']
+
+
+#Get Liked Videos
+youtube_results = []
+base_url = 'https://www.googleapis.com/youtube/v3'
+headers = {'Authorization': f'Bearer {access_token}'}
+
+next_page_token = None
+while True:
+  # Fetch videos
+  videos_url = f"{base_url}/playlistItems?part=snippet&playlistId=LL&maxResults=50"
+  if next_page_token:
+      videos_url += f"&pageToken={next_page_token}"
+  
+  videos_response = requests.get(videos_url, headers=headers)
+  if videos_response.status_code != 200:
+      raise Exception(f"HTTP error! status: {videos_response.status_code}")
+  
+  videos_data = videos_response.json()
+  # Fetch video details in batches
+  video_ids = ','.join([item['snippet']['resourceId']['videoId'] for item in videos_data['items']])
+  video_details_url = f"{base_url}/videos?part=snippet,statistics,contentDetails&id={video_ids}"
+  
+  video_details_response = requests.get(video_details_url, headers=headers)
+  if video_details_response.status_code != 200:
+      raise Exception(f"HTTP error! status: {video_details_response.status_code}")
+  
+  video_details_data = video_details_response.json()
+  for video in video_details_data['items']:
+      youtube_results.append({
+          'id': video['id'],
+          'publishedAt': video['snippet']['publishedAt'],
+          'channelId': video['snippet']['channelId'],
+          'title': video['snippet']['title'],
+          'description': video['snippet']['description'],
+          'thumbnailId': video['snippet']['thumbnails']['default']['url'],
+          'channelTitle': video['snippet']['channelTitle'],
+          'tags': video['snippet'].get('tags',[]),
+          'duration': video['contentDetails']['duration'],
+          'views': video['statistics'].get('viewCount',0),
+          'likes': video['statistics'].get('likeCount',0),
+          'comments': video['statistics'].get('commentCount',0),
+          'favorites': video['statistics']['favoriteCount'],
+          'language': video['snippet'].get('defaultAudioLanguage','')
+      })
+  next_page_token = videos_data.get('nextPageToken')
+  if not next_page_token:
+      break
\ No newline at end of file
diff --git a/apps/dashboard-app/public/samplePythonCodes/youtube/getYoutubeTranscript.txt b/apps/dashboard-app/public/samplePythonCodes/youtube/getYoutubeTranscript.txt
new file mode 100644
index 0000000..c39684e
--- /dev/null
+++ b/apps/dashboard-app/public/samplePythonCodes/youtube/getYoutubeTranscript.txt
@@ -0,0 +1,8 @@
+video_id = ""
+try:
+    present = True
+    dicts = youtube_transcript_api.YouTubeTranscriptApi.get_transcript(video_id)
+    text = ' '.join([x['text'] for x in dicts])
+except: 
+    present = False
+    text = "Subtitles disabled for this video so cannot summarize"
\ No newline at end of file
diff --git a/apps/flask-worker/app/main/services/pythonCode.py b/apps/flask-worker/app/main/services/pythonCode.py
index 5fa24b0..35ef8ec 100644
--- a/apps/flask-worker/app/main/services/pythonCode.py
+++ b/apps/flask-worker/app/main/services/pythonCode.py
@@ -7,15 +7,25 @@
 import time
 import io
 import sys
+import youtube_transcript_api
+import openai
+import contextlib
+import logging
 
 def pythonCode(code_string):
     print("Executing code")
-    local_namespace = {"os": os, "json": json, "datetime": datetime, "pytz": pytz, "requests": requests, "schedule": schedule, "time": time}
+    logging.info("Executing code")
+    logging.info("Code: " + code_string)
+    local_namespace = {"os": os, "json": json, "datetime": datetime, "pytz": pytz, "requests": requests, "schedule": schedule,
+                       "time": time, "youtube_transcript_api": youtube_transcript_api, "openai": openai}
     output_buffer = io.StringIO()
-    sys.stdout = output_buffer
-    try:
-        exec(code_string,{},local_namespace)
-        result = output_buffer.getvalue()
-    except Exception as e:
-        result = str(e)
-    return {"log":"Executed successfully","result":result}
\ No newline at end of file
+    # Use contextlib to redirect stdout to the output buffer for capturing print statements
+    with contextlib.redirect_stdout(output_buffer):
+        try:
+            exec(code_string, {}, local_namespace)  # Execute the code string within the specified namespace
+            result = output_buffer.getvalue()  # Retrieve output from buffer
+            print("Execution Result:", result)  # Optional: print the execution result
+        except Exception as e:
+            result = f"Error: {e}"  # Capture and format exceptions
+    
+    return {"log": "Executed successfully", "result": result}
\ No newline at end of file
diff --git a/apps/flask-worker/app/requirements.txt b/apps/flask-worker/app/requirements.txt
index e56d1db..6b6a82f 100644
--- a/apps/flask-worker/app/requirements.txt
+++ b/apps/flask-worker/app/requirements.txt
@@ -5,4 +5,6 @@ kafka-python
 psycopg2-binary
 pytz
 schedule
-requests
\ No newline at end of file
+requests
+youtube_transcript_api
+openai
\ No newline at end of file
diff --git a/packages/prisma-db/src/connection.ts b/packages/prisma-db/src/connection.ts
index fd5bfd6..88106ff 100644
--- a/packages/prisma-db/src/connection.ts
+++ b/packages/prisma-db/src/connection.ts
@@ -142,6 +142,19 @@ export const createConnection = async ({type, userId, accessToken, workspaceName
         })
         return connection;
     }
+    else if (type === 'Drive'){
+        const connection = await db.connection.create({
+            data:{
+                name: 'My Drive Account',
+                userId: userId,
+                accessToken: accessToken,
+                refreshToken: refreshToken,
+                scopes: scopes,
+                type: type,
+            }
+        })
+        return connection;
+    }
     return null;
 }
 
diff --git a/packages/ui/src/components/atoms/shadcn/Label.tsx b/packages/ui/src/components/atoms/shadcn/Label.tsx
index 0b4ee1b..ad6bf4c 100644
--- a/packages/ui/src/components/atoms/shadcn/Label.tsx
+++ b/packages/ui/src/components/atoms/shadcn/Label.tsx
@@ -7,7 +7,7 @@ import { cva, type VariantProps } from "class-variance-authority"
 import { cn } from "../../../lib/utils"
 
 const labelVariants = cva(
-  "text-sm font-medium leading-none peer-disabled:cursor-not-allowed peer-disabled:opacity-70"
+  "text-paragraph leading-none peer-disabled:cursor-not-allowed peer-disabled:opacity-70"
 )
 
 const Label = React.forwardRef<