Implement Polling for Image Generation Status Using Request IDs

[skushagra9]

Instead of polling all the images, and checking all of them if they are Generated or not, once we generate images, pass on the request_ids to the frontend, and use these request_ids to poll the backend,

Here's the backend polling endpoint :

router.get("/poll/images/:requestId", async (req, res) => {
    const requestId = req.params.requestId;
    const request = await prisma.requests.findUnique({
        where: {
            requestId: requestId
        }
    })
    if (request?.status === "completed") {
        const image = request?.imageGenerated;
        res.status(200).json({ message: "success", imageGenerated: image });
    } else {
        res.status(400).json({ message: "pending" });
    }
})

Please note for adding this change, it would require structural DB change, where we will have another Table - Requests

model Requests {
  id             String   @id @default(auto()) @map("_id") @db.ObjectId
  status         String   @default("pending") // pending, completed, failed
  requestId      String   @unique
  stripeId       String
  imageGenerated String?
  createdAt      DateTime @default(now())
  updatedAt      DateTime @updatedAt
}

[skushagra9]

If anyone is interested to pick this up , I can provide help
I have already implemented it in my project
@hkirat @yashmakhija

[hkirat]

yeah this is good

[skushagra9]

yeah this is good

Thanks @hkirat
I have done it exactly what I have shown in the description, and works fine
I can help if anyone interested to solve it