Doubleupai commited on
Commit
2646b5e
·
verified ·
1 Parent(s): 47e9963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -18
app.py CHANGED
@@ -1,17 +1,8 @@
1
- from fontTools.ttLib import TTFont
2
  from PIL import Image, ImageDraw, ImageFont
3
  import gradio as gr
4
  import time
5
- import os
6
 
7
- def check_font(font_path):
8
- try:
9
- font = TTFont(font_path)
10
- return True
11
- except:
12
- return False
13
-
14
- def generate_logo(text, font_size, background_color, text_color):
15
  # Задержка 12 секунд
16
  time.sleep(12)
17
 
@@ -22,14 +13,9 @@ def generate_logo(text, font_size, background_color, text_color):
22
  # Создаем объект для рисования
23
  draw = ImageDraw.Draw(image)
24
 
25
- # Путь к шрифту
26
- font_path = "arial.ttf"
27
- if not os.path.exists(font_path) or not check_font(font_path):
28
- font_path = None # Используем стандартный шрифт
29
-
30
- # Загружаем шрифт
31
  try:
32
- font = ImageFont.truetype(font_path, font_size) if font_path else ImageFont.load_default()
33
  except IOError:
34
  font = ImageFont.load_default()
35
 
@@ -49,6 +35,10 @@ def generate_logo(text, font_size, background_color, text_color):
49
  # Рисуем текст на изображении
50
  draw.text((x, y), text, font=font, fill=text_color)
51
 
 
 
 
 
52
  # Сохраняем изображение
53
  image.save("logo.png")
54
 
@@ -61,7 +51,9 @@ iface = gr.Interface(
61
  gr.Textbox(label="Текст логотипа", value="My Logo"),
62
  gr.Slider(minimum=10, maximum=100, value=40, label="Размер шрифта"),
63
  gr.ColorPicker(label="Цвет фона", value="#FFFFFF"),
64
- gr.ColorPicker(label="Цвет текста", value="#000000")
 
 
65
  ],
66
  outputs=gr.Image(label="Сгенерированный логотип"),
67
  title="Генератор логотипов",
 
 
1
  from PIL import Image, ImageDraw, ImageFont
2
  import gradio as gr
3
  import time
 
4
 
5
+ def generate_logo(text, font_size, background_color, text_color, industry, description):
 
 
 
 
 
 
 
6
  # Задержка 12 секунд
7
  time.sleep(12)
8
 
 
13
  # Создаем объект для рисования
14
  draw = ImageDraw.Draw(image)
15
 
16
+ # Загружаем шрифт (если шрифт не найден, используется стандартный)
 
 
 
 
 
17
  try:
18
+ font = ImageFont.truetype("arial.ttf", font_size)
19
  except IOError:
20
  font = ImageFont.load_default()
21
 
 
35
  # Рисуем текст на изображении
36
  draw.text((x, y), text, font=font, fill=text_color)
37
 
38
+ # Добавляем информацию об отрасли и описании (опционально)
39
+ info = f"Отрасль: {industry}\nОписание: {description}"
40
+ draw.text((10, height - 40), info, font=ImageFont.load_default(), fill="gray")
41
+
42
  # Сохраняем изображение
43
  image.save("logo.png")
44
 
 
51
  gr.Textbox(label="Текст логотипа", value="My Logo"),
52
  gr.Slider(minimum=10, maximum=100, value=40, label="Размер шрифта"),
53
  gr.ColorPicker(label="Цвет фона", value="#FFFFFF"),
54
+ gr.ColorPicker(label="Цвет текста", value="#000000"),
55
+ gr.Textbox(label="Отрасль", placeholder="Введите отрасль (например, IT, рестораны, мода)"),
56
+ gr.Textbox(label="О чем ваш логотип?", placeholder="Краткое описание логотипа")
57
  ],
58
  outputs=gr.Image(label="Сгенерированный логотип"),
59
  title="Генератор логотипов",