Update app.py

app.py

@@ -21,72 +21,44 @@ def run_setup_script():
 # Выполнить setup.sh перед запуском приложения
 run_setup_script()
 
-# Структура состояния приложения
-class GUIState:
-    def __init__(self):
-        self.pattern_state = None  # Заглушка для состояния лекал
-        self.local_path_3d = Path('./tmp_gui/garm_3d')
-        self.local_path_3d.mkdir(parents=True, exist_ok=True)
-        self.garm_3d_filename = None
 
-    def update_pattern_ui_state(self, param_dict=None, param=None, new_value=None):
-        """Обновление состояния лекал (заглушка)"""
-        print("Обновление лекал...")
-
-    def update_3d_scene(self):
-        """Обновление 3D модели (заглушка)"""
-        print("Обновление 3D модели...")
-        self.garm_3d_filename = f'garm_3d_{time.time()}.glb'
-        shutil.copy2('./assets/example.glb', self.local_path_3d / self.garm_3d_filename)
-        return str(self.local_path_3d / self.garm_3d_filename)
-
-# Инициализация состояния
-state = GUIState()
-
-# Функция для загрузки файлов лекал и тела
-def upload_files(pattern_file, body_file):
-    if pattern_file:
-        print(f"Загружен файл лекал: {pattern_file.name}")
-    if body_file:
-        print(f"Загружен файл тела: {body_file.name}")
-    return "Файлы успешно загружены!"
-
-# Функция для обновления 3D сцены
-def update_3d():
-    path_to_model = state.update_3d_scene()
-    return path_to_model
-
-# Интерфейс Gradio
-with gr.Blocks() as demo:
+with gr.Blocks(title="3D Garment Generator") as demo:
     with gr.Row():
-        # Левая панель: загрузка файлов и параметры
+        # Левый столбец (ввод параметров)
         with gr.Column(scale=1):
-            gr.Markdown("### Парамры генерации")
-            pattern_upload = gr.File(label="Загрузите лекала (JSON/SVG)", file_types=[".json", ".svg"])
-            body_upload = gr.File(label="Загрузите модель тела (OBJ/USD/FBX)", file_types=[".obj", ".usd", ".fbx"])
-            upload_button = gr.Button("Загрузить файлы")
-            upload_status = gr.Textbox(label="Статус загрузки", interactive=False)
-
-            upload_button.click(
-                fn=upload_files,
-                inputs=[pattern_upload, body_upload],
-                outputs=upload_status,
-            )
-
-            gr.Markdown("### Дополнительные параметры")
-            dropdown_menu = gr.Dropdown(["Опция 1", "Опция 2"], label="Выберите режим")
+            gr.Markdown("### Параметры генерации")
+            
+            # Загрузка файлов
+            pattern_upload = gr.File(label="1. Загрузите лекала (JSON/SVG)", file_types=[".json", ".svg"])
+            body_upload = gr.File(label="2. Загрузите 3D-модель тела (OBJ/USD/FBX)", file_types=[".obj", ".usd", ".fbx"])
+            
+            # Параметры (заглушки)
+            with gr.Accordion("3. Параметры генерации", open=False):
+                gr.Textbox("Скоро добавим...")
+            
+            # Дополнительные опции (заглушки)
+            with gr.Accordion("4. Дополнительные настройки", open=False):
+                gr.Dropdown(["Опция 1", "Опция 2"], label="Выберите режим")
 
-        # Правая панель: 3D просмотрщик
+        # Правый столбец (3D просмотрщик)
         with gr.Column(scale=2):
             gr.Markdown("### 3D Просмотрщик")
+            
+            # 3D просмотрщик (поддерживает OBJ/GLB)
             viewer = gr.Model3D(label="Результат генерации", interactive=True)
-            generate_button = gr.Button("Сгенерировать 3D модель")
             
-            generate_button.click(
-                fn=update_3d,
-                inputs=[],
-                outputs=viewer,
-            )
+            # Кнопки сохранения
+            with gr.Row():
+                save_obj_btn = gr.Button("Сохранить OBJ")
+                save_render_btn = gr.Button("Сохранить рендер")
+
+    # Логика обновления просмотрщика
+    body_upload.change(
+        fn=lambda file: file.name if file else None,
+        inputs=body_upload,
+        outputs=viewer
+    )
 
 # Запуск приложения
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()