Add LoginButton

app.py

@@ -90,7 +90,28 @@ import spaces
 
 input_image_palette = []
 current_prerendered_image = gr.State("./images/Beeuty-1.png")
-
+user_info = {
+    "username": "guest",
+    "session_hash": None,
+    "headers": None,
+    "client": None,
+    "query_params": None,
+    "path_params": None,
+    "level" : 0
+}
+# Define a function to handle the login button click and retrieve user information.
+def handle_login(request: gr.Request):
+    # Extract user information from the request
+    user_info = {
+        "username": request.username,
+        "session_hash": request.session_hash,
+        "headers": dict(request.headers),
+        "client": request.client,
+        "query_params": dict(request.query_params),
+        "path_params": dict(request.path_params),
+        "level" : (0 if request.username == "guest" else 2)
+    }
+    return user_info, gr.update(logout_value=f"Logout {user_info['username']} ({user_info['level']})", value=f"Login {user_info['username']} ({user_info['level']})")
 #---if workspace = local or colab---
 
 # Authenticate with Hugging Face
@@ -858,6 +879,13 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty',
                                 randomize_seed = gr.Checkbox(False, label="Randomize seed",elem_id="rnd_seed_chk")
                                 seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True, elem_id="rnd_seed")
                                 lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=1.01)
+    with gr.Row():
+        login_button = gr.LoginButton(logout_value=f"Logout({user_info['username']} ({user_info['level']}))", size="md", elem_classes="solid centered", elem_id="hf_login_btn", icon="./assets/favicon.ico")
+        # Create a JSON component to display the user information
+        user_info_output = gr.JSON(label="User Information:")
+    
+        # Set up the event listener for the login button click
+        login_button.click(fn=handle_login, inputs=[], outputs=[user_info_output, login_button])
     with gr.Row():
         gr.HTML(value=getVersions(), visible=True, elem_id="versions")