diff --git a/index.html b/index.html index 43dfb5f8..ac8655ab 100644 --- a/index.html +++ b/index.html @@ -87,7 +87,8 @@


-
+
+


@@ -118,12 +119,15 @@ let widthField = document.querySelector('#width') let heightField = document.querySelector('#height') let initImageSelector = document.querySelector("#init_image") let initImagePreview = document.querySelector("#init_image_preview") +let promptStrengthField = document.querySelector('#prompt_strength') +let promptStrengthValueLabel = document.querySelector('#prompt_strength_value') let makeImageBtn = document.querySelector('#makeImage') let imagesContainer = document.querySelector('#images') let initImagePreviewContainer = document.querySelector('#init_image_preview_container') let initImageClearBtn = document.querySelector('#init_image_clear') +let promptStrengthContainer = document.querySelector('#prompt_strength_container') let showConfigToggle = document.querySelector('#configToggleBtn') let configBox = document.querySelector('#config') @@ -201,6 +205,7 @@ async function makeImage() { if (initImagePreview.src.indexOf('data:image/png;base64') !== -1) { reqBody['init_image'] = initImagePreview.src + reqBody['prompt_strength'] = promptStrengthField.value / 10 } let res = '' @@ -289,6 +294,7 @@ async function makeImage() { initImagePreview.src = imgBody initImagePreviewContainer.style.display = 'block' + promptStrengthContainer.style.display = 'block' randomSeedField.checked = false seedField.value = seed @@ -320,6 +326,13 @@ function updateGuidanceScale() { guidanceScaleField.addEventListener('input', updateGuidanceScale) updateGuidanceScale() +function updatePromptStrength() { + promptStrengthValueLabel.innerHTML = promptStrengthField.value / 10 +} + +promptStrengthField.addEventListener('input', updatePromptStrength) +updatePromptStrength() + function checkRandomSeed() { if (randomSeedField.checked) { seedField.disabled = true @@ -334,6 +347,7 @@ checkRandomSeed() function showInitImagePreview() { if (initImageSelector.files.length === 0) { initImagePreviewContainer.style.display = 'none' + promptStrengthContainer.style.display = 'none' return } @@ -344,6 +358,7 @@ function showInitImagePreview() { // console.log(file.name, reader.result) initImagePreview.src = reader.result initImagePreviewContainer.style.display = 'block' + promptStrengthContainer.style.display = 'block' }) if (file) { @@ -357,6 +372,7 @@ initImageClearBtn.addEventListener('click', function() { initImageSelector.value = null initImagePreview.src = '' initImagePreviewContainer.style.display = 'none' + promptStrengthContainer.style.display = 'none' }) setInterval(healthCheck, HEALTH_PING_INTERVAL * 1000) diff --git a/main.py b/main.py index 08c10258..41c96db5 100644 --- a/main.py +++ b/main.py @@ -19,6 +19,7 @@ class ImageRequest(BaseModel): width: str = "512" height: str = "512" seed: str = "30000" + prompt_strength: str = "0.8" @app.get('/') def read_root(): @@ -48,6 +49,7 @@ async def image(req : ImageRequest): if req.init_image is not None: data['input']['init_image'] = req.init_image + data['input']['prompt_strength'] = req.prompt_strength if req.seed == "-1": del data['input']['seed']