Update app.py

app.py

@@ -172,7 +172,7 @@ def analyze_txt(file):
     return f"Содержимое файла:\n{text}"
 
 # Функция для генерации изображения с текстом и шрифтом
-def generate_text_image(text, font_file, font_size=40):
+def generate_text_image(text, font_file, font_size=40, bg_color="#FFFFFF", text_color="#000000"):
     if not text:
         return "Введите текст."
     
@@ -193,9 +193,16 @@ def generate_text_image(text, font_file, font_size=40):
             font = ImageFont.truetype(font_file.name, font_size)
     
     # Создаем изображение с текстом
-    image = Image.new("RGB", (800, 200), color="white")
+    image = Image.new("RGB", (800, 200), color=bg_color)
     draw = ImageDraw.Draw(image)
-    draw.text((10, 10), text, font=font, fill="black")
+    
+    # Рассчитываем позицию текста (по центру)
+    text_width, text_height = draw.textsize(text, font=font)
+    x = (800 - text_width) / 2
+    y = (200 - text_height) / 2
+    
+    # Рисуем текст на изображении
+    draw.text((x, y), text, font=font, fill=text_color)
     
     # Сохраняем изображение во временный файл
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
@@ -262,9 +269,16 @@ with gr.Blocks() as demo:
             text_input = gr.Textbox(label="Введите текст", placeholder="Введите текст здесь...")
             font_input = gr.File(label="Загрузите шрифт (.ttf, .otf, .zip)", file_types=[".ttf", ".otf", ".zip"])
             font_size_input = gr.Slider(minimum=10, maximum=100, value=40, label="Размер шрифта")
+            bg_color_input = gr.ColorPicker(label="Цвет фона", value="#FFFFFF")
+            text_color_input = gr.ColorPicker(label="Цвет текста", value="#000000")
             generate_button = gr.Button("Сгенерировать изображение")
             image_output = gr.Image(label="Результат", type="filepath")
-            generate_button.click(fn=generate_text_image, inputs=[text_input, font_input, font_size_input], outputs=image_output)
+            
+            generate_button.click(
+                fn=generate_text_image,
+                inputs=[text_input, font_input, font_size_input, bg_color_input, text_color_input],
+                outputs=image_output
+            )
 
     # Интерфейс чата
     chat_interface = gr.ChatInterface(