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