Update app.py

app.py

@@ -136,26 +136,29 @@ def render_message(history):
     <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;">
         <div id="messages" style="display: block; margin-bottom: 10px;">"""
 
-    for user_message, assistant_message, user_pic, assistant_pic in history:
-        if user_message:
-            sanitized_message = bleach.clean(user_message, tags=allowed_tags, attributes=allowed_attributes, strip=True)
-            rendered_message = markdown.markdown(sanitized_message)
-
-            messages_html += f"""
-            <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
-                <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
-                <div style='color: white; white-space: pre-wrap;'>{rendered_message}</div>
-            </div>"""
-
-        if assistant_message:
-            sanitized_message = bleach.clean(assistant_message, tags=allowed_tags, attributes=allowed_attributes, strip=True)
-            rendered_message = markdown.markdown(sanitized_message)
-
-            messages_html += f"""
-            <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
-                <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
-                <div style='color: white; white-space: pre-wrap;'>{rendered_message}</div>
-            </div>"""
+    for message_tuple in history:
+        if len(message_tuple) >= 4:
+            user_message, assistant_message, user_pic, assistant_pic = message_tuple[:4]  # Safely unpack first 4 items
+
+            if user_message:
+                sanitized_message = bleach.clean(user_message, tags=allowed_tags, attributes=allowed_attributes, strip=True)
+                rendered_message = markdown.markdown(sanitized_message)
+                messages_html += f"""
+                <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
+                    <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
+                    <div style='color: white; white-space: pre-wrap;'>{rendered_message}</div>
+                </div>"""
+
+            if assistant_message:
+                sanitized_message = bleach.clean(assistant_message, tags=allowed_tags, attributes=allowed_attributes, strip=True)
+                rendered_message = markdown.markdown(sanitized_message)
+                messages_html += f"""
+                <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
+                    <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
+                    <div style='color: white; white-space: pre-wrap;'>{rendered_message}</div>
+                </div>"""
+        else:
+            print(f"Skipping invalid entry in history: {message_tuple}")
 
     messages_html += "</div></div>"
     return messages_html