Merge pull request #9 from caranicas/beta-react-creation-options

Beta react creation options
This commit is contained in:
caranicas
2022-09-14 16:50:17 -04:00
committed by GitHub
3 changed files with 13 additions and 7 deletions

View File

@@ -29,6 +29,7 @@ function SettingsList() {
const toggleUseFaceCorrection = useImageCreate( const toggleUseFaceCorrection = useImageCreate(
(state) => state.toggleUseFaceCorrection (state) => state.toggleUseFaceCorrection
); );
const isUsingFaceCorrection = useImageCreate((state) => const isUsingFaceCorrection = useImageCreate((state) =>
state.isUsingFaceCorrection() state.isUsingFaceCorrection()
); );
@@ -111,6 +112,9 @@ function SettingsList() {
name="upscale_model" name="upscale_model"
disabled={!isUsingUpscaling} disabled={!isUsingUpscaling}
value={use_upscale} value={use_upscale}
onChange={(e) => {
setRequestOption("use_upscale", e.target.value);
}}
> >
<option value="RealESRGAN_x4plus">RealESRGAN_x4plus</option> <option value="RealESRGAN_x4plus">RealESRGAN_x4plus</option>
<option value="RealESRGAN_x4plus_anime_6B"> <option value="RealESRGAN_x4plus_anime_6B">

View File

@@ -50,10 +50,11 @@ export default function DisplayPanel() {
}) })
.flat() .flat()
.reverse(); .reverse();
setCompletedImages(temp); setCompletedImages(temp);
} else { } else {
setCompletedImages([]); setCompletedImages([]);
} }
}, [setCompletedImages, queryClient, completedIds]); }, [setCompletedImages, queryClient, completedIds]);
return ( return (

View File

@@ -113,7 +113,7 @@ export const useImageCreate = create<ImageCreateState>(
use_cpu: false, use_cpu: false,
use_full_precision: true, use_full_precision: true,
save_to_disk_path: "null", save_to_disk_path: "null",
use_face_correction: null, use_face_correction: 'GFPGANv1.3',
use_upscale: "RealESRGAN_x4plus", use_upscale: "RealESRGAN_x4plus",
show_only_filtered_image: false, show_only_filtered_image: false,
} as ImageRequest, } as ImageRequest,
@@ -184,6 +184,7 @@ export const useImageCreate = create<ImageCreateState>(
if (!state.uiOptions.isCheckUseFaceCorrection) { if (!state.uiOptions.isCheckUseFaceCorrection) {
request.use_face_correction = null; request.use_face_correction = null;
} }
// if we arent using upscaling clear the upscaling // if we arent using upscaling clear the upscaling
if (!state.uiOptions.isCheckedUseUpscaling) { if (!state.uiOptions.isCheckedUseUpscaling) {
request.use_upscale = null; request.use_upscale = null;
@@ -238,7 +239,7 @@ export const useImageCreate = create<ImageCreateState>(
state.requestOptions.use_upscale = state.uiOptions state.requestOptions.use_upscale = state.uiOptions
.isCheckedUseUpscaling .isCheckedUseUpscaling
? "RealESRGAN_x4plus" ? "RealESRGAN_x4plus"
: null; : undefined;
localStorage.setItem( localStorage.setItem(
"ui:isCheckedUseUpscaling", "ui:isCheckedUseUpscaling",
state.uiOptions.isCheckedUseUpscaling state.uiOptions.isCheckedUseUpscaling
@@ -259,10 +260,10 @@ export const useImageCreate = create<ImageCreateState>(
state.use_face_correction = state.uiOptions.isCheckUseFaceCorrection state.use_face_correction = state.uiOptions.isCheckUseFaceCorrection
? "GFPGANv1.3" ? "GFPGANv1.3"
: null; : null;
localStorage.setItem( // localStorage.setItem(
"ui:isCheckUseFaceCorrection", // "ui:isCheckUseFaceCorrection",
state.uiOptions.isCheckUseFaceCorrection // state.uiOptions.isCheckUseFaceCorrection
); // );
}) })
); );
}, },