pvanand's picture
Upload 2 files
0400f01
raw
history blame
1.35 kB
import streamlit as st
import requests
import json
st.title("Rasa Chatbot Interface")
# Initialize or get the chat history
if 'chat_history' not in st.session_state:
st.session_state.chat_history = []
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('http://localhost:5005/webhooks/rest/webhook', json=payload)
bot_reply = response.json()
# 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))
# 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};