Fix buttons

app.py

@@ -37,9 +37,10 @@ def toggle_slider(checked, value):
         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)
+    reset_values = [gr.update(value=0, interactive=False) for _ in range(4)]
+    if selected_type != "None":
+        index = ["Persuasive", "Descriptive", "Narrative", "Expository"].index(selected_type)
+        reset_values[index] = gr.update(value=0, interactive=True)
     return reset_values
 
 demo = gr.Blocks()
@@ -72,31 +73,31 @@ with demo:
             
             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():
-                    if "Writing Type" in label:
-                        checkbox = gr.Radio(label=label, choices=["None", "Selected"], type="index", value="None")
-                        writing_type_checkboxes.append(checkbox)
-                    else:
+                    if "Writing Type" not in label:
                         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)
+                        non_writing_type_sliders.append(slider)
+                    else:
+                        slider = gr.Slider(label=label, minimum=min_val, maximum=max_val, step=0.01, value=default, interactive=False)
+                        writing_type_sliders.append(slider)
                     
-                    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)
+            
+            writing_type_radio = gr.Radio(
+                label="Writing Type",
+                choices=["None", "Persuasive", "Descriptive", "Narrative", "Expository"],
+                value="None"
+            )
             
             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)
+            writing_type_radio.change(fn=reset_writing_type_sliders, inputs=writing_type_radio, outputs=writing_type_sliders)
 
         with gr.Column(variant="panel"):
             output = gr.Textbox(label="Output")