require user to be logged in

README.md

@@ -11,6 +11,7 @@ short_description: Clean your room with the help of AI
 tags:
   - inpainting
   - image-to-image
+hf_oauth: true
 ---
 
 # Room Cleaner

app.py

@@ -12,6 +12,7 @@ from io import BytesIO
 import base64
 import numpy as np
 
+
 dotenv.load_dotenv()
 
 
@@ -36,6 +37,7 @@ def get_base64_from_image(image: Image.Image) -> str:
 def process_image(
     image: Image.Image | str | None,
     mask: Image.Image | str | None,
+    user_data: dict,
     progress: gr.Progress = gr.Progress(),
 ) -> Image.Image | None:
     progress(0, desc="Preparing inputs...")
@@ -54,6 +56,12 @@ def process_image(
     inputs: dict = {
         "image": f"data:image/png;base64,{image_base64}",
         "mask": f"data:image/png;base64,{mask_base64}",
+        # "run_metatada": str(
+        #     {
+        #         "source": "HF",
+        #         "user": user_data,
+        #     }
+        # ),
     }
 
     # Call ComfyDeploy API
@@ -139,14 +147,32 @@ def resize_image(img: Image.Image, min_side_length: int = 768) -> Image.Image:
     return img.resize((new_width, min_side_length))
 
 
+def get_profile(profile) -> dict:
+    return {
+        "username": profile.username,
+        "profile": profile.profile,
+        "name": profile.name,
+    }
+
+
 async def process(
     image_and_mask: EditorValue | None,
     progress: gr.Progress = gr.Progress(),
+    profile: gr.OAuthProfile | None = None,
 ) -> tuple[Image.Image, Image.Image] | None:
     if not image_and_mask:
         gr.Info("Please upload an image and draw a mask")
         return None
 
+    if profile is None:
+        gr.Info("Please log in to process the image.")
+        return None
+
+    user_data = get_profile(profile)
+    print("--------- RUN ----------")
+    print(user_data)
+    print("--------- RUN ----------")
+
     image_np = image_and_mask["background"]
     image_np = cast(np.ndarray, image_np)
 
@@ -173,6 +199,7 @@ async def process(
     output = process_image(
         image,  # type: ignore
         mask,  # type: ignore
+        user_data,
         progress,
     )
 
@@ -236,6 +263,8 @@ with gr.Blocks() as demo:
                 interactive=False,
             )
 
+            login_button = gr.LoginButton(scale=8)
+
             process_btn = gr.ClearButton(
                 value="Run",
                 variant="primary",

requirements.txt

@@ -1,4 +1,5 @@
 comfydeploy==0.3.4
 gradio==4.44.0
 gradio_imageslider==0.0.20
-python-dotenv
+python-dotenv
+gradio[oauth]