oauth

README.md

@@ -9,6 +9,8 @@ app_file: app.py
 pinned: false
 license: other
 short_description: Transform flat-lay shots into on-model photos
+
+hf_oauth:  true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -209,9 +209,12 @@ def load_description(file_path):
     return content
 
 
-def generate_image(main_image, background_image, did, request: gr.Request):
+def generate_image(main_image, background_image, 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 main_image is None or background_image is None:
         m = "Please upload both the main image and the background reference image before generating."
         return gr.Warning(m), did
@@ -405,6 +408,8 @@ with gr.Blocks(css=css) as WeShop:
                     outputs=[output, current_did],
                     concurrency_limit=None
                 )
+            with gr.Row():
+                gr.LoginButton()
             with gr.Column():
                 show_case = gr.Examples(
                     examples=[