Fixes

ad-daniel · ad-daniel · commit 56361537e672 · 2023-07-02T23:06:41.000+02:00
diff --git a/webuiapi/webuiapi.py b/webuiapi/webuiapi.py
@@ -111,7 +111,8 @@ def to_dict(self):
 
 
 def b64_img(image: Image) -> str:
-    return  "data:image/png;base64," + raw_b64_img(image)
+    return "data:image/png;base64," + raw_b64_img(image)
+
 
 def raw_b64_img(image: Image) -> str:
     # XXX controlnet only accepts RAW base64 without headers
@@ -230,7 +231,7 @@ async def _to_api_result_async(self, response):
     def txt2img(
         self,
         enable_hr=False,
-        denoising_strength=0.7,
+        denoising_strength=0.0,
         firstphase_width=0,
         firstphase_height=0,
         hr_scale=2,
@@ -625,10 +626,22 @@ def get_samplers(self):
         response = self.session.get(url=f"{self.baseurl}/samplers")
         return response.json()
 
+    def get_vaes(self):
+        response = self.session.get(url=f"{self.baseurl}/sd-vae")
+        return response.json()
+
     def get_upscalers(self):
         response = self.session.get(url=f"{self.baseurl}/upscalers")
         return response.json()
 
+    def get_latent_upscale_modes(self):
+        response = self.session.get(url=f"{self.baseurl}/latent-upscale-modes")
+        return response.json()
+
+    def get_loras(self):
+        response = self.session.get(url=f"{self.baseurl}/loras")
+        return response.json()
+
     def get_sd_models(self):
         response = self.session.get(url=f"{self.baseurl}/sd-models")
         return response.json()
@@ -775,7 +788,7 @@ def util_wait_for_ready(self, check_interval=5.0):
                 time.sleep(check_interval)
 
 
-## Interface for extensions
+# Interface for extensions
 
 
 # https://github.com/mix1009/model-keyword
