Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import PyPDF2
-from PIL import Image, ImageFont, ImageDraw
+from PIL import Image, ImageEnhance, ImageFont, ImageDraw
 import cv2
 import numpy as np
 from pydub import AudioSegment
@@ -16,6 +16,8 @@ import time
 import requests
 import zipfile
 import os
+import torchaudio
+from transformers import pipeline
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -209,6 +211,49 @@ def generate_text_image(text, font_file, font_size=40, bg_color="#FFFFFF", text_
         image.save(tmp_file.name)
         return tmp_file.name
 
+# Функция для улучшения качества изображения
+def enhance_image(image, contrast_factor=1.5, brightness_factor=1.2, sharpness_factor=1.5):
+    if image is None:
+        return None
+    
+    img = Image.open(image.name)
+    
+    # Улучшение контраста
+    enhancer = ImageEnhance.Contrast(img)
+    img = enhancer.enhance(contrast_factor)
+    
+    # Улучшение яркости
+    enhancer = ImageEnhance.Brightness(img)
+    img = enhancer.enhance(brightness_factor)
+    
+    # Улучшение резкости
+    enhancer = ImageEnhance.Sharpness(img)
+    img = enhancer.enhance(sharpness_factor)
+    
+    # Сохраняем изображение во временный файл
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
+        img.save(tmp_file.name)
+        return tmp_file.name
+
+# Функция для обрезки аудио
+def trim_audio(file, start_time: float, end_time: float):
+    if file is None:
+        return "Файл не загружен."
+    
+    audio = AudioSegment.from_file(file.name)
+    trimmed_audio = audio[start_time * 1000:end_time * 1000]
+    
+    # Сохраняем обрезанное аудио во временный файл
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
+        trimmed_audio.export(tmp_file.name, format="wav")
+        return tmp_file.name
+
+# Функция для суммаризации текста
+def summarize_text(text: str, max_length: int = 100):
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    summary = summarizer(text, max_length=max_length, min_length=30, do_sample=False)
+    return summary[0]['summary_text']
+
 # Создание интерфейса
 with gr.Blocks() as demo:
     # Заголовок с изображением
@@ -263,22 +308,35 @@ with gr.Blocks() as demo:
             convert_button = gr.Button("Конвертировать")
             convert_button.click(fn=convert_image, inputs=[img_input, img_format], outputs=img_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="Размер шрифта")
-            bg_color_input = gr.ColorPicker(label="Цвет фона", value="#FFFFFF")
-            text_color_input = gr.ColorPicker(label="Цвет текста", value="#000000")
-            generate_button = gr.Button("Сгенерировать изображение")
+        # Вкладка Felguk-ImageEnhancer
+        with gr.Tab("Felguk-ImageEnhancer"):
+            gr.Markdown("## Felguk-ImageEnhancer: Улучшение качества изображений")
+            image_input = gr.File(label="Загрузите изображение", file_types=["image"])
+            contrast_slider = gr.Slider(label="Контраст", minimum=0.5, maximum=2.0, value=1.5)
+            brightness_slider = gr.Slider(label="Яркость", minimum=0.5, maximum=2.0, value=1.2)
+            sharpness_slider = gr.Slider(label="Резкость", minimum=0.5, maximum=2.0, value=1.5)
+            enhance_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, bg_color_input, text_color_input],
-                outputs=image_output
-            )
+            enhance_button.click(fn=enhance_image, inputs=[image_input, contrast_slider, brightness_slider, sharpness_slider], outputs=image_output)
+
+        # Вкладка Felguk-AudioTrimmer
+        with gr.Tab("Felguk-AudioTrimmer"):
+            gr.Markdown("## Felguk-AudioTrimmer: Обрезка аудиофайлов")
+            audio_input = gr.File(label="Загрузите аудиофайл", file_types=[".mp3", ".wav"])
+            start_time = gr.Number(label="Начало обрезки (сек)", value=0)
+            end_time = gr.Number(label="Конец обрезки (сек)", value=10)
+            trim_button = gr.Button("Обрезать аудио")
+            audio_output = gr.Audio(label="Результат")
+            trim_button.click(fn=trim_audio, inputs=[audio_input, start_time, end_time], outputs=audio_output)
+
+        # Вкладка Felguk-TextSummarizer
+        with gr.Tab("Felguk-TextSummarizer"):
+            gr.Markdown("## Felguk-TextSummarizer: Суммаризация текста")
+            text_input = gr.Textbox(label="Введите текст для суммаризации", lines=5)
+            max_length_slider = gr.Slider(label="Максимальная длина текста", minimum=30, maximum=200, value=100)
+            summarize_button = gr.Button("Суммаризировать")
+            summary_output = gr.Textbox(label="Результат", interactive=False)
+            summarize_button.click(fn=summarize_text, inputs=[text_input, max_length_slider], outputs=summary_output)
 
     # Интерфейс чата
     chat_interface = gr.ChatInterface(