Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -123,10 +123,11 @@ def render_message(history):
|
|
| 123 |
<div id="chatbox-container" class="chatbox" style="height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex; flex-direction: column-reverse;">
|
| 124 |
<div id="messages" style="display: block; margin-bottom: 10px;">"""
|
| 125 |
|
| 126 |
-
seen_messages = set()
|
| 127 |
|
| 128 |
allowed_tags = [
|
| 129 |
-
'p', 'strong', 'em', 'ul', 'ol', 'li', 'a', 'code', 'pre', 'br', 'blockquote', 'hr',
|
|
|
|
| 130 |
]
|
| 131 |
allowed_attributes = {
|
| 132 |
'*': ['class', 'style'],
|
|
@@ -138,10 +139,12 @@ def render_message(history):
|
|
| 138 |
if user_message and ("user", user_message) not in seen_messages:
|
| 139 |
seen_messages.add(("user", user_message))
|
| 140 |
user_message_html = markdown.markdown(
|
| 141 |
-
user_message,
|
| 142 |
extensions=["fenced_code", "codehilite"]
|
| 143 |
)
|
| 144 |
-
user_message_html = bleach.clean(
|
|
|
|
|
|
|
| 145 |
messages_html += f"""
|
| 146 |
<div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
|
| 147 |
<img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
|
|
@@ -151,10 +154,12 @@ def render_message(history):
|
|
| 151 |
if assistant_message and ("assistant", assistant_message) not in seen_messages:
|
| 152 |
seen_messages.add(("assistant", assistant_message))
|
| 153 |
assistant_message_html = markdown.markdown(
|
| 154 |
-
assistant_message,
|
| 155 |
extensions=["fenced_code", "codehilite"]
|
| 156 |
)
|
| 157 |
-
assistant_message_html = bleach.clean(
|
|
|
|
|
|
|
| 158 |
messages_html += f"""
|
| 159 |
<div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
|
| 160 |
<img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
|
|
|
|
| 123 |
<div id="chatbox-container" class="chatbox" style="height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex; flex-direction: column-reverse;">
|
| 124 |
<div id="messages" style="display: block; margin-bottom: 10px;">"""
|
| 125 |
|
| 126 |
+
seen_messages = set()
|
| 127 |
|
| 128 |
allowed_tags = [
|
| 129 |
+
'p', 'strong', 'em', 'ul', 'ol', 'li', 'a', 'code', 'pre', 'br', 'blockquote', 'hr',
|
| 130 |
+
'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'img'
|
| 131 |
]
|
| 132 |
allowed_attributes = {
|
| 133 |
'*': ['class', 'style'],
|
|
|
|
| 139 |
if user_message and ("user", user_message) not in seen_messages:
|
| 140 |
seen_messages.add(("user", user_message))
|
| 141 |
user_message_html = markdown.markdown(
|
| 142 |
+
user_message,
|
| 143 |
extensions=["fenced_code", "codehilite"]
|
| 144 |
)
|
| 145 |
+
user_message_html = bleach.clean(
|
| 146 |
+
user_message_html, tags=allowed_tags, attributes=allowed_attributes, strip=True
|
| 147 |
+
)
|
| 148 |
messages_html += f"""
|
| 149 |
<div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
|
| 150 |
<img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
|
|
|
|
| 154 |
if assistant_message and ("assistant", assistant_message) not in seen_messages:
|
| 155 |
seen_messages.add(("assistant", assistant_message))
|
| 156 |
assistant_message_html = markdown.markdown(
|
| 157 |
+
assistant_message,
|
| 158 |
extensions=["fenced_code", "codehilite"]
|
| 159 |
)
|
| 160 |
+
assistant_message_html = bleach.clean(
|
| 161 |
+
assistant_message_html, tags=allowed_tags, attributes=allowed_attributes, strip=True
|
| 162 |
+
)
|
| 163 |
messages_html += f"""
|
| 164 |
<div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
|
| 165 |
<img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
|