Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import PyPDF2
-from PIL import Image
+from PIL import Image, ImageFont, ImageDraw
 import cv2
 import numpy as np
 from pydub import AudioSegment
@@ -14,6 +14,8 @@ from torchvision.transforms import functional as F
 import tempfile
 import time
 import requests
+import zipfile
+import os
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -22,36 +24,6 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 model = fasterrcnn_resnet50_fpn(pretrained=True)
 model.eval()
 
-# API для модели NSFW
-API_URL = "https://api-inference.huggingface.co/models/Falconsai/nsfw_image_detection"
-headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}  # Замените на ваш API-токен
-
-# Функция для определения NSFW в изображении
-def detect_nsfw(image):
-    if image is None:
-        return "Изображение не загружено."
-    
-    # Сохраняем изображение во временный файл
-    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp_file:
-        image.save(tmp_file.name)
-        filename = tmp_file.name
-    
-    # Отправка изображения в модель NSFW через API
-    try:
-        with open(filename, "rb") as f:
-            data = f.read()
-        response = requests.post(API_URL, headers=headers, data=data)
-        result = response.json()
-        
-        # Обработка результата
-        if isinstance(result, list) and len(result) > 0 and "score" in result[0]:
-            nsfw_score = result[0]["score"]
-            return f"Процент NSFW: {nsfw_score * 100:.2f}%"
-        else:
-            return "Не удалось определить NSFW. Попробуйте другое изображение."
-    except Exception as e:
-        return f"Ошибка при обработке изображения: {str(e)}"
-
 # Функция для обработки PDF
 def process_pdf(file):
     pdf_reader = PyPDF2.PdfReader(file)
@@ -199,6 +171,37 @@ def analyze_txt(file):
     text = process_txt(file)
     return f"Содержимое файла:\n{text}"
 
+# Функция для генерации изображения с текстом и шрифтом
+def generate_text_image(text, font_file, font_size=40):
+    if not text:
+        return "Введите текст."
+    
+    # Если шрифт не загружен, используем стандартный
+    if font_file is None:
+        font = ImageFont.load_default()
+    else:
+        # Если шрифт в архиве, извлекаем его
+        if font_file.name.endswith(".zip"):
+            with zipfile.ZipFile(font_file.name, 'r') as zip_ref:
+                zip_ref.extractall("temp_fonts")
+                font_files = [f for f in os.listdir("temp_fonts") if f.endswith(('.ttf', '.otf'))]
+                if not font_files:
+                    return "В архиве нет файлов шрифтов (.ttf, .otf)."
+                font_path = os.path.join("temp_fonts", font_files[0])
+                font = ImageFont.truetype(font_path, font_size)
+        else:
+            font = ImageFont.truetype(font_file.name, font_size)
+    
+    # Создаем изображение с текстом
+    image = Image.new("RGB", (800, 200), color="white")
+    draw = ImageDraw.Draw(image)
+    draw.text((10, 10), text, font=font, fill="black")
+    
+    # Сохраняем изображение во временный файл
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
+        image.save(tmp_file.name)
+        return tmp_file.name
+
 # Создание интерфейса
 with gr.Blocks() as demo:
     # Заголовок с изображением
@@ -253,13 +256,15 @@ with gr.Blocks() as demo:
             convert_button = gr.Button("Конвертировать")
             convert_button.click(fn=convert_image, inputs=[img_input, img_format], outputs=img_output)
 
-        # Вкладка Felguk Don't Like NSFW
-        with gr.Tab("Felguk Don't Like NSFW"):
-            gr.Markdown("## Felguk Don't Like NSFW: Определение NSFW в изображении")
-            nsfw_input = gr.Image(label="Загрузите изображение", type="pil")
-            nsfw_output = gr.Textbox(label="Результат", interactive=False)
-            nsfw_button = gr.Button("Проверить на NSFW")
-            nsfw_button.click(fn=detect_nsfw, inputs=nsfw_input, outputs=nsfw_output)
+        # Вкладка Felguk Fonts
+        with gr.Tab("Felguk Fonts"):
+            gr.Markdown("## Felguk Fonts: Генерация текста с шрифтом")
+            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="Размер шрифта")
+            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)
 
     # Интерфейс чата
     chat_interface = gr.ChatInterface(