From 4a07665f0fe0d13fb49271b89bd5571924415b92 Mon Sep 17 00:00:00 2001
From: ykethan <ykethan@amazon.com>
Date: Fri, 29 Nov 2024 14:51:05 -0500
Subject: [PATCH] update nextjs AI generate

---
 src/pages/[platform]/ai/set-up-ai/index.mdx | 27 ++++++++++++++++++---
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/src/pages/[platform]/ai/set-up-ai/index.mdx b/src/pages/[platform]/ai/set-up-ai/index.mdx
index 83590478d79..7526472567a 100644
--- a/src/pages/[platform]/ai/set-up-ai/index.mdx
+++ b/src/pages/[platform]/ai/set-up-ai/index.mdx
@@ -381,8 +381,17 @@ export default function App() {
 <Block name="Pages router">
 
 ```tsx title="pages/index.tsx"
-import { Flex, TextAreaField, Loader, Text, View } from "@aws-amplify/ui-react"
-import { useAIConversation } from "@/client";
+import { useAIGeneration } from "@/client";
+import {
+  Button,
+  Flex,
+  Heading,
+  Loader,
+  Text,
+  TextAreaField,
+  View,
+} from "@aws-amplify/ui-react";
+import React from "react";
 
 export default function Page() {
   const [description, setDescription] = React.useState("");
@@ -429,8 +438,18 @@ export default function Page() {
 <Block name="App router">
 
 ```tsx title="app/page.tsx"
-'use client'
-import { useAIConversation } from "@/client";
+"use client";
+import { useAIGeneration } from "@/client";
+import {
+  Button,
+  Flex,
+  Heading,
+  Loader,
+  Text,
+  TextAreaField,
+  View,
+} from "@aws-amplify/ui-react";
+import React from "react";
 
 export default function Page() {
   const [description, setDescription] = React.useState("");