Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import PyPDF2
-from PIL import Image, ImageEnhance, ImageFont, ImageDraw
+from PIL import Image, ImageFont, ImageDraw
 import cv2
 import numpy as np
 from pydub import AudioSegment
@@ -24,7 +24,7 @@ from googletrans import Translator
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 # Загрузка предобученной модели для обнаружения объектов
-model = fasterrcnn_resnet50_fpn(pretrained=True)
+model = fasterrcnn_resnet50_fpn(weights="DEFAULT")
 model.eval()
 
 # Функция для обработки PDF
@@ -174,92 +174,6 @@ def analyze_txt(file):
     text = process_txt(file)
     return f"Содержимое файла:\n{text}"
 
-# Функция для генерации изображения с текстом и шрифтом
-def generate_text_image(text, font_file, font_size=40, bg_color="#FFFFFF", text_color="#000000"):
-    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=bg_color)
-    draw = ImageDraw.Draw(image)
-    
-    # Рассчитываем позицию текста (по центру)
-    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:
-        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']
-
-# Функция для поиска пространства на Hugging Face
-def search_space(author: str, space_name: str):
-    url = f"https://huggingface.co/space/{author}/{space_name}"
-    return url
-
 # Функция для изменения размера изображения
 def resize_image(image, width: int, height: int):
     if image is None:
@@ -387,15 +301,6 @@ with gr.Blocks() as demo:
             convert_button = gr.Button("Конвертировать")
             convert_button.click(fn=convert_image, inputs=[img_input, img_format], outputs=img_output)
 
-        # Вкладка Felguk-SpaceSearcher
-        with gr.Tab("Felguk-SpaceSearcher"):
-            gr.Markdown("## Felguk-SpaceSearcher: Поиск пространства на Hugging Face")
-            author_input = gr.Textbox(label="Имя автора")
-            space_name_input = gr.Textbox(label="Название пространства")
-            search_button = gr.Button("Найти пространство")
-            space_url_output = gr.Textbox(label="Ссылка на пространство", interactive=False)
-            search_button.click(fn=search_space, inputs=[author_input, space_name_input], outputs=space_url_output)
-
         # Вкладка Felguk-ImageResizer
         with gr.Tab("Felguk-ImageResizer"):
             gr.Markdown("## Felguk-ImageResizer: Изменение размера изображения")
@@ -445,7 +350,7 @@ with gr.Blocks() as demo:
         # Вкладка Felguk-PDFMerger
         with gr.Tab("Felguk-PDFMerger"):
             gr.Markdown("## Felguk-PDFMerger: Объединение PDF-файлов")
-            pdf_files_input = gr.File(label="Загрузите PDF-файлы", file_types=[".pdf"], multiple=True)
+            pdf_files_input = gr.Files(label="Загрузите PDF-файлы", file_types=[".pdf"])
             merge_button = gr.Button("Объединить PDF")
             merged_pdf_output = gr.File(label="Результат")
             merge_button.click(fn=merge_pdfs, inputs=pdf_files_input, outputs=merged_pdf_output)