Update gradio_keylock/component.py

gradio_keylock/component.py

@@ -274,7 +274,9 @@ class KeylockDecoderComponent:
             )
 
     def build_ui(self):
+        gr.HTML("<div style='background:blue;'>")
         with gr.Column(elem_id="login-container"):
+        
             gr.HTML("""
                 <div id="keylock-logo">
                     <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor"><path d="M18 8h-1V6c0-2.76-2.24-5-5-5S7 3.24 7 6v2H6c-1.1 0-2 .9-2 2v10c0 1.1.9 2 2 2h12c1.1 0 2-.9 2-2V10c0-1.1-.9-2-2-2zM9 6c0-1.66 1.34-3 3-3s3 1.34 3 3v2H9V6zm9 14H6V10h12v10zm-6-3c1.1 0 2-.9 2-2s-.9-2-2-2-2 .9-2 2 .9 2 2 2z"></path></svg>
@@ -317,6 +319,7 @@ class KeylockDecoderComponent:
         def generate_wrapper(kv_string):
             payload_dict = self.server_logic._parse_kv_string(kv_string)
             return self.server_logic.generate_encrypted_image(payload_dict)
+        gr.HTML("</div>")
 
         self.image_input.upload(
             fn=self._handle_login_attempt,