Update app.py

app.py

@@ -1,111 +1,84 @@
 import gradio as gr
-import subprocess
-import os
+from custom_utils.setup_env import run_setup_script
 
-# Запуск setup.sh для настройки окружения и source ~/.bashrc
-def run_setup_script():
-    setup_script_path = os.path.join(os.path.dirname(__file__), "startup.sh")
-    try:
-        print("Running setup script...")
-        # Выполнить setup.sh и source ~/.bashrc в одной сессии bash
-        subprocess.run(
-            f"bash -c 'source ~/.bashrc && bash {setup_script_path}'",
-            shell=True,
-            check=True,
-        )       
-        print("Setup script and environment initialization completed successfully!")
-    except subprocess.CalledProcessError as e:
-        print(f"Error while running setup script or sourcing .bashrc: {e}")
-        exit(1)
-
-# Выполнить setup.sh перед запуском приложения
-run_setup_script()
-
-
-import gradio as gr
 
 with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="sm")) as demo:
-    # Верхний блок с загрузкой файлов
+    # Состояния для хранения выбранных значений
+    front_collar = gr.State()
+    back_collar = gr.State()
+    render_engine = gr.State()
+    
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Параметры генерации")
             
-            # Ряд для загрузки файлов
             with gr.Row():
-                with gr.Column(min_width=300,):
-                    pattern_upload = gr.File(
-                        label="1. Загрузите лекала (JSON/SVG)",
-                        file_types=[".json", ".svg"],
-                        elem_classes="upload-box"
-                    )
-                
-                with gr.Column(min_width=300):
-                    body_upload = gr.File(
-                        label="2. Загрузите 3D-модель тела (OBJ/USD/FBX)",
-                        file_types=[".obj", ".usd", ".fbx"],
-                        elem_classes="upload-box"
-                    )
+                pattern_upload = gr.File(
+                    label="1. Лекала (JSON/SVG)",
+                    file_types=[".json", ".svg"]
+                )
+                body_upload = gr.File(
+                    label="2. Тело (OBJ/USD/FBX)",
+                    file_types=[".obj", ".usd", ".fbx"]
+                )
 
-    # Основной контент
     with gr.Row():
-        # Левая панель параметров
         with gr.Column(scale=1):
             with gr.Accordion("3. Параметры генерации", open=False):
-                gr.Dropdown(
-                    ["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf", 
-                     "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                front_dropdown = gr.Dropdown(
                     label="Front Collar",
-                    info="Форма переднего воротника"
+                    choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf", 
+                            "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                    interactive=True
                 )
-                gr.Dropdown(
-                    ["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf", 
-                     "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                back_dropdown = gr.Dropdown(
                     label="Back Collar",
-                    info="Форма заднего воротника"
+                    choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf", 
+                            "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                    interactive=True
                 )
 
             with gr.Accordion("4. Render engine", open=False):
-                gr.Radio(
-                    ["BPY", "Pyrender/Open3D", "Maya"],
-                    label="Выберите рендер движок",
+                render_radio = gr.Radio(
+                    label="Движок рендеринга",
+                    choices=["BPY", "Pyrender/Open3D", "Maya"],
                     value="Pyrender/Open3D"
                 )
 
-        # Правая панель 3D просмотра
         with gr.Column(scale=3):
-            # Основной 3D вьювер
-            viewer = gr.Model3D(
-                label="3D Просмотрщик",
-                interactive=True,
-                height=600
-            )
-
-            # Панель управления сохранением
+            viewer = gr.Model3D(label="3D Просмотрщик", interactive=True)
+            
             with gr.Row():
-                with gr.Column(show_progress=True):
-                    with gr.Row():
-                        save_render_btn = gr.Button("Сохранить рендер", variant="primary")
-                        save_obj_btn = gr.Button("Сохранить OBJ", variant="secondary")
+                save_render_btn = gr.Button("Сохранить рендер")
+                save_obj_btn = gr.Button("Сохранить OBJ")
+
+    # Обработчики событий
+    front_dropdown.change(
+        lambda x: x,
+        inputs=front_dropdown,
+        outputs=front_collar
+    )
+    
+    back_dropdown.change(
+        lambda x: x,
+        inputs=back_dropdown,
+        outputs=back_collar
+    )
+    
+    render_radio.change(
+        lambda x: x,
+        inputs=render_radio,
+        outputs=render_engine
+    )
 
-    # Логика обновления
     body_upload.change(
-        fn=lambda file: file.name if file else None,
+        lambda file: file.name if file else None,
         inputs=body_upload,
         outputs=viewer
     )
 
-# Стилизация
-demo.css = """
-.upload-box {
-    margin: 1rem 0; 
-    padding: 1rem;
-    border: 2px dashed #e0e0e0;
-    border-radius: 8px;
-}
-"""
-
-
-# Запуск приложения
 if __name__ == "__main__":
     run_setup_script()
     demo.launch()
+
+