Update app.py

app.py

@@ -55,10 +55,27 @@ def process_txt(file):
         text = f.read()
     return text
 
-def remove_background(image):
+# Функция для удаления фона с возможностью выбора фона или цвета
+def remove_background(image, background=None, background_color=None):
     if image is None:
         return "**Ошибка:** Чем я буду удалять, если ты не загрузил изображение?"
+    
+    # Удаляем фон с изображения
     output = remove(image)
+    
+    # Если выбран цвет фона
+    if background_color:
+        background_image = Image.new("RGB", output.size, background_color)
+        background_image.paste(output, mask=output)
+        return background_image
+    
+    # Если загружен фон
+    if background:
+        background_image = Image.open(background).resize(output.size)
+        background_image.paste(output, mask=output)
+        return background_image
+    
+    # Если ничего не выбрано, возвращаем изображение без фона
     return output
 
 def count_objects(image):
@@ -163,63 +180,8 @@ def translate_text(text: str, target_language: str):
     except Exception as e:
         return f"Ошибка перевода: {str(e)}"
 
-# Кастомные CSS-стили
-custom_css = """
-/* Основной фон */
-body, .gradio-container {
-    background-color: #f5f5f5;
-    font-family: 'Arial', sans-serif;
-}
-
-/* Заголовок */
-h1, h2, h3 {
-    color: #333;
-    font-weight: bold;
-}
-
-/* Кнопки */
-button {
-    background-color: #4CAF50;
-    color: white;
-    border: none;
-    padding: 10px 20px;
-    border-radius: 5px;
-    cursor: pointer;
-}
-
-button:hover {
-    background-color: #45a049;
-}
-
-/* Вкладки */
-.tab-button {
-    background-color: #e0e0e0;
-    color: #333;
-    border-radius: 5px 5px 0 0;
-}
-
-.tab-button.selected {
-    background-color: #4CAF50;
-    color: white;
-}
-
-/* Текстовые поля */
-input, textarea {
-    border: 1px solid #ccc;
-    border-radius: 5px;
-    padding: 10px;
-    width: 100%;
-}
-
-/* Изображения */
-img {
-    border-radius: 10px;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
-}
-"""
-
-# Создание интерфейса с кастомными стилями
-with gr.Blocks(css=custom_css) as demo:
+# Создание интерфейса
+with gr.Blocks() as demo:
     # Заголовок с изображением
     gr.HTML("""
     <div style="text-align: center;">
@@ -230,7 +192,7 @@ with gr.Blocks(css=custom_css) as demo:
     # Felguk News
     gr.Markdown("""
     ## Felguk News
-    **Последнее обновление:** 2024-01-03
+    **Последнее обновление:** 2023-10-25  
     - Добавлена поддержка новых форматов файлов.  
     - Улучшена производительность обработки изображений.  
     - Исправлены ошибки в переводе текста.  
@@ -256,9 +218,15 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Tab("Удаление фона"):
             gr.Markdown("## Удаление фона с изображения")
             image_input = gr.Image(label="Загрузите изображение", type="pil")
-            image_output = gr.Image(label="Результат (без фона)", type="pil")
+            background_input = gr.Image(label="Загрузите фон (опционально)", type="pil")
+            background_color_input = gr.ColorPicker(label="Выберите цвет фона (опционально)")
             remove_bg_button = gr.Button("Удалить фон")
-            remove_bg_button.click(fn=remove_background, inputs=image_input, outputs=image_output)
+            image_output = gr.Image(label="Результат (без фона)", type="pil")
+            remove_bg_button.click(
+                fn=remove_background,
+                inputs=[image_input, background_input, background_color_input],
+                outputs=image_output
+            )
 
         # Вкладка Numage
         with gr.Tab("Numage"):