File size: 1,847 Bytes
03d9a71 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import { memo, useEffect } from 'react';
const Preview = memo(() => {
useEffect(() => {
console.time('🤯 [layout] inject - Split Previewer');
// tab_txt2img
const txt2imgToprow = gradioApp().querySelector('#txt2img_toprow') as HTMLDivElement;
const txt2imgSettings = gradioApp().querySelector('#txt2img_settings') as HTMLDivElement;
const txt2imgGenerate = gradioApp().querySelector('#txt2img_generate_box') as HTMLDivElement;
const txt2imgPreview = gradioApp().querySelector(
'#txt2img_gallery_container',
) as HTMLDivElement;
if (txt2imgToprow && txt2imgSettings && txt2imgGenerate && txt2imgPreview) {
txt2imgSettings.prepend(txt2imgToprow);
txt2imgPreview.prepend(txt2imgGenerate);
}
// tab_img2img
const img2imgToprow = gradioApp().querySelector('#img2img_toprow') as HTMLDivElement;
const img2imgSettings = gradioApp().querySelector('#img2img_settings') as HTMLDivElement;
const img2imgGenerate = gradioApp().querySelector('#img2img_generate_box') as HTMLDivElement;
const img2imgPreview = gradioApp().querySelector(
'#img2img_gallery_container',
) as HTMLDivElement;
if (img2imgSettings && img2imgToprow && img2imgGenerate && img2imgPreview) {
img2imgSettings.prepend(img2imgToprow);
img2imgPreview.prepend(img2imgGenerate);
}
// extras_img2img
const extrasGenerate = gradioApp().querySelector('#extras_generate') as HTMLDivElement;
const extrasPreview = gradioApp().querySelector('#extras_results') as HTMLDivElement;
if (extrasGenerate && extrasPreview) {
(extrasPreview?.parentNode as HTMLDivElement).id = '#extras_gallery_container';
extrasPreview.prepend(extrasGenerate);
}
console.timeEnd('🤯 [layout] inject - Split Previewer');
}, []);
return null;
});
export default Preview;
|