Update app.py

app.py

@@ -22,45 +22,60 @@ model.to(device)
 
 # Функция для проверки изображения
 def check_input_image(input_image):
-    if input_image is None:
-        raise gr.Error("No image uploaded!")
+    try:
+        if input_image is None:
+            raise gr.Error("No image uploaded!")
+    except Exception as e:
+        print(f"Error in check_input_image: {e}")
+        raise
 
 
 # Функция обработки изображения
 def preprocess(input_image, do_remove_background, foreground_ratio):
-    def fill_background(image):
-        image = np.array(image).astype(np.float32) / 255.0
-        image = image[:, :, :3] * image[:, :, 3:4] + (1 - image[:, :, 3:4]) * 0.5
-        image = Image.fromarray((image * 255.0).astype(np.uint8))
-        return image
-
-    if do_remove_background:
-        image = input_image.convert("RGB")
-        image = remove_background(image)
-        image = resize_foreground(image, foreground_ratio)
-        image = fill_background(image)
-    else:
-        image = input_image
-        if image.mode == "RGBA":
+    try:
+        def fill_background(image):
+            image = np.array(image).astype(np.float32) / 255.0
+            image = image[:, :, :3] * image[:, :, 3:4] + (1 - image[:, :, 3:4]) * 0.5
+            image = Image.fromarray((image * 255.0).astype(np.uint8))
+            return image
+
+        if do_remove_background:
+            image = input_image.convert("RGB")
+            image = remove_background(image)
+            image = resize_foreground(image, foreground_ratio)
             image = fill_background(image)
-    return image
+        else:
+            image = input_image
+            if image.mode == "RGBA":
+                image = fill_background(image)
+        return image
+    except Exception as e:
+        print(f"Error in preprocess: {e}")
+        raise
 
 
 # Функция генерации 3D модели
 def generate(image):
-    time.sleep(3)  # Эмуляция времени обработки
-    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
-
-
+    try:
+        time.sleep(3)  # Эмуляция времени обработки
+        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
+    except Exception as e:
+        print(f"Error in generate: {e}")
+        raise
+
+
+# Функции управления лоадером
 def start_loading():
+    print("Start loading triggered")
     return gr.HTML.update(visible=True)
 
 def stop_loading():
+    print("Stop loading triggered")
     return gr.HTML.update(visible=False)
 
 
@@ -101,54 +116,23 @@ h1, h2, h3, h4, h5, h6 {
     font-weight: 700;
     color: #FFFFFF;
 }
-/* Стиль для текстовых полей и кнопок */
-input[type="text"], textarea {
-    background-color: #191a1e !important;
-    color: #FFFFFF;
-    border: 1px solid #FFFFFF;
-}
-/* Слайдер */
-.gr-slider {
-    height: 40px !important; /* Увеличение высоты слайдера */
-    transition: all 0.3s ease !important; /* Плавные переходы */
-}
-.gr-slider .slider-value {
-    display: none !important; /* Прячем текст значения */
-}
-.gr-slider .slider::-webkit-slider-thumb {
-    height: 15px !important;
-    width: 15px !important;
-    background-color: #5271FF !important; /* Цвет совпадает с кнопкой */
-    border-radius: 50%;
-}
-.gr-slider .slider::-webkit-slider-runnable-track {
-    background: #5271FF !important; /* Цвет заполнения слайдера */
-}
-/* Кнопка Generate */
-.generate-button {
-    background-color: #5271FF !important;
-    color: #FFFFFF !important;
-    border: none;
-    font-weight: bold;
-}
-.generate-button:hover {
-    background-color: #405BBF !important; /* Цвет при наведении */
-}
+/* Лоадер внутри кнопки */
 .generate-button {
     position: relative; /* Кнопка становится родительским контейнером */
+    display: flex;
+    align-items: center;
+    justify-content: center;
 }
-/* Выделяем текст для Prompt */
-.prompt-text {
-    font-weight: bold;
-    color: #FFFFFF;
-}
-/* Лоадер (анимация) */
+
+/* Лоадер внутри кнопки */
 #loading-bar {
     display: none; /* Скрыт по умолчанию */
-    position: relative; /* Изменено на relative для использования в кнопке */
-    margin: 0 auto; /* Центрируем внутри родительского элемента */
-    width: 30px; /* Уменьшен размер для кнопки */
-    height: 30px;
+    position: absolute; /* Абсолютная позиция внутри кнопки */
+    top: 50%;
+    left: 50%;
+    transform: translate(-50%, -50%);
+    width: 20px;
+    height: 20px;
     border: 4px solid #f3f3f3;
     border-top: 4px solid #5271FF; /* Цвет лоадера */
     border-radius: 50%;
@@ -156,12 +140,8 @@ input[type="text"], textarea {
 }
 /* Анимация вращения */
 @keyframes spin {
-    0% { transform: translate(-50%, -50%) rotate(0deg); }
-    100% { transform: translate(-50%, -50%) rotate(360deg); }
-}
-/* Текст CheckBox в белый */
-.gr-checkbox label {
-    color: #FFFFFF !important;
+    0% { transform: rotate(0deg); }
+    100% { transform: rotate(360deg); }
 }
 """
 
@@ -192,11 +172,11 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             step=0.05,
         )
         do_remove_background = gr.Checkbox(
-            label="Remove Background",  # Текст перекрашен в белый через CSS
+            label="Remove Background",
             value=True,
         )
         submit = gr.Button("Generate", elem_classes="generate-button")
-        loading_bar = gr.HTML("<div id='loading-bar'></div>")
+        loading_bar = gr.HTML("<div id='loading-bar'></div>", visible=False)
         output_model = gr.Model3D(
             label="Generated GLB Model",
             interactive=False,
@@ -204,29 +184,30 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
         )
 
     submit.click(
-    fn=start_loading,  # Включить прогресс-бар
-    inputs=[], 
-    outputs=[loading_bar]
-).then(
-    fn=check_input_image,
-    inputs=[input_image],
-    outputs=[]
-).then(
-    fn=preprocess,
-    inputs=[input_image, do_remove_background, foreground_ratio],
-    outputs=[processed_image]
-).then(
-    fn=generate,
-    inputs=[processed_image],
-    outputs=[output_model]
-).then(
-    fn=stop_loading,  # Отключаем прогресс-бар после завершения генерации
-    inputs=[],
-    outputs=[loading_bar]
-)
+        fn=start_loading,  # Включить прогресс-бар
+        inputs=[], 
+        outputs=[loading_bar]
+    ).then(
+        fn=check_input_image,
+        inputs=[input_image],
+        outputs=[]
+    ).then(
+        fn=preprocess,
+        inputs=[input_image, do_remove_background, foreground_ratio],
+        outputs=[processed_image]
+    ).then(
+        fn=generate,
+        inputs=[processed_image],
+        outputs=[output_model]
+    ).then(
+        fn=stop_loading,  # Отключаем прогресс-бар после завершения генерации
+        inputs=[],
+        outputs=[loading_bar]
+    )
+
 # Запуск приложения
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     share=True,
-)
+)