diff --git a/templates/index.html b/templates/index.html index 509c460..b422c53 100644 --- a/templates/index.html +++ b/templates/index.html @@ -32,14 +32,30 @@
+ {% set ns = namespace(model_content=[]) %} {% for message in chat_history %} -
- {{ message.role.capitalize() }} -
-
- {{ message.content }} -
+ {% if message.parts and message.parts[0].text and message.parts[0].text.strip() %} + {% if message.role == 'user' %} + {% if ns.model_content %} +
Model
+
{{ ns.model_content | join('') }}
+ {% set ns.model_content = [] %} + {% endif %} + +
User
+
{{ message.parts[0].text }}
+ + {% elif message.role == 'model' %} + {% set _ = ns.model_content.append(message.parts[0].text) %} + {% endif %} + + {% endif %} {% endfor %} + + {% if ns.model_content %} +
Model
+
{{ ns.model_content | join('') }}
+ {% endif %}