File size: 7,527 Bytes
52809d4 fa0a2c5 52809d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 | #import the necessary libraries
import streamlit as st
import pandas as pd
import plotly.express as px
from models import generate_response
from functions import authenticate_user, register_user, save_emotions_to_supabase, get_user_emotions
#set the cover image
st.image('portada.webp', use_container_width=True)
#set the title
st.title('🤖 :blue[EmotiBot]')
st.subheader(':blue[Therapeutic Chatbot with Emotion Analysis]')
#customize the appearance of the application
st.markdown(
f"""
<style>
.stApp {{
background-color: #0ee3bf;
}}
.block-container {{
max-width: 1200px;
margin: auto;
}}
.chatbot-response {{
background-color: #b3d9ff;
padding: 10px;
margin-top: 10px;
border-radius: 5px;
}}
div.stButton > button {{
background-color: #0c99eb;
color: white;
border-radius: 5px;
border: none;
padding: 8px 16px;
font-size: 16px;
}}
div.stButton > button:hover {{
background-color: #075be0;
}}
textarea {{
background-color: #b3d9ff !important;
color: black !important;
}}
</style>
""",
unsafe_allow_html=True
)
#manage the session state in Streamlit using st.session_state.
if "authenticated" not in st.session_state:
st.session_state.authenticated = False
if "username" not in st.session_state:
st.session_state.username = ""
if "user_id" not in st.session_state:
st.session_state.user_id = None
if "show_bar_chart" not in st.session_state:
st.session_state.show_bar_chart = False
if "show_line_chart" not in st.session_state:
st.session_state.show_line_chart = False
#login or register
if not st.session_state.authenticated:
tab_login, tab_register = st.tabs(["Login", "Register"])
with tab_login:
st.subheader("Login")
username = st.text_input("Username")
password = st.text_input("Password", type="password")
if st.button("Log in"):
user_id = authenticate_user(username, password)
if user_id is None:
st.error("Invalid username or password")
else:
st.session_state.authenticated = True
st.session_state.user_id = user_id
st.session_state.username = username
st.success(f"Login successful, {username}!")
st.rerun()
with tab_register:
st.subheader("Register")
new_email = st.text_input("Email")
new_password = st.text_input("New password", type="password")
if st.button("Sign up"):
user_id, result = register_user(new_email, new_password)
if user_id:
st.success(result)
else:
st.error(result)
#if the user is authenticated, display the chat interface and graphics
if st.session_state.authenticated:
st.markdown(f"Welcome, {st.session_state.username}")
#distribute the chat and graphics in two columns
col1, col2 = st.columns([3, 3], gap="large")
with col1:
user_input = st.text_area("Tell me how you feel:", key="user_input")
if st.button("Send Message"):
if user_input.strip():
response, detected_emotions = generate_response(user_input)
st.session_state.last_response = response
st.session_state.last_emotion = ", ".join([emotion[0] for emotion in detected_emotions])
save_emotions_to_supabase(st.session_state.user_id, detected_emotions)
st.rerun()
if "last_response" in st.session_state:
st.markdown(f'<div class="chatbot-response">{st.session_state.last_response}</div>', unsafe_allow_html=True)
st.write(f"**Detected emotions:** {st.session_state.last_emotion}")
with col2:
user_emotions = get_user_emotions(st.session_state.user_id)
if user_emotions.empty:
st.warning("No data available. Try sending a message first.")
else:
if st.button("Show Emotion Frequency"):
st.session_state.show_bar_chart = True
st.session_state.show_line_chart = False
if st.button("Show Emotion Evolution"):
st.session_state.show_line_chart = True
st.session_state.show_bar_chart = False
emotion_colors = {
"sadness": "#0761e8",
"fear": "#000000",
"joy": "#FFA500",
"surprise": "#8A2BE2",
"disgust": "#228B22",
"neutral": "#D3D3D3",
"anger": "#FF0000"
}
#show bar chart if activated
if st.session_state.show_bar_chart:
emotion_counts = user_emotions["emotion"].explode().value_counts().reset_index()
emotion_counts.columns = ["Emotion", "Frequency"]
fig_bar = px.bar(
emotion_counts,
x="Emotion",
y="Frequency",
color="Emotion",
color_discrete_map=emotion_colors
)
fig_bar.update_layout(
paper_bgcolor='rgba(0,0,0,0)',
plot_bgcolor='rgba(0,0,0,0)',
showlegend=False,
title_text="",
xaxis=dict(showgrid=False, tickfont=dict(color="black"), title_text=""),
yaxis=dict(showgrid=False, tickfont=dict(color="black"), title_text="")
)
st.plotly_chart(fig_bar, use_container_width=True)
#show line bar if activated
if st.session_state.show_line_chart:
emotion_timeline = user_emotions.explode("emotion")
emotion_timeline["week"] = emotion_timeline["timestamp"].dt.to_period("W").astype(str)
unique_weeks = sorted(emotion_timeline["week"].unique())
week_labels = {week: f"Week {i+1}" for i, week in enumerate(unique_weeks)}
emotion_timeline["week"] = emotion_timeline["week"].map(week_labels)
emotion_weekly = emotion_timeline.groupby(["week", "emotion"]).size().reset_index(name="count")
fig_line = px.line(
emotion_weekly,
x="week",
y="count",
color="emotion",
markers=True,
color_discrete_map=emotion_colors
)
fig_line.update_layout(
paper_bgcolor='rgba(0,0,0,0)',
plot_bgcolor='rgba(0,0,0,0)',
showlegend=True,
title_text="",
legend_title=dict(text="Emotion", font=dict(color="black")),
xaxis=dict(
showgrid=False,
tickfont=dict(color="black"),
title_text=""
),
yaxis=dict(
showgrid=False,
tickfont=dict(color="black"),
title_text=""
)
)
st.plotly_chart(fig_line, use_container_width=True)
|