Update app.py

app.py

@@ -1,5 +1,4 @@
-
-# -* coding:UTF-8 -*
+# -*- coding: UTF-8 -*-
 # !/usr/bin/env python
 import numpy as np
 import gradio as gr
@@ -13,11 +12,33 @@ from roop.core import (
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
 import os
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
+
+
+def add_watermark(image_path, text="Made with ❤️ by Aadhi"):
+    image = Image.open(image_path).convert("RGBA")
+    watermark_layer = Image.new("RGBA", image.size, (255, 255, 255, 0))
+    draw = ImageDraw.Draw(watermark_layer)
+
+    font_size = int(min(image.size) / 20)
+    try:
+        font = ImageFont.truetype("arial.ttf", font_size)
+    except:
+        font = ImageFont.load_default()
+
+    text_width, text_height = draw.textsize(text, font=font)
+    x = image.width - text_width - 20
+    y = image.height - text_height - 20
+
+    draw.text((x, y), text, font=font, fill=(255, 255, 255, 180))  # semi-transparent white
 
+    watermarked_image = Image.alpha_composite(image, watermark_layer)
+    watermarked_path = "output_watermarked.jpg"
+    watermarked_image.convert("RGB").save(watermarked_path)
+    return watermarked_path
 
-def swap_face(source_file, target_file,doFaceEnhancer):
 
+def swap_face(source_file, target_file, doFaceEnhancer):
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -35,10 +56,12 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
-    if doFaceEnhancer == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
+
+    if doFaceEnhancer:
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
+
     roop.globals.headless = True
     roop.globals.keep_fps = True
     roop.globals.keep_audio = True
@@ -50,24 +73,28 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.execution_providers = decode_execution_providers(["cuda"])
     roop.globals.execution_threads = suggest_execution_threads()
 
-    print(
-        "start process",
-        roop.globals.source_path,
-        roop.globals.target_path,
-        roop.globals.output_path,
-    )
+    print("start process", roop.globals.source_path, roop.globals.target_path, roop.globals.output_path)
 
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
+    for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
         if not frame_processor.pre_check():
             return
 
     start()
-    return output_path
+
+    # Add watermark after face swap
+    watermarked_path = add_watermark(output_path, text="Made with ❤️ by Aadhi")
+    return watermarked_path
 
 
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+    fn=swap_face,
+    inputs=[
+        gr.Image(label="Source Image"),
+        gr.Image(label="Target Image"),
+        gr.Checkbox(label="Enable Face Enhancer?", info="Improves output quality")
+    ],
+    outputs=gr.Image(label="Output Image with Watermark"),
+    title="Face Swap with Watermark",
 )
-app.launch()
+
+app.launch()