Update app.py

app.py

@@ -18,6 +18,7 @@ import zipfile
 import os
 import torchaudio
 from transformers import pipeline
+from googletrans import Translator
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -254,6 +255,84 @@ def summarize_text(text: str, max_length: int = 100):
     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:
+        return None
+    
+    img = Image.open(image.name)
+    resized_img = img.resize((width, height))
+    
+    # Сохраняем изображение во временный файл
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
+        resized_img.save(tmp_file.name)
+        return tmp_file.name
+
+# Функция для конвертации аудио
+def convert_audio(file, target_format: str):
+    if file is None:
+        return "Файл не загружен."
+    
+    audio = AudioSegment.from_file(file.name)
+    
+    # Сохраняем аудио в новом формате
+    with tempfile.NamedTemporaryFile(delete=False, suffix=f".{target_format.lower()}") as tmp_file:
+        audio.export(tmp_file.name, format=target_format)
+        return tmp_file.name
+
+# Функция для перевода текста
+def translate_text(text: str, target_language: str):
+    translator = Translator()
+    translation = translator.translate(text, dest=target_language)
+    return translation.text
+
+# Функция для обрезки видео
+def trim_video(file, start_time: float, end_time: float):
+    if file is None:
+        return "Файл не загружен."
+    
+    cap = cv2.VideoCapture(file.name)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    
+    # Создаем объект для записи видео
+    output_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+    
+    # Обрезка видео
+    cap.set(cv2.CAP_PROP_POS_MSEC, start_time * 1000)
+    while cap.isOpened():
+        ret, frame = cap.read()
+        if not ret or cap.get(cv2.CAP_PROP_POS_MSEC) >= end_time * 1000:
+            break
+        out.write(frame)
+    
+    cap.release()
+    out.release()
+    return output_path
+
+# Функция для объединения PDF-файлов
+def merge_pdfs(files):
+    if not files:
+        return "Файлы не загружены."
+    
+    merger = PyPDF2.PdfMerger()
+    for file in files:
+        merger.append(file.name)
+    
+    # Сохраняем объединенный PDF во временный файл
+    output_path = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
+    merger.write(output_path)
+    merger.close()
+    return output_path
+
 # Создание интерфейса
 with gr.Blocks() as demo:
     # Заголовок с изображением
@@ -308,35 +387,68 @@ with gr.Blocks() as demo:
             convert_button = gr.Button("Конвертировать")
             convert_button.click(fn=convert_image, inputs=[img_input, img_format], outputs=img_output)
 
-        # Вкладка Felguk-ImageEnhancer
-        with gr.Tab("Felguk-ImageEnhancer"):
-            gr.Markdown("## Felguk-ImageEnhancer: Улучшение качества изображений")
+        # Вкладка 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: Изменение размера изображения")
             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")
-            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: Обрезка аудиофайлов")
+            width_input = gr.Number(label="Ширина", value=300)
+            height_input = gr.Number(label="Высота", value=300)
+            resize_button = gr.Button("Изменить размер")
+            resized_image_output = gr.Image(label="Результат", type="filepath")
+            resize_button.click(fn=resize_image, inputs=[image_input, width_input, height_input], outputs=resized_image_output)
+
+        # Вкладка Felguk-AudioConverter
+        with gr.Tab("Felguk-AudioConverter"):
+            gr.Markdown("## Felguk-AudioConverter: Конвертация аудиофайлов")
             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)
+            target_format_input = gr.Dropdown(
+                choices=["mp3", "wav", "ogg"],
+                label="Выберите формат для конвертации",
+                value="mp3"
+            )
+            convert_button = gr.Button("Конвертировать")
+            converted_audio_output = gr.Audio(label="Результат")
+            convert_button.click(fn=convert_audio, inputs=[audio_input, target_format_input], outputs=converted_audio_output)
+
+        # Вкладка Felguk-TextTranslator
+        with gr.Tab("Felguk-TextTranslator"):
+            gr.Markdown("## Felguk-TextTranslator: Перевод текста")
+            text_input = gr.Textbox(label="Введите текст для перевода", lines=5)
+            target_language_input = gr.Dropdown(
+                choices=["en", "ru", "es", "fr", "de"],
+                label="Выберите язык перевода",
+                value="en"
+            )
+            translate_button = gr.Button("Перевести")
+            translated_text_output = gr.Textbox(label="Результат", interactive=False)
+            translate_button.click(fn=translate_text, inputs=[text_input, target_language_input], outputs=translated_text_output)
+
+        # Вкладка Felguk-VideoTrimmer
+        with gr.Tab("Felguk-VideoTrimmer"):
+            gr.Markdown("## Felguk-VideoTrimmer: Обрезка видеофайлов")
+            video_input = gr.File(label="Загрузите видеофайл", file_types=[".mp4", ".avi"])
+            start_time_input = gr.Number(label="Начало обрезки (сек)", value=0)
+            end_time_input = gr.Number(label="Конец обрезки (сек)", value=10)
+            trim_button = gr.Button("Обрезать видео")
+            trimmed_video_output = gr.Video(label="Результат")
+            trim_button.click(fn=trim_video, inputs=[video_input, start_time_input, end_time_input], outputs=trimmed_video_output)
+
+        # Вкладка Felguk-PDFMerger
+        with gr.Tab("Felguk-PDFMerger"):
+            gr.Markdown("## Felguk-PDFMerger: Объединение PDF-файлов")
+            pdf_files_input = gr.File(label="Загрузите PDF-файлы", file_types=[".pdf"], multiple=True)
+            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)
 
     # Интерфейс чата
     chat_interface = gr.ChatInterface(