Add captcha refresh

app.py

@@ -14,6 +14,7 @@ from src.display.about import (
     TABLE_DESC,
 )
 from src.display.css_html_js import custom_css
+from src.display.formatting import styled_error, styled_message, styled_warning
 from src.display.utils import (
     BENCHMARK_COLS,
     COLS,
@@ -164,10 +165,11 @@ def generate_captcha(width=200, height=150, length=4):
     return image, text
 
 
-def validate_captcha(input, text):
+def validate_captcha(input, text, img):
     if input == text:
-        return True, "Correct!"
-    return False, "Incorrect! Please re-enter the code."
+        return True, styled_message("Correct!"), text, img
+    img.value, text.value = generate_captcha()
+    return False, styled_error("Incorrect! Please retry with the new code."), text, img
 
 
 
@@ -357,23 +359,24 @@ with demo:
                         captcha_result = gr.Markdown()
                         check_button.click(
                             fn = validate_captcha,
-                            inputs = [captcha_input, text],
-                            outputs = [captcha_correct, captcha_result],
-                        )
-                        submit_button = gr.Button("Submit Eval", interactive=True)
-                        submission_result = gr.Markdown()
-                        submit_button.click(
-                            fn = add_new_eval,
-                            inputs = [
-                                model_name_textbox,
-                                file_input,
-                                precision,
-                                hf_model_id,
-                                contact_email
-                            ],
-                            outputs = [submission_result, model_name_textbox, precision, hf_model_id, contact_email],
+                            inputs = [captcha_input, text, captcha_img],
+                            outputs = [captcha_correct, captcha_result, text, captcha_img],
                         )
 
+                    submit_button = gr.Button("Submit Eval", interactive=True)
+                    submission_result = gr.Markdown()
+                    submit_button.click(
+                        fn = add_new_eval,
+                        inputs = [
+                            model_name_textbox,
+                            file_input,
+                            precision,
+                            hf_model_id,
+                            contact_email
+                        ],
+                        outputs = [submission_result, model_name_textbox, precision, hf_model_id, contact_email],
+                    )
+
     with gr.Row():
         with gr.Accordion("📙 Citation", open=False):
             citation_button = gr.Textbox(