Update app.py

app.py

@@ -19,7 +19,7 @@ else:
     device = "cpu"
 
 d = os.environ.get("DEVICE", None)
-if d != None:
+if d is not None:
     device = d 
 
 model = TSR.from_pretrained(
@@ -61,24 +61,22 @@ def generate(image):
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes)[0]
     mesh = to_gradio_3d_orientation(mesh)
-    mesh_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False)
     mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
-    mesh.export(mesh_path.name)
     mesh.export(mesh_path2.name)
-    return mesh_path.name, mesh_path2.name
+    return mesh_path2.name
 
 
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
         self.primary_hue = "#191a1e"
-        self.background_fill_primary = "#1a1a1a"
-        self.background_fill_secondary = "#1a1a1a"
-        self.background_fill_tertiary = "#1a1a1a"
+        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 = "#1a1a1a"
+        self.input_background_fill = "#191a1e"
         self.input_text_color = "#FFFFFF"
 
 
@@ -93,7 +91,7 @@ footer {
 
 body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
-    background-color: #1a1a1a !important;
+    background-color: #191a1e !important;
     color: #FFFFFF;
 }
 
@@ -104,7 +102,7 @@ h1, h2, h3, h4, h5, h6 {
 }
 
 input[type="text"], textarea {
-    background-color: #1a1a1a !important;
+    background-color: #191a1e !important;
     color: #FFFFFF;
     border: 1px solid #FFFFFF;
 }
@@ -121,7 +119,7 @@ input[type="text"], textarea {
 }
 
 gr-image-upload {
-    background-color: #1a1a1a !important;
+    background-color: #191a1e !important;
     border: 1px dashed #FFFFFF !important;
     position: relative;
 }
@@ -157,27 +155,7 @@ gr-image-upload {
     margin-bottom: 20px;
 }
 
-/* Кнопка сброса возле слайдера */
-.slider-reset-button {
-    width: 25px;
-    height: 25px;
-    margin-left: 10px; /* Отступ от значения слайдера */
-    background-color: #5271FF; /* Цвет кнопки */
-    color: #FFFFFF;
-    border: none;
-    border-radius: 50%; /* Скругление для круглой кнопки */
-    font-size: 14px;
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    cursor: pointer;
-}
-
-.slider-reset-button:hover {
-    background-color: #405BBF; /* Цвет при наведении */
-}
-
-/* Вкладки OBJ/GLB Model */
+/* Вкладки GLB Model */
 .gr-tab {
     background-color: #252525; /* Чуть светлее общий фон */
     border: 1px solid #FFFFFF; /* Белая рамка для структуры */
@@ -189,7 +167,6 @@ gr-image-upload {
     color: #FFFFFF; /* Цвет текста */
     font-weight: bold;
 }
-
 """
 
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
@@ -222,17 +199,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                 value=0.85,
                 step=0.05,
             )
-            reset_button = gr.Button(
-                "↺",  # Символ кнопки сброса
-                elem_classes="slider-reset-button",
-                interactive=True,
-            )
-
-        reset_button.click(
-            fn=lambda: 0.85,  # Значение по умолчанию
-            inputs=[],
-            outputs=[foreground_ratio],
-        )
+            # Кнопка Reset удалена
 
         with gr.Row(elem_id="settings-row"):
             do_remove_background = gr.Checkbox(
@@ -247,11 +214,6 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             )
 
         with gr.Row(elem_id="output-model-container"):
-            with gr.Tab("OBJ Model"):
-                output_model = gr.Model3D(
-                    label="Generated OBJ Model",
-                    interactive=False,
-                )
             with gr.Tab("GLB Model"):
                 output_model2 = gr.Model3D(
                     label="Generated GLB Model",
@@ -269,9 +231,8 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
         ).then(
             fn=generate,
             inputs=[processed_image],
-            outputs=[output_model, output_model2]
+            outputs=[output_model2]
         )
 
-
 demo.queue(max_size=10)
 demo.launch()