Update app.py

app.py

@@ -1,37 +1,43 @@
-import streamlit as st
 import requests
-import json
+import streamlit as st
+import random
+import time
+
+st.title("Simple chat")
 
-st.title("Rasa Chatbot Interface")
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-# Initialize or get the chat history
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
-with st.form(key='chat_form'):
-    user_input = st.text_input("You:")
-    
-    # You can submit the form by pressing enter or by clicking the button
-    submitted = st.form_submit_button("Send")
-    
-    if submitted and user_input:
-        payload = {"sender": "user", "message": user_input}
-        response = requests.post('https://pvanand-rasa-moodbot.hf.space/webhooks/rest/webhook', json=payload)
-        bot_reply = response.json()
+# Accept user input
+if user_input := st.chat_input("What is up?"):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": user_input})
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(user_input)
 
-        # Append the user message and bot reply to the chat history
-        st.session_state.chat_history.append(("You:", user_input, True))
-        st.session_state.chat_history.append(("Bot:", bot_reply[0]['text'], False))
+    # Send user input to Rasa webhook
+    payload = {"sender": "user", "message": user_input}
+    response = requests.post('https://pvanand-rasa-moodbot.hf.space/webhooks/rest/webhook', json=payload)
+    bot_reply = response.json()
 
-# Display the chat history
-for message in st.session_state.chat_history:
-    sender, text, is_user = message
-    alignment = "right" if is_user else "left"
-    background_color = "#f5f5f5" # light background color
-    text_color = "#333333" # dark text color
-    
-    st.markdown(
-        f"<div style='text-align: {alignment};padding: 10px; margin: 5px; border-radius: 5px;'>{text}</div>",
-        unsafe_allow_html=True,
-    )
-# background-color: {background_color}; color: {text_color}; 
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        full_response = ""
+        assistant_response = random.choice(bot_reply)["text"]
+        # Simulate stream of response with milliseconds delay
+        for chunk in assistant_response.split():
+            full_response += chunk + " "
+            time.sleep(0.05)
+            # Add a blinking cursor to simulate typing
+            message_placeholder.markdown(full_response + "▌")
+        message_placeholder.markdown(full_response)
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": full_response})