Update app.py

app.py

@@ -117,7 +117,7 @@ prompt_placeholder = st.form("chat-form")
 credit_card_placeholder = st.empty()
 
 with chat_placeholder:
-    for chat in st.session_state.history:
+    for chat in st.session_state.history[:-1]:
         div = f"""
 
 <div class="chat-row 
@@ -133,6 +133,25 @@ with chat_placeholder:
 </div>
         """
         st.markdown(div, unsafe_allow_html=True)
+    last_chat = st.session_state.history[-1]
+div_start = f"""
+<div class="chat-row 
+    {'' if last_chat.origin == 'ai' else 'row-reverse'}">
+    <img class="chat-icon" src="https://cdn-icons-png.flaticon.com/{
+        '/512/3058/3058838.png' if last_chat.origin == 'ai' 
+                      else '512/1177/1177568.png'}"
+         width=32 height=32>
+    <div class="chat-bubble
+    {'ai-bubble' if last_chat.origin == 'ai' else 'human-bubble'}">
+        &#8203;"""
+    div_end = """
+    </div>
+</div>
+    """
+    new_placeholder = st.empty()
+    for j in range(len(last_chat.message)):
+        new_placeholder.markdown(div_start + last_chat.message[:j+1] + div_end, unsafe_allow_html=True)
+        time.sleep(0.05)
     
     for _ in range(3):
         st.markdown("")