Update app.py

app.py

@@ -7,9 +7,8 @@ import zipfile
 from PIL import Image
 
 # --- Required Core Logic ---
-# These functions are copied from your project's core modules.
 # Ensure keylock1 and repo_to_md are in the same directory or accessible in the Python path.
-from keylock import core as keylock_core
+from keylock1 import core as keylock_core
 from repo_to_md.core import markdown_to_files
 
 # --- Helper Functions for the Decoder UI ---
@@ -17,6 +16,7 @@ from repo_to_md.core import markdown_to_files
 def retrieve_and_process_files(stego_image_pil: Image.Image, password: str):
     """
     Extracts data from an image and prepares the file browser UI components for update.
+    This version handles cases where the password is an empty string.
     """
     # Initial state for all UI components
     status = "An error occurred."
@@ -30,12 +30,11 @@ def retrieve_and_process_files(stego_image_pil: Image.Image, password: str):
         status = "Error: Please upload or paste an image."
         return status, file_data_state, file_buffers_state, file_browser_visibility, filename_choices, accordion_update, code_update
 
-    if not password:
-        status = "Error: Please enter a decryption password."
-        return status, file_data_state, file_buffers_state, file_browser_visibility, filename_choices, accordion_update, code_update
+    # The check for a mandatory password has been removed. An empty string is now a valid input.
 
     try:
-        # Step 1: Extract and decrypt data from the image
+        # Step 1: Extract and decrypt data from the image.
+        # The password (which can be empty) is passed directly to the core function.
         extracted_data = keylock_core.extract_data_from_image(stego_image_pil.convert("RGB"))
         decrypted_bytes = keylock_core.decrypt_data(extracted_data, password)
         
@@ -67,8 +66,8 @@ def retrieve_and_process_files(stego_image_pil: Image.Image, password: str):
         else:
             status += " (No files were found in the decoded content)."
 
-    except ValueError as e:
-        status = f"Error: Failed to decrypt. Most likely an incorrect password. ({e})"
+    except ValueError:
+        status = "Error: Failed to decrypt. This usually means the password is wrong (or was required but not provided)."
     except Exception as e:
         status = f"An unexpected error occurred: {e}"
     
@@ -103,7 +102,7 @@ def download_all_zip(buffers):
 # --- GRADIO UI DEFINITION ---
 with gr.Blocks(theme=gr.themes.Soft(), title="KeyLock Decoder") as demo:
     gr.Markdown("# KeyLock Image Decoder")
-    gr.Markdown("Upload or paste your KeyLock image, enter the password, and click 'Extract Files' to view and download the contents.")
+    gr.Markdown("Upload or paste your KeyLock image, enter the password (if one was used), and click 'Extract Files' to view and download the contents.")
     
     # State variables to hold data between user interactions
     file_data_state = gr.State([])
@@ -118,9 +117,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="KeyLock Decoder") as demo:
                 sources=["upload", "clipboard"]
             )
             extract_password_input = gr.Textbox(
-                label="Decryption Password",
+                label="Decryption Password (Optional)",
                 type="password",
-                placeholder="Enter the password used for encryption"
+                placeholder="Leave blank if no password was set",
+                info="If a password was used during encryption, it is required here."
             )
             extract_button = gr.Button(
                 value="Extract Files",