POLRAMBORA commited on
Commit
2f14634
·
verified ·
1 Parent(s): e147ef5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -8
app.py CHANGED
@@ -122,21 +122,25 @@ def render_message(history):
122
  <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;">
123
  <div id="messages" style="display: block; margin-bottom: 10px;">"""
124
 
 
 
125
  for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
126
- if user_message:
 
127
  user_message_html = markdown.markdown(user_message, extensions=["fenced_code", "codehilite"])
128
  messages_html += f"""
129
- <div style='display: flex; align-items: center; margin-bottom: 10px;'>
130
- <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
131
- <span style='color: white; white-space: pre-wrap;'>{user_message_html}</span>
132
  </div>"""
133
 
134
- if assistant_message:
 
135
  assistant_message_html = markdown.markdown(assistant_message, extensions=["fenced_code", "codehilite"])
136
  messages_html += f"""
137
- <div style='display: flex; align-items: center; margin-bottom: 10px;'>
138
- <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
139
- <span style='color: white; white-space: pre-wrap;'>{assistant_message_html}</span>
140
  </div>"""
141
 
142
  messages_html += "</div></div>"
 
122
  <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;">
123
  <div id="messages" style="display: block; margin-bottom: 10px;">"""
124
 
125
+ seen_messages = set() # Track (role, message) pairs to avoid duplicates
126
+
127
  for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
128
+ if user_message and ("user", user_message) not in seen_messages:
129
+ seen_messages.add(("user", user_message))
130
  user_message_html = markdown.markdown(user_message, extensions=["fenced_code", "codehilite"])
131
  messages_html += f"""
132
+ <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
133
+ <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
134
+ <div style='color: white; white-space: pre-wrap;'>{user_message_html}</div>
135
  </div>"""
136
 
137
+ if assistant_message and ("assistant", assistant_message) not in seen_messages:
138
+ seen_messages.add(("assistant", assistant_message))
139
  assistant_message_html = markdown.markdown(assistant_message, extensions=["fenced_code", "codehilite"])
140
  messages_html += f"""
141
+ <div style='display: flex; flex-direction: column; align-items: flex-start; margin-bottom: 10px;'>
142
+ <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-bottom: 5px;'>
143
+ <div style='color: white; white-space: pre-wrap;'>{assistant_message_html}</div>
144
  </div>"""
145
 
146
  messages_html += "</div></div>"