| |
| 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 |
|
|
| |
| st.image('portada.webp', use_container_width=True) |
|
|
| |
| st.title('🤖 :blue[EmotiBot]') |
| st.subheader(':blue[Therapeutic Chatbot with Emotion Analysis]') |
|
|
| |
| 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 |
| ) |
|
|
| |
| 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 |
|
|
| |
| 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 st.session_state.authenticated: |
| st.markdown(f"Welcome, {st.session_state.username}") |
|
|
| |
| 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" |
| } |
|
|
| |
| 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) |
|
|
| |
| 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) |
| |
