Update app.py

app.py

@@ -1,17 +1,8 @@
-from fontTools.ttLib import TTFont
 from PIL import Image, ImageDraw, ImageFont
 import gradio as gr
 import time
-import os
 
-def check_font(font_path):
-    try:
-        font = TTFont(font_path)
-        return True
-    except:
-        return False
-
-def generate_logo(text, font_size, background_color, text_color):
+def generate_logo(text, font_size, background_color, text_color, industry, description):
     # Задержка 12 секунд
     time.sleep(12)
     
@@ -22,14 +13,9 @@ def generate_logo(text, font_size, background_color, text_color):
     # Создаем объект для рисования
     draw = ImageDraw.Draw(image)
     
-    # Путь к шрифту
-    font_path = "arial.ttf"
-    if not os.path.exists(font_path) or not check_font(font_path):
-        font_path = None  # Используем стандартный шрифт
-    
-    # Загружаем шрифт
+    # Загружаем шрифт (если шрифт не найден, используется стандартный)
     try:
-        font = ImageFont.truetype(font_path, font_size) if font_path else ImageFont.load_default()
+        font = ImageFont.truetype("arial.ttf", font_size)
     except IOError:
         font = ImageFont.load_default()
     
@@ -49,6 +35,10 @@ def generate_logo(text, font_size, background_color, text_color):
     # Рисуем текст на изображении
     draw.text((x, y), text, font=font, fill=text_color)
     
+    # Добавляем информацию об отрасли и описании (опционально)
+    info = f"Отрасль: {industry}\nОписание: {description}"
+    draw.text((10, height - 40), info, font=ImageFont.load_default(), fill="gray")
+    
     # Сохраняем изображение
     image.save("logo.png")
     
@@ -61,7 +51,9 @@ iface = gr.Interface(
         gr.Textbox(label="Текст логотипа", value="My Logo"),
         gr.Slider(minimum=10, maximum=100, value=40, label="Размер шрифта"),
         gr.ColorPicker(label="Цвет фона", value="#FFFFFF"),
-        gr.ColorPicker(label="Цвет текста", value="#000000")
+        gr.ColorPicker(label="Цвет текста", value="#000000"),
+        gr.Textbox(label="Отрасль", placeholder="Введите отрасль (например, IT, рестораны, мода)"),
+        gr.Textbox(label="О чем ваш логотип?", placeholder="Краткое описание логотипа")
     ],
     outputs=gr.Image(label="Сгенерированный логотип"),
     title="Генератор логотипов",