qbhf2 commited on
Commit
b001341
·
verified ·
1 Parent(s): 27d7f7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +62 -32
app.py CHANGED
@@ -21,42 +21,72 @@ def run_setup_script():
21
  # Выполнить setup.sh перед запуском приложения
22
  run_setup_script()
23
 
24
- # Основное приложение Gradio
25
- with gr.Blocks(title="3D Garment Generator") as demo:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  with gr.Row():
27
- # Левый столбец (ввод параметров)
28
  with gr.Column(scale=1):
29
  gr.Markdown("### Параметры генерации")
30
-
31
- # Загрузка файлов
32
- pattern_upload = gr.File(label="1. Загрузите лекала (JSON/SVG)", file_types=[".json", ".svg"])
33
- body_upload = gr.File(label="2. Загрузите 3D-модель тела (OBJ/USD/FBX)", file_types=[".obj", ".usd", ".fbx"])
34
-
35
- # Параметры (заглушки)
36
- with gr.Accordion("3. Параметры генерации", open=False):
37
- gr.Textbox("Скоро добавим...")
38
-
39
- # Дополнительные опции (заглушки)
40
- with gr.Accordion("4. Дополнительные настройки", open=False):
41
- gr.Dropdown(["Опция 1", "Опция 2"], label="Выберите режим")
42
 
43
- # Правый столбец (3D просмотрщик)
 
 
 
44
  with gr.Column(scale=2):
45
- gr.Markdown("### 3D Просмотр")
46
-
47
- # 3D просмотрщик (поддерживает OBJ/GLB)
48
  viewer = gr.Model3D(label="Результат генерации", interactive=True)
 
49
 
50
- # Кнопки сохранения
51
- with gr.Row():
52
- save_obj_btn = gr.Button("Сохранить OBJ")
53
- save_render_btn = gr.Button("Сохранить рендер")
54
-
55
- # Логика обновления просмотрщика
56
- body_upload.change(
57
- fn=lambda file: file.name if file else None,
58
- inputs=body_upload,
59
- outputs=viewer
60
- )
61
-
62
- demo.launch()
 
21
  # Выполнить setup.sh перед запуском приложения
22
  run_setup_script()
23
 
24
+ # Структура состояния приложения
25
+ class GUIState:
26
+ def __init__(self):
27
+ self.pattern_state = None # Заглушка для состояния лекал
28
+ self.local_path_3d = Path('./tmp_gui/garm_3d')
29
+ self.local_path_3d.mkdir(parents=True, exist_ok=True)
30
+ self.garm_3d_filename = None
31
+
32
+ def update_pattern_ui_state(self, param_dict=None, param=None, new_value=None):
33
+ """Обновление состояния лекал (заглушка)"""
34
+ print("Обновление лекал...")
35
+
36
+ def update_3d_scene(self):
37
+ """Обновление 3D модели (заглушка)"""
38
+ print("Обновление 3D модели...")
39
+ self.garm_3d_filename = f'garm_3d_{time.time()}.glb'
40
+ shutil.copy2('./assets/example.glb', self.local_path_3d / self.garm_3d_filename)
41
+ return str(self.local_path_3d / self.garm_3d_filename)
42
+
43
+ # Инициализация состояния
44
+ state = GUIState()
45
+
46
+ # Функция для загрузки файлов лекал и тела
47
+ def upload_files(pattern_file, body_file):
48
+ if pattern_file:
49
+ print(f"Загружен файл лекал: {pattern_file.name}")
50
+ if body_file:
51
+ print(f"Загружен файл тела: {body_file.name}")
52
+ return "Файлы успешно загружены!"
53
+
54
+ # Функция для обновления 3D сцены
55
+ def update_3d():
56
+ path_to_model = state.update_3d_scene()
57
+ return path_to_model
58
+
59
+ # Интерфейс Gradio
60
+ with gr.Blocks() as demo:
61
  with gr.Row():
62
+ # Левая панель: загрузка файлов и параметры
63
  with gr.Column(scale=1):
64
  gr.Markdown("### Параметры генерации")
65
+ pattern_upload = gr.File(label="Загрузите лекала (JSON/SVG)", file_types=[".json", ".svg"])
66
+ body_upload = gr.File(label="Загрузите модель тела (OBJ/USD/FBX)", file_types=[".obj", ".usd", ".fbx"])
67
+ upload_button = gr.Button("Загрузить файлы")
68
+ upload_status = gr.Textbox(label="Статус загрузки", interactive=False)
69
+
70
+ upload_button.click(
71
+ fn=upload_files,
72
+ inputs=[pattern_upload, body_upload],
73
+ outputs=upload_status,
74
+ )
 
 
75
 
76
+ gr.Markdown("### Дополнительные параметры")
77
+ dropdown_menu = gr.Dropdown(["Опция 1", "Опция 2"], label="Выберите режим")
78
+
79
+ # Правая панель: 3D просмотрщик
80
  with gr.Column(scale=2):
81
+ gr.Markdown("### 3D Просмотрщик")
 
 
82
  viewer = gr.Model3D(label="Результат генерации", interactive=True)
83
+ generate_button = gr.Button("Сгенерировать 3D модель")
84
 
85
+ generate_button.click(
86
+ fn=update_3d,
87
+ inputs=[],
88
+ outputs=viewer,
89
+ )
90
+
91
+ # Запуск приложения
92
+ demo.launch()