applied a migration, changed some other things

package.json

@@ -2,10 +2,8 @@
   "private": true,
   "scripts": {
     "dev": "next dev",
-    "build": "next build",
-    "start": "next start",
-    "vercel-build": "prisma generate && next build",
-    "prisma:generate": "prisma generate"
+    "build": "prisma generate && prisma migrate deploy && next build",
+    "start": "next start"
   },
   "dependencies": {
     "@headlessui/react": "^1.7.7",

pages/api/generate.ts

@@ -114,6 +114,10 @@ export default async function handler(
       await new Promise((resolve) => setTimeout(resolve, 1000));
     }
   }
+
+  // add the generated room to the Room Table in the User's account with Prisma
+  // get rid of rate limiting code and decrease the number of credits on the User's table
+
   res.status(200).json(
     generatedImage
       ? {

pages/api/getReplicateInfo.ts

@@ -0,0 +1,37 @@
+import type { NextApiRequest, NextApiResponse } from "next";
+
+export type GenerateResponseData = {
+  original: string | null;
+  generated: string | null;
+  id: string;
+};
+
+interface ExtendedNextApiRequest extends NextApiRequest {
+  body: {
+    imageUrl: string;
+    theme: string;
+    room: string;
+  };
+}
+
+export default async function handler(
+  req: ExtendedNextApiRequest,
+  res: NextApiResponse<GenerateResponseData | string>
+) {
+  // POST request to Replicate to start the image restoration generation process
+  let predictonResponse = await fetch(
+    "https://api.replicate.com/v1/predictions/tfda4hdsgncrpafun4ikywjbve",
+    {
+      method: "GET",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: "Token " + process.env.REPLICATE_API_KEY,
+      },
+    }
+  );
+
+  let jsonStartResponse = await predictonResponse.json();
+
+  console.log({ jsonStartResponse });
+  res.status(200).json(jsonStartResponse);
+}

prisma/migrations/20230317200723_payments/migration.sql

@@ -0,0 +1,23 @@
+-- AlterTable
+ALTER TABLE "User" ADD COLUMN     "boughtCredits" INTEGER NOT NULL DEFAULT 0,
+ADD COLUMN     "credits" INTEGER NOT NULL DEFAULT 5;
+
+-- CreateTable
+CREATE TABLE "Room" (
+    "id" TEXT NOT NULL,
+    "replicateId" TEXT NOT NULL,
+    "userId" TEXT NOT NULL,
+    "inputImage" TEXT NOT NULL,
+    "outputImage" TEXT NOT NULL,
+    "prompt" TEXT NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updatedAt" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "Room_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "Room_replicateId_userId_key" ON "Room"("replicateId", "userId");
+
+-- AddForeignKey
+ALTER TABLE "Room" ADD CONSTRAINT "Room_userId_fkey" FOREIGN KEY ("userId") REFERENCES "User"("id") ON DELETE CASCADE ON UPDATE CASCADE;

prisma/schema.prisma

@@ -40,8 +40,11 @@ model User {
   email         String?   @unique
   emailVerified DateTime?
   image         String?
+  credits       Int       @default(5)
+  boughtCredits Int       @default(0)
   accounts      Account[]
   sessions      Session[]
+  rooms         Room[]
 }
 
 model VerificationToken {
@@ -51,3 +54,17 @@ model VerificationToken {
 
   @@unique([identifier, token])
 }
+
+model Room {
+  id          String   @id @default(cuid())
+  replicateId String
+  userId      String
+  user        User     @relation(fields: [userId], references: [id], onDelete: Cascade)
+  inputImage  String
+  outputImage String
+  prompt      String
+  createdAt   DateTime @default(now())
+  updatedAt   DateTime @updatedAt
+
+  @@unique([replicateId, userId])
+}