fix path in ui

src/ui/callbacks.py

@@ -75,11 +75,25 @@ def format_checklist(steps_state: List[Dict[str, Any]]) -> str:
             lines.append(f"[❌] 🔴 {s['name']} - {s.get('message')}")
     return "\n".join(lines)
 
-def generate_wrapper(user_prompt: str) -> Generator[Tuple[Optional[str], str], None, None]:
+def generate_wrapper(user_prompt: str) -> Generator[
+    Tuple[
+        Optional[str],  # 1. output_3d (Путь к GLB модели)
+        str,            # 2. status_output (Текст чек-листа)
+        Optional[str],  # 3. output_svg (Путь к картинке для превью 2D-плана)
+        Optional[str],  # 4. draft_json_out (Путь к layout_draft.json)
+        Optional[str],  # 5. final_json_out (Путь к layout_final.json)
+        Optional[str],  # 6. svg_file_out (Путь к layout_final.svg для скачивания)
+        Optional[str],  # 7. history_json_out (Путь к llm_history.jsonl)
+        Optional[str]   # 8. draft_svg_file_out (Draft SVG)
+    ], 
+    None, 
+    None
+]:
     """Оркестратор пайплайна с передачей промпта в LLM и запуском 3D-движка."""
     tracker = setup_logger()
     local_logger = logging.getLogger(__name__)
 
