Captcha update

app.py

@@ -156,6 +156,14 @@ def validate_upload(input):
         raise gr.Error("Cannot parse file")
     
 
+def generate_captcha(width=200, height=150, length=4):
+    text = ''.join(random.choices(string.ascii_uppercase + string.digits, k=length))
+    captcha_obj = ImageCaptcha(width, height)
+    data = captcha_obj.generate(text)
+    image = Image.open(data)
+    return image, text
+    
+
 
 demo = gr.Blocks(css=custom_css)
 with demo:
@@ -351,10 +359,7 @@ with demo:
                     )
                     base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
                 """
-            captcha_text = ''.join(random.choices(string.ascii_uppercase + string.digits, k=4))
-            captcha_obj = ImageCaptcha(200, 150)
-            captcha_data = captcha_obj.generate("1234")
-            image = Image.open(captcha_data)
+            image, text = generate_captcha()
             captcha_img = gr.Image(
                 image,
                 container=False,
@@ -362,6 +367,7 @@ with demo:
                 show_fullscreen_button=False,
                 show_share_button=False,
             )
+            captcha_input = gr.Textbox(placeholder="Enter the text in the image above")
             submit_button = gr.Button("Submit Eval", interactive=True)
             submission_result = gr.Markdown()
             submit_button.click(