Update app.py

app.py

@@ -85,7 +85,7 @@ async def load_image_from_url(url):
             return Image.open(io.BytesIO(image_data))
 
 
-async def greet(my_prompt, control_weight, correct_level, padding_ratio, point_style, my_seed, size):
+async def greet(my_prompt, control_weight, correct_level, padding_ratio, hires_rate, point_style, my_seed, size):
     url = API_URL
     headers = {
         'x-qrbtf-key': f'{API_KEY}',
@@ -97,6 +97,8 @@ async def greet(my_prompt, control_weight, correct_level, padding_ratio, point_s
     size_value = sizes[size]['value']
     point_style_value = anchor_styles[point_style]['value']
 
+    is_hires = not (hires_rate < 0.01)
+
     payload = {
         'url': 'https://qrbtf.com/',
         'prompt': my_prompt,
@@ -106,6 +108,8 @@ async def greet(my_prompt, control_weight, correct_level, padding_ratio, point_s
         'point_style': point_style_value,
         'seed': my_seed,
         'size': size_value,
+        'is_hires': is_hires,
+        'hires_rate': hires_rate,
     }
 
     async with aiohttp.ClientSession(headers=headers) as session:
@@ -151,7 +155,6 @@ with gr.Blocks() as demo:
             )
             with gr.Accordion("More options", open=False):
                 with gr.Row():
-                    promptsTuning = gr.Checkbox(label="Prompts tuning", value=True, interactive=True)
                     seed = gr.Slider(
                         label="Seed",
                         minimum=-1,
@@ -160,20 +163,23 @@ with gr.Blocks() as demo:
                         value=-1,
                         interactive=True,
                     )
+                    ControlWeight = gr.Slider(0.5, 1.5, value=1.0, label="ControlNet weight", info="", interactive=True)
 
                 with gr.Row():
-                    ControlWeight = gr.Slider(0.5, 1.5, value=1.0, label="ControlNet weight", info="", interactive=True)
                     marginScale = gr.Slider(0, 0.5, value=0.2, label="Padding ratio", info="", interactive=True)
+                    hiresRate = gr.Slider(0, 0.5, value=0, label="Image restoration rate", info="", interactive=True)
 
                 with gr.Row():
                     SizeSelection = gr.Dropdown(
                         [size['label'] for size in sizes], value=sizes[0]['label'], label="Size", type="index", interactive=True)
                     errorRate = gr.Dropdown(
-                        [level['label'] for level in correct_levels], value=correct_levels[0]['label'], label="Error correction", type="index",
+                        [level['label'] for level in correct_levels], value=correct_levels[1]['label'], label="Error correction", type="index",
                         interactive=True)
-                    
-                anchorStyle = gr.Dropdown(
-                    [anchor['label'] for anchor in anchor_styles], value=anchor_styles[0]['label'], label="Anchor style", type="index", interactive=True)
+
+                with gr.Row():
+                    promptsTuning = gr.Checkbox(label="Prompts tuning", value=True, interactive=True)
+                    anchorStyle = gr.Dropdown(
+                        [anchor['label'] for anchor in anchor_styles], value=anchor_styles[0]['label'], label="Anchor style", type="index", interactive=True)
 
         with gr.Column():
             with gr.Row():
@@ -184,6 +190,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 out = gr.Image(shape=(1, 1))
 
-    btn.click(greet, [prompt, ControlWeight, errorRate, marginScale, anchorStyle, seed, SizeSelection], out)
+    btn.click(greet, [prompt, ControlWeight, errorRate, marginScale, hiresRate, anchorStyle, seed, SizeSelection], out)
 
 demo.launch()