added componentAIView for collections

1 files changed, 15 insertions, 18 deletions

diff --git a/src/client/views/nodes/ImageBox.tsx b/src/client/views/nodes/ImageBox.tsx
index a1fa9a283..6b9e170c8 100644
--- a/src/client/views/nodes/ImageBox.tsx
+++ b/src/client/views/nodes/ImageBox.tsx
@@ -580,24 +580,21 @@ export class ImageBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
                                 // style={{ alignSelf: 'flex-end' }}
                                 icon={this._regenerateLoading ? <ReactLoading type="spin" color={SettingsManager.userVariantColor} width={16} height={20} /> : <AiOutlineSend />}
                                 iconPlacement="right"
-                                onClick={undoable(
-                                    action(async () => {
-                                        this._regenerateLoading = true;
-                                        SmartDrawHandler.Instance.regenerate([this.Document], undefined, undefined, this._regenInput, true).then(newImgs => {
-                                            if (newImgs[0]) {
-                                                const url = newImgs[0].pathname;
-                                                const imgField = new ImageField(url);
-                                                this._prevImgs.length === 0 &&
-                                                    this._prevImgs.push({ prompt: StrCast(this.dataDoc.ai_firefly_prompt), seed: NumCast(this.dataDoc.ai_firefly_seed), href: this.paths.lastElement(), pathname: field.url.pathname });
-                                                this.dataDoc[this.fieldKey] = imgField;
-                                                this._prevImgs.unshift({ prompt: newImgs[0].prompt, seed: newImgs[0].seed, href: this.paths.lastElement(), pathname: url });
-                                                this._regenerateLoading = false;
-                                                this._regenInput = '';
-                                            }
-                                        });
-                                    }),
-                                    'regenerate image'
-                                )}
+                                onClick={action(async () => {
+                                    this._regenerateLoading = true;
+                                    SmartDrawHandler.Instance.regenerate([this.Document], undefined, undefined, this._regenInput, true).then(newImgs => {
+                                        if (newImgs[0]) {
+                                            const url = newImgs[0].pathname;
+                                            const imgField = new ImageField(url);
+                                            this._prevImgs.length === 0 &&
+                                                this._prevImgs.push({ prompt: StrCast(this.dataDoc.ai_firefly_prompt), seed: this.dataDoc.ai_firefly_seed as number, href: this.paths.lastElement(), pathname: field.url.pathname });
+                                            this.dataDoc[this.fieldKey] = imgField;
+                                            this._prevImgs.unshift({ prompt: newImgs[0].prompt, seed: newImgs[0].seed, href: this.paths.lastElement(), pathname: url });
+                                            this._regenerateLoading = false;
+                                            this._regenInput = '';
+                                        }
+                                    });
+                                })}
                             />
                             <Button
                                 // style={{ alignSelf: 'flex-end' }}