Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import time  # Для эмуляции времени загрузки
+import time  # For simulating loading times
 import tempfile
 import numpy as np
 import torch
@@ -7,10 +7,10 @@ from PIL import Image
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
-# Проверяем наличие GPU
+# Check for GPU availability
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-# Загружаем модель
+# Load the model
 model = TSR.from_pretrained(
     "stabilityai/TripoSR",
     config_name="config.yaml",
@@ -19,14 +19,12 @@ model = TSR.from_pretrained(
 model.renderer.set_chunk_size(131072)
 model.to(device)
 
-
-# Функция для проверки изображения
+# Function to check image input
 def check_input_image(input_image):
     if input_image is None:
         raise gr.Error("No image uploaded!")
 
-
-# Функция обработки изображения
+# Image preprocessing
 def preprocess(input_image, do_remove_background, foreground_ratio):
     def fill_background(image):
         image = np.array(image).astype(np.float32) / 255.0
@@ -45,74 +43,65 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
             image = fill_background(image)
     return image
 
-
-# Функция генерации 3D модели
+# 3D model generation
 def generate(image):
-    time.sleep(3)  # Эмуляция времени обработки
+    time.sleep(3)  # Simulate processing time
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes)[0]
     mesh = to_gradio_3d_orientation(mesh)
     mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
     mesh.export(mesh_path2.name)
     return mesh_path2.name
+
+# Loading bar controls
 def start_loading(loader_id):
     return f"<script>document.getElementById('{loader_id}').style.display = 'block';</script>"
 
 def stop_loading(loader_id):
     return f"<script>document.getElementById('{loader_id}').style.display = 'none';</script>"
 
-# Настройка темы и CSS
-class CustomTheme(gr.themes.Base):
-    def __init__(self):
-        super().__init__()
-        self.primary_hue = "#191a1e"
-        self.background_fill_primary = "#191a1e"
-        self.background_fill_secondary = "#191a1e"
-        self.background_fill_tertiary = "#191a1e"
-        self.text_color_primary = "#FFFFFF"
-        self.text_color_secondary = "#FFFFFF"
-        self.text_color_tertiary = "#FFFFFF"
-        self.input_background_fill = "#191a1e"
-        self.input_text_color = "#FFFFFF"
-
-
+# Custom CSS and theme
 css = """
-/* Общий стиль лоадеров */
+/* Loaders inside their respective rows */
 .loader {
-    display: none; /* Скрыт по умолчанию */
+    display: none;
     position: absolute;
     top: 50%;
     left: 50%;
     transform: translate(-50%, -50%);
     width: 40px;
     height: 40px;
-    border: 5px solid #f3f3f3; /* Белый круг */
-    border-top: 5px solid #5271FF; /* Синий круг */
+    border: 5px solid #f3f3f3;
+    border-top: 5px solid #5271FF;
     border-radius: 50%;
     animation: spin 1s linear infinite;
 }
 
-/* Анимация вращения */
 @keyframes spin {
-    0% {
-        transform: translate(-50%, -50%) rotate(0deg);
-    }
-    100% {
-        transform: translate(-50%, -50%) rotate(360deg);
-    }
+    0% { transform: translate(-50%, -50%) rotate(0deg); }
+    100% { transform: translate(-50%, -50%) rotate(360deg); }
 }
 
-/* Добавить позицию для контейнеров */
+/* Loader container positions */
 #image-container,
 #process-container,
 #generate-container {
-    position: relative; /* Для размещения лоадера внутри */
+    position: relative;
+}
+
+/* Button styling remains as original */
+.generate-button {
+    background-color: #5271FF !important;
+    color: #FFFFFF !important;
+    border: none;
+    font-weight: bold;
 }
 """
 
-# Интерфейс
+# Gradio Interface
 with gr.Blocks(css=css) as demo:
     with gr.Column():
+        # Row 1: Image upload
         with gr.Row(elem_id="image-container"):
             input_image = gr.Image(
                 label="Upload Image",
@@ -129,7 +118,8 @@ with gr.Blocks(css=css) as demo:
                 height=300,
             )
             loading_bar_image = gr.HTML("<div class='loader' id='image-loader'></div>")
-        
+
+        # Row 2: Processing options
         with gr.Row(elem_id="process-container"):
             foreground_ratio = gr.Slider(
                 label="Foreground Ratio",
@@ -140,7 +130,8 @@ with gr.Blocks(css=css) as demo:
             )
             do_remove_background = gr.Checkbox(label="Remove Background", value=True)
             loading_bar_process = gr.HTML("<div class='loader' id='process-loader'></div>")
-        
+
+        # Row 3: Generate button and 3D model
         with gr.Row(elem_id="generate-container"):
             submit = gr.Button("Generate", elem_classes="generate-button")
             output_model = gr.Model3D(
@@ -149,46 +140,47 @@ with gr.Blocks(css=css) as demo:
                 elem_classes="gr-model3d-container",
             )
             loading_bar_generate = gr.HTML("<div class='loader' id='generate-loader'></div>")
-            # Обновленная цепочка действий для кнопки submit
-    submit.click(
-        fn=lambda: start_loading('image-loader'),  # Показать лоадер для загрузки изображения
-        inputs=[],
-        outputs=[loading_bar_image]
-    ).then(
-        fn=check_input_image,  # Проверить изображение
-        inputs=[input_image],
-        outputs=[]
-    ).then(
-        fn=lambda: stop_loading('image-loader'),  # Скрыть лоадер для загрузки
-        inputs=[],
-        outputs=[loading_bar_image]
-    ).then(
-        fn=lambda: start_loading('process-loader'),  # Показать лоадер для обработки
-        inputs=[],
-        outputs=[loading_bar_process]
-    ).then(
-        fn=preprocess,  # Обработка изображения
-        inputs=[input_image, do_remove_background, foreground_ratio],
-        outputs=[processed_image]
-    ).then(
-        fn=lambda: stop_loading('process-loader'),  # Скрыть лоадер для обработки
-        inputs=[],
-        outputs=[loading_bar_process]
-    ).then(
-        fn=lambda: start_loading('generate-loader'),  # Показать лоадер для генерации
-        inputs=[],
-        outputs=[loading_bar_generate]
-    ).then(
-        fn=generate,  # Генерация модели
-        inputs=[processed_image],
-        outputs=[output_model]
-    ).then(
-        fn=lambda: stop_loading('generate-loader'),  # Скрыть лоадер для генерации
-        inputs=[],
-        outputs=[loading_bar_generate]
-    )
-     
 
+        # Submit button functionality
+        submit.click(
+            fn=lambda: start_loading("image-loader"),
+            inputs=[],
+            outputs=[loading_bar_image],
+        ).then(
+            fn=check_input_image,
+            inputs=[input_image],
+            outputs=[]
+        ).then(
+            fn=lambda: stop_loading("image-loader"),
+            inputs=[],
+            outputs=[loading_bar_image],
+        ).then(
+            fn=lambda: start_loading("process-loader"),
+            inputs=[],
+            outputs=[loading_bar_process],
+        ).then(
+            fn=preprocess,
+            inputs=[input_image, do_remove_background, foreground_ratio],
+            outputs=[processed_image],
+        ).then(
+            fn=lambda: stop_loading("process-loader"),
+            inputs=[],
+            outputs=[loading_bar_process],
+        ).then(
+            fn=lambda: start_loading("generate-loader"),
+            inputs=[],
+            outputs=[loading_bar_generate],
+        ).then(
+            fn=generate,
+            inputs=[processed_image],
+            outputs=[output_model],
+        ).then(
+            fn=lambda: stop_loading("generate-loader"),
+            inputs=[],
+            outputs=[loading_bar_generate],
+        )
+
+# Launch app
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,