Update app.py

app.py

@@ -96,7 +96,7 @@ def render_message(history):
             messages_html += f"""
             <div style='display: flex; align-items: center; margin-bottom: 10px;'>
                 <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
-                <span style='color: white;'>{user_message_html}</span>
+                <span id="message-content" style='color: white;'>{user_message_html}</span>
             </div><br>"""
 
         if assistant_message:
@@ -104,7 +104,7 @@ def render_message(history):
             messages_html += f"""
             <div style='display: flex; align-items: center; margin-bottom: 10px;'>
                 <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
-                <span style='color: white;'>{assistant_message_html}</span>
+                <span id="message-content" style='color: white;'>{assistant_message_html}</span>
             </div><br>"""
 
     messages_html += """
@@ -112,7 +112,7 @@ def render_message(history):
         <script src="https://cdnjs.cloudflare.com/ajax/libs/dompurify/2.3.3/purify.min.js"></script>
         <script>
             var cleanHTML = DOMPurify.sanitize(userInputHTML);
-            document.getElementById("output").innerHTML = cleanHTML;
+            document.getElementById("message-content").innerHTML = cleanHTML;
         </script>
         """ 
     return messages_html