Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2
 import numpy as np
 from pydub import AudioSegment
 from langdetect import detect
+from rembg import remove
+import io
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -42,6 +44,13 @@ def process_txt(file):
         text = f.read()
     return text
 
+# Функция для удаления фона с изображения
+def remove_background(image):
+    if image is None:
+        return None
+    output = remove(image)
+    return output
+
 # Функция для определения языка текста
 def detect_language(text):
     try:
@@ -131,7 +140,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         new_chat_button = gr.Button("Новый чат", variant="secondary")
 
-    # Felguk Tools: Txt Analyzer
+    # Felguk Tools: Txt Analyzer и Remove Background
     with gr.Tab("Felguk Tools"):
         gr.Markdown("## Txt Analyzer")
         txt_file = gr.File(label="Загрузите txt файл", file_types=[".txt"])
@@ -139,6 +148,12 @@ with gr.Blocks() as demo:
         analyze_button = gr.Button("Анализировать")
         analyze_button.click(fn=analyze_txt, inputs=txt_file, outputs=txt_output)
 
+        gr.Markdown("## Удаление фона с изображения")
+        image_input = gr.Image(label="Загрузите изображение", type="pil")
+        image_output = gr.Image(label="Результат (без фона)", type="pil")
+        remove_bg_button = gr.Button("Удалить фон")
+        remove_bg_button.click(fn=remove_background, inputs=image_input, outputs=image_output)
+
     # Интерфейс чата
     chat_interface = gr.ChatInterface(
         respond,