Upload 3 files

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+from funcs import process_image
+
+def process_image_gradio(image_path, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin):
+    # Вызываем функцию process_image из funcs.py
+    output_paths = process_image(image_path, "output", watermark_text, font_path.name, font_size, opacity, color, right_margin, bottom_margin)
+
+    # Cover obj to json str
+    generation_params_str = f"Watermark Text: {watermark_text}\nFont Size: {font_size}\nOpacity: {opacity}\nColor: {color}\nRight Margin: {right_margin}\nBottom Margin: {bottom_margin}"
+
+    # Возвращаем путь к изображению с водяным знаком
+    return output_paths[0], output_paths[1], generation_params_str
+
+# Определяем интерфейс с использованием Gradio
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            # Компоненты для настроек
+            image = gr.Image(label="Upload Image", type="filepath")
+            font_path = gr.File(label="Font Path", file_types=[".ttf"])
+            watermark_text = gr.Textbox(label="Watermark Text")
+            font_size = gr.Slider(minimum=6, maximum=100, step=1, label="Font Size", value=30)
+            opacity = gr.Slider(minimum=0, maximum=1, step=0.01, label="Opacity", value=0.5)
+            color = gr.ColorPicker(label="Color",value="#000000")
+            right_margin = gr.Slider(minimum=0, maximum=1, step=0.01, label="Right Margin", value=0.05)
+            bottom_margin = gr.Slider(minimum=0, maximum=1, step=0.01, label="Bottom Margin", value=0.02)
+
+        with gr.Column():
+            # Компонент для отображения результата
+            process_button = gr.Button("Process Image")
+            generation_params = gr.Textbox(label="Generation Parameters",interactive=False,show_copy_button=True)
+            output_image = gr.Image(label="Output Image",interactive=False)
+            output_mini = gr.Image(label="Output Image (Mini)",interactive=False)
+            
+            
+
+    # Обработчик нажатия кнопки "Process Image"
+    process_button.click(process_image_gradio, inputs=[image, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin], outputs=[output_image,output_mini,generation_params])
+
+# Запускаем интерфейс
+demo.queue()
+demo.launch()

funcs.py

@@ -0,0 +1,70 @@
+import math
+import os
+from PIL import Image, ImageDraw, ImageFont
+
+def add_watermark(image, watermark_text, font_path, font_size, opacity, color, right_margin, bottom_margin):
+    # Load the custom font
+    font = ImageFont.truetype(font_path, font_size)
+
+    # Create a drawing context
+    draw = ImageDraw.Draw(image)
+
+    # Get the image dimensions
+    image_width, image_height = image.size
+
+    # Calculate the position for the watermark
+    text_bbox = draw.textbbox((0, 0), watermark_text, font=font)
+    text_width = text_bbox[2] - text_bbox[0]
+    text_height = text_bbox[3] - text_bbox[1] + math.floor(font_size / 2)
+    x = image_width - text_width - int(image_width * right_margin)
+    y = image_height - text_height - int(image_height * bottom_margin)  # Добавляем отступ снизу
+
+    # Create a new image for the watermark with alpha channel
+    watermark = Image.new('RGBA', (text_width, text_height), (0, 0, 0, 0))
+    watermark_draw = ImageDraw.Draw(watermark)
+
+    # Convert the color tuple to a hexadecimal color string
+    hex_color = "#%02x%02x%02x" % color
+    
+    print(color, hex_color)
+
+    # Draw the watermark text on the new image
+    watermark_draw.text((0, 0), watermark_text, font=font, fill=color + (int(255 * opacity),))
+
+    # Paste the watermark onto the original image
+    image.paste(watermark, (x, y), mask=watermark)
+
+
+
+def process_image(image_path, output_dir, watermark_text=None, font_path=None, font_size=30, opacity=0.5,
+                  color="#000000", right_margin=0.05, bottom_margin=0.02):
+    # Convert the color string to a tuple
+    color_tuple = tuple(int(color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
+    
+    # Create output folder 
+    os.makedirs(output_dir, exist_ok=True)
+
+    # Открываем исходное изображение
+    with Image.open(image_path) as img:
+        # Конвертируем изображение в режим RGBA, если необходимо
+        if img.mode != 'RGBA':
+            img = img.convert('RGBA')
+
+        # Добавляем водяной знак на исходное изображение, если указан текст
+        if watermark_text and font_path:
+            add_watermark(img, watermark_text, font_path, font_size, opacity, color_tuple, right_margin, bottom_margin)
+
+        # Сохраняем исходное изображение с водяным знаком
+        watermarked_file = os.path.join(output_dir, 'watermarked.png')
+        img.save(watermarked_file, format='PNG')
+
+        # Создаем миниатюру изображения
+        thumbnail_size = (200, 200)  # Укажите нужный размер миниатюры
+        img.thumbnail(thumbnail_size)
+
+        # Сохраняем миниатюру
+        thumbnail_file = os.path.join(output_dir, 'thumbnail.png')
+        img.save(thumbnail_file, format='PNG')
+        
+    # Возвращаем список с путями к полученным файлам
+    return [watermarked_file, thumbnail_file]

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+pillow