Update app.py

app.py

@@ -162,18 +162,10 @@ 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)
-    
-          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), history, "", message
-
+           history, assistant_reply = respond(message, api_key, max_tokens, top_p, temperature)
+           history.append((message, assistant_reply, "You", "P-ALPLE", USER_PIC_PATH, ASSISTANT_PIC_PATH))
+           full_history_html = render_message(history)
+           return full_history_html, history, "", message
 
         def regenerate_response(history, last_message, max_tokens, top_p, temperature):
             return "", []
@@ -189,12 +181,12 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
             return render_message(history), history
 
         msg_input.submit(user_interaction, 
-                         inputs=[msg_input, history_state, api_key_input, max_tokens, top_p, temperature], 
-                         outputs=[chatbot_output, history_state, msg_input, last_message_state])
+                 inputs=[msg_input, history_state, api_key_input, max_tokens, top_p, temperature], 
+                 outputs=[chatbot_output, history_state, msg_input, last_message_state])
 
         send_btn.click(user_interaction, 
-                       inputs=[msg_input, history_state, api_key_input, max_tokens, top_p, temperature], 
-                       outputs=[chatbot_output, history_state, msg_input, last_message_state])
+                inputs=[msg_input, history_state, api_key_input, max_tokens, top_p, temperature], 
+                outputs=[chatbot_output, history_state, msg_input, last_message_state])
 
         regen_btn.click(clear_history, 
                         inputs=[api_key_input],