Update app.py

app.py

@@ -162,8 +162,18 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
         last_message_state = gr.State("")  
 
         def user_interaction(message, history, api_key, max_tokens, top_p, temperature):
-            history, assistant_reply = respond(message, api_key, max_tokens, top_p, temperature)
-            return render_message(history), history, "", message  
+          history, assistant_reply = respond(message, api_key, max_tokens, top_p, temperature)
+    
+          new_message_html = f"<div style='display: flex; align-items: center; margin-bottom: 10px;'>"
+          new_message_html += f"<img src='{USER_PIC_PATH}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>"
+          new_message_html += f"{message}</div><br>"
+
+          new_message_html += f"<div style='display: flex; align-items: center; margin-bottom: 10px;'>"
+          new_message_html += f"<img src='{ASSISTANT_PIC_PATH}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>"
+          new_message_html += f"{assistant_reply}</div><br>"
+
+          return gr.update(value=new_message_html, append=True), history, "", message
+
 
         def regenerate_response(history, last_message, max_tokens, top_p, temperature):
             return "", []