oauth

README.md

@@ -9,6 +9,8 @@ app_file: app.py
 pinned: false
 license: other
 short_description: Enhance hand features for flawless, natural-looking images
+
+hf_oauth:  true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -223,9 +223,12 @@ def image_to_base64(image):
     return f"data:image/png;base64,{img_base64}"
 
 
-def generate_image(edit_image_infos, did, request: gr.Request):
+def generate_image(edit_image_infos, did, request: gr.Request, oauth_token: gr.OAuthToken | None):
     if not did:
         did = str(uuid.uuid4())
+    if not oauth_token:
+        m = "Please log in to your Hugging Face account to use the features of this application."
+        return gr.Warning(m), did
     if edit_image_infos is None or not isinstance(edit_image_infos, dict):
         m = "Please upload the main image before generating."
         return gr.Warning(m), did
@@ -464,6 +467,8 @@ with gr.Blocks(css=css) as WeShop:
                     outputs=[output, current_did],
                     concurrency_limit=None
                 )
+            with gr.Row():
+                gr.LoginButton()
             with gr.Column():
                 main_image_middleware = gr.Image(
                     image_mode='RGBA',