Update templates.py

templates.py

@@ -1033,8 +1033,8 @@ class TemplateManager:
             )
         }
         
-        self.component_index = self._build_component_index()
-        self.category_index = self._build_category_index()
+        self.component_index: Dict[str, List[str]] = self._build_component_index()
+        self.category_index: Dict[str, List[str]] = self._build_category_index()
 
     def _build_component_index(self) -> Dict[str, List[str]]:
         """Build index of templates by component"""
@@ -1142,6 +1142,7 @@ class TemplateManager:
         try:
             data = {
                 name: {
+                    "code": template.code,
                     "description": template.description,
                     "components": template.components,
                     "metadata": template.metadata,
@@ -1155,6 +1156,9 @@ class TemplateManager:
 
             logger.info(f"Templates exported to {path}")
 
+        except FileNotFoundError:
+            logger.error(f"File not found: {path}")
+            raise
         except Exception as e:
             logger.error(f"Error exporting templates: {str(e)}")
             raise
@@ -1167,7 +1171,7 @@ class TemplateManager:
 
             for name, template_data in data.items():
                 self.templates[name] = Template(
-                    code="",  # Code should be loaded separately
+                    code=template_data.get("code", ""),  # Load the actual code from the JSON file
                     description=template_data["description"],
                     components=template_data["components"],
                     metadata=template_data["metadata"],
@@ -1180,6 +1184,12 @@ class TemplateManager:
 
             logger.info(f"Templates imported from {path}")
 
+        except FileNotFoundError:
+            logger.error(f"File not found: {path}")
+            raise
+        except json.JSONDecodeError:
+            logger.error(f"Error decoding JSON from file: {path}")
+            raise
         except Exception as e:
             logger.error(f"Error importing templates: {str(e)}")
             raise