add error control

myapp/app.py

@@ -10,6 +10,7 @@ from gradio.components import Component
 from huggingface_hub import InferenceClient
 from PIL import Image
 from qrcode_artistic import write_artistic
+from segno.consts import ERROR_MAPPING
 
 from myapp.colorutils import array_to_hex
 from myapp.palette import extract_color_clusters, sort_color_clusters
@@ -45,14 +46,15 @@ with gr.Blocks() as demo:
                 str(static_path / "example.webp"), visible=False, type="filepath"
             )
             scale = gr.Slider(3, 15, 9, step=1, label="Scale")
+            error = gr.Radio(list(ERROR_MAPPING), value="H", label="Error")
 
             with gr.Row():
                 color_dark = gr.ColorPicker("#000000", label="Dark")
                 color_light = gr.ColorPicker("#FFFFFF", label="Light")
 
             with gr.Row():
-                gr.ClearButton([color_dark, color_light], value="Reset")
                 extract_colors = gr.Button("Extract")
+                gr.ClearButton([color_dark, color_light], value="Reset")
 
     def generate_background(data: dict[Component, Any]):
         if not data.get(prompt):
@@ -77,7 +79,7 @@ with gr.Blocks() as demo:
             return f"#{r:02X}{g:02X}{b:02X}"
 
         image = Image.open(data[background])
-        qr_code = segno.make(data[text], error="h")
+        qr_code = segno.make(data[text], error=data[error])
 
         with BytesIO() as buffer:
             write_artistic(
@@ -121,6 +123,7 @@ with gr.Blocks() as demo:
             text.submit,
             background.change,
             scale.change,
+            error.change,
             color_light.change,
             color_dark.change,
         ],
@@ -129,6 +132,7 @@ with gr.Blocks() as demo:
             text,
             background,
             scale,
+            error,
             color_light,
             color_dark,
         },