Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
-from keylock_component import KeylockDecoderComponent
+from keylock_component import KeylockDecoderComponent, AppServerLogic
+
+shared_server_logic = AppServerLogic()
 
-# Mock credentials for login
 MOCK_CREDENTIALS = {
     "demo-user": {"password": "password123"},
     "admin-user": {"password": "adminpass"}
@@ -9,6 +10,7 @@ MOCK_CREDENTIALS = {
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="sky")) as demo:
     gr.Markdown("# 🔑 KeyLock Application Login")
+
     with gr.Row():
         with gr.Column(scale=2):
             gr.Markdown("### 1. Login with Credentials")
@@ -17,13 +19,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="sky")) as demo:
             password_input = gr.Textbox(label="Password", type="password", interactive=True)
             login_button = gr.Button("Login", variant="primary")
             login_status = gr.Markdown()
+        
         with gr.Column(scale=3):
             gr.Markdown("### 2. Use a KeyLock Image to Auto-Fill")
-            keylock_decoder = KeylockDecoderComponent()
-            # State to store the decoded payload
-            decoded_payload = gr.State()
-            # Button to trigger payload check (workaround for change event issue)
-            check_payload_button = gr.Button("Check Decoded Payload", variant="secondary", visible=False)
+            keylock_decoder = KeylockDecoderComponent(
+                server_logic=shared_server_logic
+            )
 
     def handle_login(username, password):
         user_data = MOCK_CREDENTIALS.get(username)
@@ -33,40 +34,22 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="sky")) as demo:
         else:
             gr.Error("Login Failed: Invalid username or password.")
             return f"<p style='color:red;font-weight:bold'>❌ Login failed.</p>"
-
-    def autofill_from_payload(payload):
-        """
-        Extracts USER and PASSWORD from the raw JSON payload and updates textboxes.
-        """
-        if payload and payload.get("status") == "Success":
-            payload_data = payload.get("payload", {})
-            user = payload_data.get("USER", "")
-            password = payload_data.get("PASSWORD", "")
+            
+    def autofill_from_key(decoder_result):
+        if decoder_result and decoder_result.get("status") == "Success":
+            payload = decoder_result.get("payload", {})
+            user = payload.get("USER", "")
+            user_data = MOCK_CREDENTIALS.get(user)
+            password = user_data.get("password", "") if user_data else ""
             return gr.update(value=user), gr.update(value=password)
         return gr.update(), gr.update()
 
-    def check_component_value():
-        """
-        Retrieves the current value of the KeylockDecoderComponent.
-        """
-        return keylock_decoder.value
-
-    # Trigger check_payload_button automatically when keylock_decoder value changes
     keylock_decoder.change(
-        fn=check_component_value,
-        inputs=[],
-        outputs=[decoded_payload],
-        js="() => { setTimeout(() => document.querySelector('button[aria-label=\"Check Decoded Payload\"]').click(), 100); return []; }"
-    )
-
-    # When check_payload_button is clicked (manually or via JS), update textboxes
-    check_payload_button.click(
-        fn=autofill_from_payload,
-        inputs=[decoded_payload],
+        fn=autofill_from_key,
+        inputs=[keylock_decoder],
         outputs=[username_input, password_input]
     )
-
-    # Login button validation
+    
     login_button.click(
         fn=handle_login,
         inputs=[username_input, password_input],