fix/AI rejected generation

Dereje1 · Sep 23, 2023 · 7c6b602 · 7c6b602
1 parent 448a20b
commit 7c6b602
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/client/src/components/mypins/pincreatemodal.tsx b/client/src/components/mypins/pincreatemodal.tsx
@@ -174,10 +174,10 @@ class PinCreate extends Component<PinCreateProps, PinCreateState> {
       description: title.trim(),
       AIimageStatus: {
         _id,
-        generatedImage: true,
+        generatedImage: Boolean(_id),
         generatingImage: false,
       },
-    }, updateGeneratedImages);
+    }, () => (_id ? updateGeneratedImages() : null));
   };
 
   handleImageTypes = () => {

diff --git a/server/controllers/post.ts b/server/controllers/post.ts
@@ -105,6 +105,7 @@ export const generateAIimage = async (req: Request, res: genericResponseType) =>
     const [titleObject] = titleResponse.choices;
     res.json({ imgURL: linkObject.url, title: titleObject.text?.trim().replace(/[".]/g, ''), _id });
   } catch (error) {
-    res.json(error);
+    debug(`Error Generating AI image and title for -> UserId -> ${userId} and description: ${description}, Error: ${error}`);
+    res.json({ imgURL: '', title: '', _id: null });
   }
 };
diff --git a/tests/server/controllers/post.test.ts b/tests/server/controllers/post.test.ts
@@ -362,6 +362,10 @@ describe('generating an AI image', () => {
       },
     };
     await generateAIimage(req as genericRequest, res as unknown as Response);
-    expect(res.json).toHaveBeenCalledWith(Error('Mocked rejection'));
+    expect(res.json).toHaveBeenCalledWith({
+      imgURL: '',
+      title: '',
+      _id: null,
+    });
   });
 });