Update app.py

app.py

@@ -22,42 +22,70 @@ def run_setup_script():
 run_setup_script()
 
 
+import gradio as gr
+
 with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="sm")) as demo:
+    # Верхний блок с загрузкой файлов
     with gr.Row():
-        # Левая колонка (ввод параметров)
         with gr.Column(scale=1):
             gr.Markdown("### Параметры генерации")
             
-            pattern_upload = gr.File(label="1. Загрузите лекала (JSON/SVG)", 
-                                   file_types=[".json", ".svg"])
-            
+            # Ряд для загрузки файлов
+            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"
+                    )
+
+    # Основной контент
+    with gr.Row():
+        # Левая панель параметров
+        with gr.Column(scale=1):
             with gr.Accordion("3. Параметры генерации", open=False):
-                gr.Radio(["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf",...], 
-                       label="Front Collar", 
-                       info="Front collar shape")
-                gr.Radio(["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf",...], 
-                       label="Back Collar", 
-                       info="Back collar shape")
+                gr.Radio(
+                    ["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf", 
+                     "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                    label="Front Collar",
+                    info="Форма переднего воротника"
+                )
+                gr.Radio(
+                    ["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf",
+                     "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                    label="Back Collar",
+                    info="Форма заднего воротника"
+                )
 
             with gr.Accordion("4. Render engine", open=False):
-                gr.Dropdown(["BPY", "Pyrender/Open3D","Maya"], 
-                          label="Выберите режим")
+                gr.Dropdown(
+                    ["BPY", "Pyrender/Open3D", "Maya"],
+                    label="Выберите рендер движок",
+                    value="Pyrender/Open3D"
+                )
 
-            with gr.Row():
-                save_obj_btn = gr.Button("Сохранить OBJ")
+        # Правая панель 3D просмотра
+        with gr.Column(scale=3):
+            # Основной 3D вьювер
+            viewer = gr.Model3D(
+                label="3D Просмотрщик",
+                interactive=True,
+                height=600
+            )
 
-        # Правая колонка (3D просмотрщик)
-        with gr.Column(scale=2):
-            gr.Markdown("### ")
-            body_upload = gr.File(label="2. Загрузите 3D-модель тела (OBJ/USD/FBX)", 
-                                file_types=[".obj", ".usd", ".fbx"])
-            
-        gr.Markdown("### 3D Просмотрщик")
-        viewer = gr.Model3D(label="Результат генерации", 
-                            interactive=True)
-            
+            # Панель управления сохранением
             with gr.Row():
-                save_render_btn = gr.Button("Сохранить рендер")
+                with gr.Column(align="center"):
+                    with gr.Row():
+                        save_render_btn = gr.Button("Сохранить рендер", variant="primary")
+                        save_obj_btn = gr.Button("Сохранить OBJ", variant="secondary")
 
     # Логика обновления
     body_upload.change(
@@ -66,7 +94,18 @@ with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="
         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()