Update app.py

app.py

@@ -5,19 +5,33 @@ import os
 import subprocess
 import shutil
 import threading
+from concurrent.futures import ThreadPoolExecutor
+import logging
+
+logging.basicConfig(level=logging.INFO)
 
 
 iframe_html = """
 <iframe src="http://localhost:8080" width="100%" height="500" style="border:none;"></iframe>
 """
 
+async def run_simulation(data_path, config_path):
+    proc = await asyncio.create_subprocess_exec(
+        "python3", "GarmentCode/pattern_data_sim.py",
+        "--data", data_path,
+        "--config", config_path
+    )
+    await proc.wait()  # дождаться завершения процесса
+    return
 
-
-async def start_smpl():
-    proc = await asyncio.create_subprocess_exec("python", "smpl_visualizer.py")
-    # Можно добавить логику ожидания запуска
-    await proc.wait()
-    return proc
+async def run_fitting(data_path, config_path):
+    proc = await asyncio.create_subprocess_exec(
+        "python3", "GarmentCode/pattern_fitter.py",
+        "--data", data_path,
+        "--config", config_path
+    )
+    await proc.wait()  # дождаться завершения процесса
+    return
 
 def organize_images_into_structure(source_folder, new_folder):
     os.makedirs(new_folder, exist_ok=True)
@@ -31,34 +45,38 @@ def organize_images_into_structure(source_folder, new_folder):
             shutil.copy(source_file, destination_file)
 
 def update_submenu(clothing_type):
-    """Функция для динамического обновления подменю в зависимости от типа одежды."""
     if clothing_type == "Top":
         return (
-            gr.Dropdown.update(visible=True, label="Front Collar",
-                               choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf",
-                                        "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"]),
-            gr.Dropdown.update(visible=True, label="Back Collar",
-                               choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf",
-                                        "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"])
+            gr.Dropdown(visible=True, label="Front Collar",
+                        choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf",
+                                 "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                        interactive=True),
+            gr.Dropdown(visible=True, label="Back Collar",
+                        choices=["CircleNeckHalf", "CurvyNeckHalf", "VNeckHalf", "SquareNeckHalf",
+                                 "TrapezoidNeckHalf", "CircleArcNeckHalf", "Bezier2NeckHalf"],
+                        interactive=True)
         )
     elif clothing_type == "Down":
         return (
-            gr.Dropdown.update(visible=True, label="Waist Style",
-                               choices=["High Waist", "Low Waist", "Medium Waist"]),
-            gr.Dropdown.update(visible=False)
+            gr.Dropdown(visible=True, label="Waist Style",
+                        choices=["High Waist", "Low Waist", "Medium Waist"],
+                        interactive=True),
+            gr.Dropdown(visible=False, interactive=True)
         )
     elif clothing_type == "Full body":
         return (
-            gr.Dropdown.update(visible=True, label="Overall Style",
-                               choices=["Slim Fit", "Loose Fit", "Regular Fit"]),
-            gr.Dropdown.update(visible=False)
+            gr.Dropdown(visible=True, label="Overall Style",
+                        choices=["Slim Fit", "Loose Fit", "Regular Fit"],
+                        interactive=True),
+            gr.Dropdown(visible=False, interactive=True)
         )
     else:
         return (
-            gr.Dropdown.update(visible=False),
-            gr.Dropdown.update(visible=False)
+            gr.Dropdown(visible=False, interactive=True),
+            gr.Dropdown(visible=False, interactive=True)
         )
 
+
 with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="sm")) as demo:
     front_collar = gr.State()
     back_collar = gr.State()
@@ -68,6 +86,7 @@ with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="
         # Левая колонка: рендер и кнопки
         with gr.Column(scale=3):
             viewer = gr.HTML(iframe_html, label="3D Рендер")
+            start_sim = gr.Button("Запустить симуляцию",variant="primary",scale=2)
             with gr.Row():
                 save_render_btn = gr.Button("Сохранить рендер")
                 save_obj_btn = gr.Button("Сохранить OBJ")
@@ -128,8 +147,8 @@ with gr.Blocks(title="3D Garment Generator", theme=gr.themes.Default(text_size="
 
 def run_smpl():
     subprocess.run([
-        "python3", "custom_utils/smpl_visualizer.py",
-        "--model-path", "custom_utils/SMPL_FEMALE.npz"
+        "python3", "smpl_visualizer.py",
+        "--model-path", "/Users/user/Downloads/SMPL/SMPL_julia/SMPL_FEMALE.npz"
     ])
 
 if __name__ == "__main__":