diff --git a/backend/core/management/commands/import_template.py b/backend/core/management/commands/import_template.py
index d055293..20074f8 100644
--- a/backend/core/management/commands/import_template.py
+++ b/backend/core/management/commands/import_template.py
@@ -1,5 +1,6 @@
import yaml
from django.core.management.base import BaseCommand
+from django.template import Template, Context
from backend.core.models import DocumentTemplate
class Command(BaseCommand):
@@ -13,7 +14,6 @@ class Command(BaseCommand):
with open(yaml_file_path, 'r') as file:
content = file.read()
- yaml_data = yaml.safe_load(content)
DocumentTemplate.objects.update_or_create(
name="Default Template",
diff --git a/backend/core/templates/document.html b/backend/core/templates/document.html
index f1ec85c..1fa2cc1 100644
--- a/backend/core/templates/document.html
+++ b/backend/core/templates/document.html
@@ -2,35 +2,13 @@
{% block content %}
-
-
-
- {% for segment in segments %}
- {% if segment.segment_type == 'title' %}
- {{ segment.content }}
- {% elif segment.segment_type == 'subtitle' %}
- {{ segment.content }}
- {% elif segment.segment_type == 'h1' %}
- {{ segment.content }}
- {% elif segment.segment_type == 'h2' %}
- {{ segment.content }}
- {% elif segment.segment_type == 'h3' %}
- {{ segment.content }}
- {% elif segment.segment_type == 'quote' %}
- {{ segment.content }}
- {% else %}
- {{ segment.content }}
- {% endif %}
- {% endfor %}
-
-
+ {% if error %}
+ {{ error }}
+ {% endif %}
+
+ {{ rendered_html|safe }}
+