deving

src/utils/html_generator.py

@@ -275,25 +275,30 @@ class HTMLGenerator:
             raise
     
     def generate_and_save_visualization(
-        self, 
-        geometry_data: Dict[str, Any], 
-        stp_filename: str
+        self,
+        geometry_data: Dict[str, Any],
+        stp_filename: str,
+        cavity_data: Optional[Dict[str, Any]] = None
     ) -> str:
         """生成并保存可视化HTML文件"""
         try:
             # 生成HTML内容
-            html_content = self.generate_3d_viewer_html(geometry_data, stp_filename)
-            
+            html_content = self.generate_3d_viewer_html(
+                geometry_data,
+                stp_filename,
+                cavity_data=cavity_data
+            )
+
             # 创建文件名
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             safe_filename = stp_filename.replace('.', '_').replace(' ', '_')
             html_filename = f"{safe_filename}_{timestamp}.html"
-            
+
             # 保存文件
             file_path = self.save_html_file(html_content, html_filename)
-            
+
             return file_path
-            
+
         except Exception as e:
             logger.error(f"生成可视化文件失败: {e}")
             raise