Add Reset button to clear all inputs and outputs

app.py

@@ -297,6 +297,31 @@ def add_category_field(current_count):
     return updates
 
 
+def reset_all():
+    """Reset all inputs and outputs to initial state."""
+    updates = [
+        None,  # spreadsheet_file
+        gr.update(choices=[], value=None),  # spreadsheet_column
+    ]
+    # Reset category inputs (first 3 visible, rest hidden, all empty)
+    for i in range(MAX_CATEGORIES):
+        updates.append(gr.update(value="", visible=(i < INITIAL_CATEGORIES)))
+    updates.extend([
+        gr.update(visible=True),  # add_category_btn
+        INITIAL_CATEGORIES,  # category_count
+        "Free Models",  # model_tier
+        FREE_MODEL_CHOICES[0],  # model
+        "auto",  # model_source
+        "",  # api_key
+        "**Free tier** - no API key required! We cover the cost while CatLLM is in review.",  # api_key_status
+        "Ready to classify",  # status
+        None,  # results
+        None,  # download_file
+        gr.update(value="", visible=False),  # code_output
+    ])
+    return updates
+
+
 def generate_code(spreadsheet_file, spreadsheet_column,
                   cat1, cat2, cat3, cat4, cat5, cat6, cat7, cat8, cat9, cat10,
                   model_tier, model, model_source_input):
@@ -450,6 +475,7 @@ https://github.com/chrissoria/cat-llm
             with gr.Row():
                 classify_btn = gr.Button("Classify", variant="primary")
                 see_code_btn = gr.Button("See the Code", variant="secondary")
+                reset_btn = gr.Button("Reset", variant="stop")
 
         with gr.Column():
             status = gr.Markdown("Ready to classify")
@@ -507,6 +533,12 @@ https://github.com/chrissoria/cat-llm
         outputs=[code_output]
     )
 
+    reset_btn.click(
+        fn=reset_all,
+        inputs=[],
+        outputs=[spreadsheet_file, spreadsheet_column] + category_inputs + [add_category_btn, category_count, model_tier, model, model_source, api_key, api_key_status, status, results, download_file, code_output]
+    )
+
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)