dev(narugo): update GUI

app.py

@@ -1,3 +1,4 @@
+import os
 import re
 from typing import Mapping, Tuple, Dict
 
@@ -152,23 +153,32 @@ def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
 
 
 if __name__ == '__main__':
-    interface = gr.Interface(
-        image_to_wd14_tags,
-        inputs=[
-            gr.Image(type='pil', label='Original Image'),
-            gr.Radio(list(WAIFU_MODELS.keys()), value='wd14-vit', label='Waifu Model'),
-            gr.Slider(0.0, 1.0, value=0.5, label='Tagging Confidence Threshold'),
-            gr.Checkbox(value=False, label='Use Space Instead Of _'),
-            gr.Checkbox(value=True, label='Use Text Escape'),
-            gr.Checkbox(value=False, label='Keep Confidences'),
-            gr.Checkbox(value=True, label='Descend By Confidence'),
-        ],
-        outputs=[
-            gr.Label(label='Ratings'),
-            gr.TextArea(label='Exported Text'),
-            gr.Label(label='Tags'),
-        ],
-        interpretation="default"
-
-    )
-    interface.launch()
+    with gr.Blocks() as demo:
+        with gr.Row():
+            with gr.Column():
+                gr_input_image = gr.Image(type='pil', label='Original Image')
+                with gr.Row():
+                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='wd14-vit', label='Waifu Model')
+                    gr_threshold = gr.Slider(0.0, 1.0, 0.5, label='Tagging Confidence Threshold')
+                with gr.Row():
+                    gr_space = gr.Checkbox(value=False, label='Use Space Instead Of _')
+                    gr_escape = gr.Checkbox(value=True, label='Use Text Escape')
+                    gr_confidence = gr.Checkbox(value=False, label='Keep Confidences')
+                    gr_order = gr.Checkbox(value=True, label='Descend By Confidence')
+
+                gr_btn_submit = gr.Button(value='Tagging', variant='primary')
+
+            with gr.Column():
+                gr_ratings = gr.Label(label='Ratings')
+                with gr.Tabs():
+                    with gr.Tab("Tags"):
+                        gr_tags = gr.Label(label='Tags')
+                    with gr.Tab("Exported Text"):
+                        gr_output_text = gr.TextArea(label='Exported Text')
+
+        gr_btn_submit.click(
+            image_to_wd14_tags,
+            inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape, gr_confidence, gr_order],
+            outputs=[gr_ratings, gr_output_text, gr_tags],
+        )
+    demo.queue(os.cpu_count()).launch()