+    # Чек-лист остается без изменений...
     steps_state: List[Dict[str, Any]] = [
         {"name": "ИИ-анализ и граф", "status": "pending"},
         {"name": "Инициализация", "status": "pending"},
@@ -90,7 +104,9 @@ def generate_wrapper(user_prompt: str) -> Generator[Tuple[Optional[str], str], N
         {"name": "Экспорт GLB (Draco)", "status": "pending"}
     ]
 
-    yield None, format_checklist(steps_state)
+    # Важно: yield теперь должен возвращать 8 элементов (None для всех файлов, пока идет генерация)
+    yield None, format_checklist(steps_state), None, None, None, None, None, None
+    
     total_start: float = time.time()
     task_queue: queue.Queue = queue.Queue()
 
@@ -106,70 +122,46 @@ def generate_wrapper(user_prompt: str) -> Generator[Tuple[Optional[str], str], N
                 task_queue.put({"type": "step_update", "idx": 0, "update": {"status": "running"}})
                 llm_start: float = time.time()
                 
-                # ПРЕОБРАЗОВАНИЕ В СЫРОЙ СЛОВАРЬ: 
-                # Отключаем авто-разрешение интерполяций (resolve=False), чтобы избежать InterpolationKeyError
                 asset_cfg_raw = OmegaConf.to_container(asset_cfg, resolve=False)
                 
-                # Собираем категории товаров из безопасного словаря
                 categories: List[str] = []
                 for group in asset_cfg_raw.get("products_hierarchy", {}).values():
                     if isinstance(group, dict):
                         categories.extend(list(group.keys()))
                 
-                # Динамический сбор каталога оборудования (Вычисление AABB напрямую из мешей)
                 equipment_lines: List[str] = []
                 fixtures_cfg = asset_cfg_raw.get("fixtures", asset_cfg_raw.get("equipment", {})) 
-                
-                # Достаем базовый путь к папке с ассетами
                 base_assets_dir = asset_cfg_raw.get("assets_dir_path", "assets")
                 
                 for fix_id, fix_data in fixtures_cfg.items():
-                    if not isinstance(fix_data, dict):
-                        continue
-                        
+                    if not isinstance(fix_data, dict): continue
                     raw_path = fix_data.get("asset_file_path", "")
-                    if not raw_path:
-                        continue
-                        
-                    # Разрешаем интерполяцию вручную (меняем ${assets.assets_dir_path} на реальную папку)
+                    if not raw_path: continue
                     clean_path = raw_path.replace("${assets.assets_dir_path}", base_assets_dir)
                     full_path = PROJECT_DIR / clean_path
-                    
                     w, d = None, None
-                    
                     try:
                         import trimesh
-                        # Загружаем геометрию с диска. 
-                        # force='mesh' гарантирует получение объекта с AABB
                         scene = trimesh.load(str(full_path), force='mesh')
-                        
-                        # В 3D обычно: extents[0] = X (ширина), extents[1] = Y (высота), extents[2] = Z (глубина)
                         w = round(float(scene.extents[0]), 2)
                         d = round(float(scene.extents[1]), 2)
-                        
                     except Exception as e:
-                        local_logger.debug(f"⚠️ Не удалось вычислить габариты для {fix_id} ({full_path}): {e}")
+                        local_logger.debug(f"⚠️ Не удалось вычислить габариты для {fix_id}: {e}")
                     
                     if w is not None and d is not None:
-                        # Используем asset_name для описания или ID как запасной вариант
                         desc: str = fix_data.get("asset_name", fix_id)
                         equipment_lines.append(f"- {fix_id} ({desc}): {w}m x {d}m")
                 
                 equipment_catalog: str = "\n".join(equipment_lines)
                 
-                # Защита: если trimesh не установлен или модели не прочитались
                 if not equipment_catalog:
-                    local_logger.warning("⚠️ Не удалось извлечь габариты геометрии, используем фолбэк.")
                     equipment_catalog = (
                         "- showcase_glb (Холодильник пристенный): 1.2m x 0.6m\n"
                         "- shelf_metal (Стеллаж высокий): 1.0m x 0.4m\n"
                         "- small_shelf_two_sided (Островок двусторонний): 1.2m x 0.8m"
                     )
 
-                # Инициализируем LLM клиент
                 client = InventoryLLMClient(main_cfg)
-
-                # Передаем динамический каталог оборудования и габариты комнаты в LLM
                 result = client.generate_layout(
                     user_request=user_prompt, 
                     available_categories=categories,
@@ -205,13 +197,11 @@ def generate_wrapper(user_prompt: str) -> Generator[Tuple[Optional[str], str], N
             local_logger.error(f"Ошибка воркера: {e}", exc_info=True)
             task_queue.put({"type": "error", "message": str(e)})
 
-    # Запускаем наш поток
     thread = threading.Thread(target=worker_thread)
     thread.start()
 
     res_path: Optional[str] = None
     
-    # Ждем только наш конкретный поток, а не все потоки веб-сервера
     while thread.is_alive() or not task_queue.empty():
         try:
             msg = task_queue.get(timeout=0.1)
@@ -220,22 +210,34 @@ def generate_wrapper(user_prompt: str) -> Generator[Tuple[Optional[str], str], N
                 if upd.get("status") == "running" and "start_time" not in steps_state[idx]:
                     steps_state[idx]["start_time"] = time.time()
                 steps_state[idx].update(upd)
-                yield None, format_checklist(steps_state)
+                yield None, format_checklist(steps_state), None, None, None, None, None, None
             elif msg["type"] == "finish": 
                 res_path = msg["result_path"]
             elif msg["type"] == "error": 
                 raise gr.Error(msg["message"])
         except queue.Empty:
-            yield None, format_checklist(steps_state)
+            yield None, format_checklist(steps_state), None, None, None, None, None, None
 
-    # Завершение и передача результата в интерфейс
     if res_path:
         final_text: str = format_checklist(steps_state)
         final_text += "\n" + "-"*40 + "\n"
         final_text += f"🏁 Бэкенд отработал успешно! Время: {time.time() - total_start:.1f} сек.\n"
-        final_text += "⏳ Браузер скачивает геометрию. Идет отрисовка 3D-сцены..."
+        final_text += "⏳ Браузер скачивает артефакты..."
         
-        # Надежно конвертируем путь для компонента Gradio
         absolute_model_path: str = str(Path(res_path).resolve())
-        yield absolute_model_path, final_text
+        
+        # === Обновленный сбор путей к артефактам (всё из папки logs) ===
+        logs_dir = PROJECT_DIR / "logs"
+        
+        # Ищем JSON
+        draft_path: Optional[str] = str(logs_dir / "layout_draft.json") if (logs_dir / "layout_draft.json").exists() else None
+        final_path: Optional[str] = str(logs_dir / "layout_final.json") if (logs_dir / "layout_final.json").exists() else None
+        history_path: Optional[str] = str(logs_dir / "llm_history.jsonl") if (logs_dir / "llm_history.jsonl").exists() else None
+        
+        # Ищем SVG тоже в logs
+        svg_path: Optional[str] = str(logs_dir / "layout_final.svg") if (logs_dir / "layout_final.svg").exists() else None
+        draft_svg_path: Optional[str] = str(logs_dir / "layout_draft.svg") if (logs_dir / "layout_draft.svg").exists() else None
+        
+        # Возвращаем: 3D_модель, Текст, SVG_превью, Draft_JSON, Final_JSON, SVG_файл, History_JSONL, Draft_SVG
+        yield absolute_model_path, final_text, svg_path, draft_path, final_path, svg_path, history_path, draft_svg_path 
         

src/ui/layout.py

@@ -3,6 +3,7 @@
 """
 Модуль верстки пользовательского интерфейса Gradio.
 ИСПРАВЛЕНО: Удалена тема Soft для возврата стандартных шрифтов.
+ДОБАВЛЕНО: Компоненты для просмотра 2D SVG-плана и скачивания JSON-артефактов.
 """
 import gradio as gr
 from omegaconf import DictConfig
@@ -17,7 +18,6 @@ def create_ui(cfg: DictConfig) -> gr.Blocks:
     """
     Создает структуру веб-интерфейса.
     """
-    # Удаляем theme=gr.themes.Soft() для возврата к стандартному виду
     with gr.Blocks(title="ИИ-Мерчандайзер 3D") as app:
         gr.Markdown("# 🏪 Интеллектуальный генератор 3D-планировок")
 
@@ -46,7 +46,7 @@ def create_ui(cfg: DictConfig) -> gr.Blocks:
             )
             generate_btn = gr.Button("🚀 Сгенерировать", variant="primary", scale=1)
 
-        # 4. Область вывода результатов
+        # 4. Область вывода результатов генерации
         with gr.Row():
             with gr.Column(scale=1):
                 status_output = gr.Textbox(
@@ -55,18 +55,46 @@ def create_ui(cfg: DictConfig) -> gr.Blocks:
                     interactive=False,
                     lines=15
                 )
+            
             with gr.Column(scale=3):
-                output_3d = gr.Model3D(
-                    label="Результат 3D-рендера",
-                    clear_color=[0.9, 0.9, 0.9, 1.0],
-                    height=600
-                )
+                # Вкладки для переключения между 3D и 2D
+                with gr.Tabs():
+                    with gr.TabItem("🎮 3D Сцена"):
+                        output_3d = gr.Model3D(
+                            label="Результат 3D-рендера",
+                            clear_color=[0.9, 0.9, 0.9, 1.0],
+                            height=600
+                        )
+                    with gr.TabItem("🗺️ 2D План (SVG)"):
+                        output_svg = gr.Image(
+                            label="Векторный план расстановки",
+                            type="filepath",
+                            interactive=False
+                        )
+
+        # 5. Панель артефактов (для скачивания)
+        with gr.Row():
+            draft_json_out = gr.File(label="📥 Скачать черновик LLM (Draft JSON)")
+            final_json_out = gr.File(label="📥 Скачать финальный расчет (Final JSON)")
+            history_json_out = gr.File(label="📥 Скачать историю LLM (JSONL)")
+            draft_svg_file_out = gr.File(label="📥 Draft SVG")
+            svg_file_out = gr.File(label="📥 Скачать 2D План (SVG)")
 
-        # Привязка событий
+        # Привязка событий: теперь функция должна возвращать 5 значений (3D, Чеклист, SVG-превью, Draft, Final, SVG-файл)
+        # Так как SVG-файл и SVG-превью - это один и тот же файл, мы просто передадим его путь дважды в outputs
         generate_btn.click(
             fn=generate_wrapper,
             inputs=[prompt_input],
-            outputs=[output_3d, status_output],
+            outputs=[
+                output_3d, 
+                status_output, 
+                output_svg, 
+                draft_json_out, 
+                final_json_out, 
+                svg_file_out, 
+                history_json_out,
+                draft_svg_file_out
+            ],
             show_progress="hidden"
         )