Upload folder using huggingface_hub

gradio_demo/app.py

@@ -208,6 +208,47 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
     # def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     #     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     #     return p.replace("{prompt}", positive), n + ' ' + negative
+    
+    def add_watermark(image, watermark=logo, opacity=128, position="bottom_right", padding=10):
+        # Convert NumPy array to PIL Image if needed
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image)
+
+        if isinstance(watermark, np.ndarray):
+            watermark = Image.fromarray(watermark)
+
+        # Convert images to 'RGBA' mode to handle transparency
+        image = image.convert("RGBA")
+        watermark = watermark.convert("RGBA")
+
+        # Adjust the watermark opacity
+        watermark = watermark.copy()
+        watermark.putalpha(opacity)
+
+        # Calculate the position for the watermark
+        if position == "bottom_right":
+            x = image.width - watermark.width - padding
+            y = image.height - watermark.height - padding
+        elif position == "bottom_left":
+            x = padding
+            y = image.height - watermark.height - padding
+        elif position == "top_right":
+            x = image.width - watermark.width - padding
+            y = padding
+        elif position == "top_left":
+            x = padding
+            y = padding
+        else:
+            raise ValueError("Unsupported position. Choose from 'bottom_right', 'bottom_left', 'top_right', 'top_left'.")
+
+        # Paste the watermark onto the image
+        image.paste(watermark, (x, y), watermark)
+
+        # Convert back to 'RGB' if the original image was not 'RGBA'
+        if image.mode != "RGBA":
+            image = image.convert("RGB")
+
+        return image
 
     def generate_image(face_image,prompt,negative_prompt):
         pose_image_path = None
@@ -298,8 +339,11 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F
             generator=generator,
             # num_images_per_prompt = 4
         ).images
+        
+        watermarked_image = add_watermark(images[0])
 
-        return images[0]
+        # return images[0]
+        return watermarked_image
 
     ### Description
     title = r"""