Update app.py

app.py

@@ -71,15 +71,15 @@ class Prodia:
         }
 
     def generate(self, params):
-        response = self._post(f"{self.base}/sdxl/generate", params)
+        response = self._post(f"{self.base}/sd/generate", params)
         return response.json()
 
     def transform(self, params):
-        response = self._post(f"{self.base}/sdxl/transform", params)
+        response = self._post(f"{self.base}/sd/transform", params)
         return response.json()
 
     def controlnet(self, params):
-        response = self._post(f"{self.base}/sdxl/controlnet", params)
+        response = self._post(f"{self.base}/sd/controlnet", params)
         return response.json()
 
     def get_job(self, job_id):
@@ -96,11 +96,11 @@ class Prodia:
         return job_result
 
     def list_models(self):
-        response = self._get(f"{self.base}/sdxl/models")
+        response = self._get(f"{self.base}/sd/models")
         return response.json()
 
     def list_samplers(self):
-        response = self._get(f"{self.base}/sdxl/samplers")
+        response = self._get(f"{self.base}/sd/samplers")
         return response.json()
 
     def _post(self, url, params):
@@ -213,7 +213,7 @@ for model_name in model_list:
     model_names[name_without_ext] = model_name
 
 
-def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, progress=gr.Progress()):
+def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, upscale, seed, progress=gr.Progress()):
     progress(0, desc="Starting")
     time.sleep(2.5)
     progress(0.25, desc="Generating")
@@ -226,6 +226,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
         "cfg_scale": cfg_scale,
         "width": width,
         "height": height,
+        "upscale": upscale,
         "seed": seed
     })
     progress(0.75, desc="Opening image")
@@ -234,7 +235,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
     return [job["imageUrl"]], job["imageUrl"]
 
 
-def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, progress=gr.Progress()):
+def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, upscale, seed, progress=gr.Progress()):
     progress(0, desc="Starting")
     time.sleep(1.5)
     progress(0.10, desc="Uploading input image")
@@ -251,6 +252,7 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
         "cfg_scale": cfg_scale,
         "width": width,
         "height": height,
+        "upscale": upscale,
         "seed": seed
     })
     progress(0.75, desc="Opening image")
@@ -290,10 +292,11 @@ with gr.Blocks(css=css) as demo:
                             with gr.Column(scale=1):
                                 sampler = gr.Dropdown(value="DPM++ 2M Karras", show_label=True, label="Sampling Method",
                                                       choices=prodia_client.list_samplers())
-
+    
                             with gr.Column(scale=1):
                                 steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=25, value=20, step=1)
-
+                            with gr.Column(scale=1):
+                                upscale = gr.Checkbox(label="Upscale", value=False, interactive=True)
                         with gr.Row():
                             with gr.Column(scale=1):
                                 width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
@@ -312,7 +315,7 @@ with gr.Blocks(css=css) as demo:
                     send_to_png = gr.Button(value="Send OUTPUT IMAGE to PNG Info")
                     past_url = gr.Textbox(visible=False, interactive=False)
 
-            text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height,
+            text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, upscale,
                                                seed], outputs=[image_output, past_url], concurrency_limit=64)
 
         with gr.Tab("img2img", id='i2i'):
@@ -336,7 +339,8 @@ with gr.Blocks(css=css) as demo:
 
                             with gr.Column(scale=1):
                                 i2i_steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=25, value=20, step=1)
-
+                            with gr.Column(scale=1):
+                                i2i_upscale = gr.Checkbox(label="Upscale", value=False, interactive=True)
                         with gr.Row():
                             with gr.Column(scale=1):
                                 i2i_width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
@@ -357,7 +361,7 @@ with gr.Blocks(css=css) as demo:
                     i2i_past_url = gr.Textbox(visible=False, interactive=False)
 
             i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt,
-                                                   model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height,
+                                                   model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height, i2i_upscale,
                                                    i2i_seed], outputs=[i2i_image_output, i2i_past_url], concurrency_limit=64)
         send_to_img2img.click(send_to_img2img_def, inputs=past_url, outputs=i2i_image_input)