Update app.py

app.py

@@ -45,10 +45,10 @@ def main_ui():
                 with gr.Row():
                     keylock_image_input = gr.Image(
                         label="Stego Image (PNG containing HF_TOKEN)",
-                        type="pil", # Pass PIL image to backend
-                        # sources=["upload"], # Optional: Restrict to upload only
-                        # height=200, # Optional: for display consistency
-                        # width=200,
+                        type="pil",
+                        image_mode="RGBA",  # << TRY ADDING THIS. Use "RGB" if your images are RGB.
+                                            # If you're unsure, you can omit it first to see if the debug saving helps.
+                        # sources=["upload"], 
                     )
                     keylock_password_input = gr.Textbox(
                         label="Image Password",