Toggle buttons for Writing Type

app.py

@@ -36,6 +36,12 @@ def toggle_slider(checked, value):
     else:
         return gr.update(value=0, interactive=False)
 
+def reset_writing_type_sliders(selected_type):
+    reset_values = [gr.update(value=0, interactive=False)] * 4
+    if selected_type is not None:
+        reset_values[selected_type] = gr.update(value=0, interactive=True)
+    return reset_values
+
 demo = gr.Blocks()
 
 with demo:
@@ -64,17 +70,34 @@ with demo:
                 ("Writing Type: Expository (None \u2192 More)", 0, 1, 0)
             ]
             
-            for label, min_val, max_val, default in slider_values:
+            non_writing_type_sliders = []
+            writing_type_sliders = []
+            writing_type_checkboxes = []
+            
+            for idx, (label, min_val, max_val, default) in enumerate(slider_values):
                 with gr.Row():
-                    checkbox = gr.Checkbox(label=label)
+                    if "Writing Type" in label:
+                        checkbox = gr.Radio(label=label, choices=["None", "Selected"], type="index", value="None")
+                        writing_type_checkboxes.append(checkbox)
+                    else:
+                        checkbox = gr.Checkbox(label=label)
+                    
                     slider = gr.Slider(label=label, minimum=min_val, maximum=max_val, step=0.01, value=default, interactive=False)
                     checkbox.change(fn=toggle_slider, inputs=[checkbox, gr.State(default)], outputs=slider)
                     sliders.append(slider)
+                    
+                    if "Writing Type" in label:
+                        writing_type_sliders.append(slider)
+                    else:
+                        non_writing_type_sliders.append(slider)
             
             obfuscate_button = gr.Button("Obfuscate Text")
             
             reset_button.click(fn=reset_sliders, inputs=[], outputs=sliders)
             
+            for i, checkbox in enumerate(writing_type_checkboxes):
+                checkbox.change(fn=reset_writing_type_sliders, inputs=[gr.State(i)], outputs=writing_type_sliders)
+
         with gr.Column(variant="panel"):
             output = gr.Textbox(label="Output")
             obfuscate_button.click(fn=greet, inputs=[input_text] + sliders, outputs=output)