Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
75 changes: 43 additions & 32 deletions app/(main)/page.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -56,45 +56,56 @@ export default function Home() {
scrollTo({ delay: 0.5 });
}

setStatus("creating");
setGeneratedCode("");
try {
setStatus("creating");
setGeneratedCode("");

let res = await fetch("/api/generateCode", {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({
model,
shadcn,
messages: [{ role: "user", content: prompt }],
}),
});
let res = await fetch("/api/generateCode", {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({
model,
shadcn,
messages: [{ role: "user", content: prompt }],
}),
});

if (!res.ok) {
throw new Error(res.statusText);
}
if (!res.ok) {
const errorMessage = await res.text();
setGeneratedCode(`Error: ${errorMessage || res.statusText || 'Something went wrong'}`);
setStatus("initial");
return;
}

if (!res.body) {
throw new Error("No response body");
}
if (!res.body) {
setGeneratedCode("Error: No response received from the server");
setStatus("initial");
return;
}

const reader = res.body.getReader();
let receivedData = "";
const reader = res.body.getReader();
let receivedData = "";

while (true) {
const { done, value } = await reader.read();
if (done) {
break;
while (true) {
const { done, value } = await reader.read();
if (done) {
break;
}
receivedData += new TextDecoder().decode(value);
const cleanedData = removeCodeFormatting(receivedData);
setGeneratedCode(cleanedData);
}
receivedData += new TextDecoder().decode(value);
const cleanedData = removeCodeFormatting(receivedData);
setGeneratedCode(cleanedData);
}

setMessages([{ role: "user", content: prompt }]);
setInitialAppConfig({ model, shadcn });
setStatus("created");
setMessages([{ role: "user", content: prompt }]);
setInitialAppConfig({ model, shadcn });
setStatus("created");
} catch (error) {
console.error('Error generating code:', error);
setGeneratedCode(`Error: ${error instanceof Error ? error.message : 'Something went wrong'}`);
setStatus("initial");
}
}

useEffect(() => {
Expand Down
Loading