Correctly create schemas for Layouts

django_forms_as_jsonschema/forms.py

@@ -4,8 +4,13 @@ from .jsonschema import JSONSchema
 
 
 class JSONSchemaFormMixin:
+    layout = None
+
     def as_jsonschema(self: Form) -> dict:
         schema = JSONSchema()
-        for name, field in self.fields.items():
-            schema.add_field(name, field)
+        if self.layout is None:
+            for name, field in self.fields.items():
+                schema.add_field(name, field)
+        else:
+            schema.update_properties(self.layout.build_schema(schema, self.fields)["properties"])
         return schema.schema