diff --git a/packages/frontend/src/pages/AiBuilder/components/AIFormModalContent.tsx b/packages/frontend/src/pages/AiBuilder/components/AIFormModalContent.tsx
index 0d02e8bf01..bb0ef6bb74 100644
--- a/packages/frontend/src/pages/AiBuilder/components/AIFormModalContent.tsx
+++ b/packages/frontend/src/pages/AiBuilder/components/AIFormModalContent.tsx
@@ -6,6 +6,7 @@ import {
   FormControl,
   FormErrorMessage,
   ModalBody,
+  ModalCloseButton,
   ModalFooter,
   ModalHeader,
   Text,
@@ -17,7 +18,14 @@ import { FormLabel, useIsMobile } from '@opengovsg/design-system-react'
 import pairLogo from '@/assets/pair-logo.svg'
 import { ImageBox } from '@/components/FlowStepConfigurationModal/ChooseAndAddConnection/ConfigureExcelConnection'
 import { AI_FORM_SCHEMA, AiFormData } from '@/pages/AiBuilder/schema'
-import { AI_FORM_FIELDS, AI_FORM_IDEAS } from '@/pages/Flows/constants'
+import {
+  AI_FORM_FIELDS,
+  AI_FORM_IDEAS,
+  AiChatIdea,
+  AiFormIdea,
+} from '@/pages/Flows/constants'
+
+import IdeaButtons from './IdeaButtons'
 
 export const AIFormModalContent = ({
   flowName,
@@ -50,17 +58,20 @@ export const AIFormModalContent = ({
     },
   })
 
-  const handleIdeaClick = (idea: (typeof AI_FORM_IDEAS)[number]) => {
-    setValue('trigger', idea.trigger, { shouldValidate: true })
-    setValue('actions', idea.actions, { shouldValidate: true })
+  const handleIdeaClick = (idea: AiChatIdea | AiFormIdea) => {
+    setValue('trigger', (idea as AiFormIdea).trigger, { shouldValidate: true })
+    setValue('actions', (idea as AiFormIdea).actions, { shouldValidate: true })
   }
 
+  const isCreate = type === 'create'
+
   return (
     <>
       <Form onSubmit={handleSubmit(onSubmit)}>
         <ModalHeader p="2.5rem 2rem 1.5rem">
           <Text textStyle="h4">Build with AI</Text>
         </ModalHeader>
+        {!isCreate && <ModalCloseButton onClick={onBack} />}
         <ModalBody>
           <Flex gap={4} flexDir="column">
             {AI_FORM_FIELDS.map((field) => (
@@ -69,73 +80,60 @@ export const AIFormModalContent = ({
                 isInvalid={!!errors[field.key]}
                 key={field.key}
               >
-                <FormLabel>{field.label}</FormLabel>
-                <Textarea
-                  {...register(field.key)}
-                  placeholder={field.placeholder}
-                  resize={field.resize}
-                  minH={field.minH}
-                  maxH={field.maxH}
-                  required={field.required}
-                />
-                {errors[field.key] && (
-                  <FormErrorMessage>
-                    {errors[field.key]?.message}
-                  </FormErrorMessage>
-                )}
+                <Flex gap={2} flexDir="column">
+                  <FormLabel>{field.label}</FormLabel>
+                  <Textarea
+                    {...register(field.key)}
+                    placeholder={field.placeholder}
+                    resize={field.resize}
+                    minH={field.minH}
+                    maxH={field.maxH}
+                    required={field.required}
+                  />
+                  {errors[field.key] && (
+                    <FormErrorMessage>
+                      {errors[field.key]?.message}
+                    </FormErrorMessage>
+                  )}
+                  {isCreate && field.key === 'actions' && (
+                    <IdeaButtons
+                      ideas={AI_FORM_IDEAS}
+                      onClick={(idea: AiChatIdea | AiFormIdea) =>
+                        handleIdeaClick(idea)
+                      }
+                    />
+                  )}
+                </Flex>
               </FormControl>
             ))}
-            <Flex flexDir="row" alignItems="center">
-              <FormLabel
-                isRequired
-                style={{ margin: 0, marginRight: '0.5rem' }}
-              >
-                {/* arbitrary isRequired to hide optional text */}
-                Try:
-              </FormLabel>
-              <Flex
-                flexDir="row"
-                gap={2}
-                justifyContent="space-between"
-                flexWrap="wrap"
-              >
-                {AI_FORM_IDEAS.map((idea) => (
-                  <Button
-                    key={idea.label}
-                    size="sm"
-                    bgColor="interaction.sub-subtle.default"
-                    color="#5D6785"
-                    variant="clear"
-                    _hover={{
-                      bgColor: 'interaction.sub-subtle.hover',
-                    }}
-                    onClick={() => handleIdeaClick(idea)}
-                    px={3}
-                    minH={4}
-                    w={isMobile ? 'calc(50% - 4px)' : 'auto'}
-                    flexShrink={0}
-                  >
-                    <Text textStyle="caption-1">{idea.label}</Text>
-                  </Button>
-                ))}
-              </Flex>
-            </Flex>
           </Flex>
         </ModalBody>
         <ModalFooter>
-          <Flex justifyContent="space-between" alignItems="center" w="100%">
-            <Flex gap={1} alignItems="center">
-              <Text fontSize="xs" color="gray.500">
-                Powered by{' '}
-              </Text>
-              <ImageBox imageUrl={pairLogo} boxSize={6} />
-            </Flex>
+          <Flex
+            justifyContent={isMobile ? 'flex-end' : 'space-between'}
+            alignItems="center"
+            w="100%"
+          >
+            {!isMobile && (
+              <Flex gap={1} alignItems="center" justifyContent="center">
+                <Text fontSize="xs" color="gray.500">
+                  Powered by{' '}
+                </Text>
+                <ImageBox imageUrl={pairLogo} boxSize={6} />
+              </Flex>
+            )}
             <Flex gap={4}>
-              <Button variant="clear" colorScheme="secondary" onClick={onBack}>
-                Back
-              </Button>
+              {isCreate && (
+                <Button
+                  variant="clear"
+                  colorScheme="secondary"
+                  onClick={onBack}
+                >
+                  Back
+                </Button>
+              )}
               <Button type="submit" isDisabled={!isValid}>
-                {type === 'update' ? 'Update workflow' : 'Create'}
+                {isCreate ? 'Create' : 'Update'}
               </Button>
             </Flex>
           </Flex>
diff --git a/packages/frontend/src/pages/AiBuilder/components/ChatInterface/PromptInput.tsx b/packages/frontend/src/pages/AiBuilder/components/ChatInterface/PromptInput.tsx
index b69c084192..fcdf8dee74 100644
--- a/packages/frontend/src/pages/AiBuilder/components/ChatInterface/PromptInput.tsx
+++ b/packages/frontend/src/pages/AiBuilder/components/ChatInterface/PromptInput.tsx
@@ -8,6 +8,7 @@ import {
 import { FaArrowCircleRight } from 'react-icons/fa'
 import { FaCircleStop } from 'react-icons/fa6'
 import { Box, Flex, Icon, Text, Textarea } from '@chakra-ui/react'
+import { useIsMobile } from '@opengovsg/design-system-react'
 
 import pairLogo from '@/assets/pair-logo.svg'
 import { ImageBox } from '@/components/FlowStepConfigurationModal/ChooseAndAddConnection/ConfigureExcelConnection'
@@ -29,6 +30,7 @@ export default function PromptInput({
   sendMessage,
   cancelStream,
 }: PromptInputProps) {
+  const isMobile = useIsMobile()
   const [input, setInput] = useState<string>('')
   const textareaRef = useRef<HTMLTextAreaElement>(null)
 
@@ -74,7 +76,7 @@ export default function PromptInput({
         w="full"
         minH={showIdeas ? '120px' : '50px'}
         height="auto"
-        mb={6}
+        mb={isMobile ? 0 : 6}
       >
         <Textarea
           ref={textareaRef}
@@ -156,12 +158,14 @@ export default function PromptInput({
         />
       )}
 
-      <Flex gap={1} alignItems="center" justify="center" mt={3}>
-        <Text fontSize="xs" color="gray.500">
-          Powered by{' '}
-        </Text>
-        <ImageBox imageUrl={pairLogo} boxSize={6} />
-      </Flex>
+      {!isMobile && (
+        <Flex gap={1} alignItems="center" justify="center" mt={3}>
+          <Text fontSize="xs" color="gray.500">
+            Powered by{' '}
+          </Text>
+          <ImageBox imageUrl={pairLogo} boxSize={6} />
+        </Flex>
+      )}
     </Box>
   )
 }
diff --git a/packages/frontend/src/pages/AiBuilder/components/IdeaButtons.tsx b/packages/frontend/src/pages/AiBuilder/components/IdeaButtons.tsx
index 146f7c2141..bb8cf62cbb 100644
--- a/packages/frontend/src/pages/AiBuilder/components/IdeaButtons.tsx
+++ b/packages/frontend/src/pages/AiBuilder/components/IdeaButtons.tsx
@@ -19,10 +19,11 @@ export default function IdeaButtons({ ideas, onClick }: IdeaButtonsProps) {
         Try:
       </FormLabel>
       <Flex
-        flexDir="row"
+        flexDir={isMobile ? 'column' : 'row'}
         gap={2}
         justifyContent="space-between"
         flexWrap="wrap"
+        w={isMobile ? 'full' : 'auto'}
       >
         {ideas.map((idea) => (
           <Button
@@ -37,7 +38,7 @@ export default function IdeaButtons({ ideas, onClick }: IdeaButtonsProps) {
             onClick={() => onClick(idea)}
             px={3}
             minH={4}
-            w={isMobile ? 'calc(50% - 4px)' : 'auto'}
+            w={isMobile ? 'full' : 'auto'}
             flexShrink={0}
           >
             <TemplateIcon iconName={idea.icon} fontSize="1rem" />
diff --git a/packages/frontend/src/pages/Flows/constants.ts b/packages/frontend/src/pages/Flows/constants.ts
index f682f4010c..74a0ce8ae2 100644
--- a/packages/frontend/src/pages/Flows/constants.ts
+++ b/packages/frontend/src/pages/Flows/constants.ts
@@ -68,14 +68,8 @@ export const AI_CHAT_IDEAS = [
     label: 'Attendance taking',
     icon: 'BiCheckDouble',
     input:
-      'I have a FormSG that I am using to track attendance. I need this workflow to track who has signed up and is present. It should also note down if this person is a new registrant.',
+      "When a new event attendance is received, find the attendee in Tiles. If the attendee is found, update the Attended? column to Yes. If the attendee is not found, create a new row in Tiles with the attendee's details.",
   },
-  // {
-  //   label: 'Send follow ups',
-  //   icon: 'BiEnvelope',
-  //   input:
-  //     'When a new form submission is received, send a follow up email to the customer.',
-  // },
 ]
 
 export type AiChatIdea = {