Update app.py

app.py

@@ -4,6 +4,12 @@ import json
 
 server_logic = AppServerLogic()
 
+example_db = {"USER":"TestUser","PASS":"TestPass"}
+
+def login_fn(username,password):
+    if username==example_db.USER and password==example_db.PASS:
+        return gr.Update("### Logged in")
+
 def process_image_and_display_payload(image):
     if image is None:
         return "Upload a KeyLock image to auto-fill credentials.", {}
@@ -38,35 +44,24 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     with gr.Row():
         with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("## Login")
-                    user_name=gr.Textbox(label='User Name')
-                    user_pass=gr.Textbox(label='Password', type='password')
-                    gr.Markdown("## API Keys")
-                    user_GROQ=gr.Textbox(label='GROQ_API_KEY', type='password')
-                    user_HF=gr.Textbox(label='HF_TOKEN', type='password')
-                    user_OPENAI=gr.Textbox(label='OPENAI_API_KEY', type='password')
-                    user_OPENROUTER=gr.Textbox(label='OPENROUTER_API_KEY', type='password')
-                    
-                with gr.Column():
-                    gr.Markdown("## Login")
-                    user_name=gr.Textbox(label='User Name')
-                    user_pass=gr.Textbox(label='Password', type='password')
-                    login_btn=gr.Button("Login")
-                    login_msg=gr.Markdown("### Enter user name and password")
-                    gr.Markdown("## API Keys (Demo)")
-                    user_GROQ=gr.Textbox(label='GROQ_API_KEY')
-                    user_HF=gr.Textbox(label='HF_TOKEN')
-                    user_OPENAI=gr.Textbox(label='OPENAI_API_KEY')
-                    user_OPENROUTER=gr.Textbox(label='OPENROUTER_API_KEY')
+            gr.Markdown("## Login")
+            user_name=gr.Textbox(label='User Name')
+            user_pass=gr.Textbox(label='Password', type='password')
+            login_btn=gr.Button("Login")
+            login_msg=gr.Markdown("### Enter user name and password")
+            gr.Markdown("## API Keys (Demo)")
+            user_GROQ=gr.Textbox(label='GROQ_API_KEY')
+            user_HF=gr.Textbox(label='HF_TOKEN')
+            user_OPENAI=gr.Textbox(label='OPENAI_API_KEY')
+            user_OPENROUTER=gr.Textbox(label='OPENROUTER_API_KEY')
             
         with gr.Column():
             keylock_builder = KeylockDecoderComponent(server_logic)
             image_input, status_display = keylock_builder.build_ui()
         
             output_json = gr.JSON(label="Decoded Payload")
-
+    
+    login_btn.click(login_fn,[user_name,user_pass], login_msg)
     image_input.upload(
         fn=process_image_and_display_payload,
         inputs=image_input,