| import streamlit as st | |
| import matplotlib | |
| import matplotlib.pyplot as plt | |
| import plotly.graph_objects as go | |
| import plotly.express as px | |
| def indicator_plot(value, title, value_range, domain): | |
| plot = go.Indicator( | |
| mode = 'gauge+delta', | |
| value = value, | |
| domain = domain, | |
| title = title, | |
| delta = { | |
| 'reference': 0, | |
| 'decreasing': {'color': '#ec4899'}, | |
| 'increasing': {'color': '#36def1'} | |
| }, | |
| gauge = { | |
| 'axis': {'range': value_range, 'tickwidth': 1, 'tickcolor': 'black'}, | |
| 'bar': {'color': '#4361ee'}, | |
| 'bgcolor': 'white', | |
| 'borderwidth': 2, | |
| 'bordercolor': '#efefef', | |
| 'steps': [ | |
| {'range': [value_range[0], 0], 'color': '#efefef'}, | |
| {'range': [0, value_range[1]], 'color': '#efefef'} | |
| ], | |
| 'threshold': { | |
| 'line': {'color': '#4361ee', 'width': 8}, | |
| 'thickness': 0.75, | |
| 'value': value | |
| } | |
| } | |
| ) | |
| return plot | |
| def scatter_plot(df, group_var): | |
| colors = ['#36def1', '#4361ee'] if group_var else ['#4361ee'] | |
| plot = px.scatter( | |
| df, | |
| x='Machine-ratings', | |
| y='Human-ratings', | |
| color=group_var, | |
| facet_col='x_group', | |
| facet_col_wrap=2, | |
| trendline='ols', | |
| trendline_scope='trace', | |
| hover_data={ | |
| 'Text': df.text, | |
| 'Language': False, | |
| 'x_group': False, | |
| 'Human-ratings': ':.2f', | |
| 'Machine-ratings': ':.2f', | |
| 'Study': df.study, | |
| 'Instrument': df.instrument, | |
| }, | |
| width=400, | |
| height=400, | |
| color_discrete_sequence=colors | |
| ) | |
| plot.for_each_annotation(lambda a: a.update(text=a.text.split('=')[-1])) | |
| plot.update_layout( | |
| legend={ | |
| 'orientation':'h', | |
| 'yanchor': 'bottom', | |
| 'y': -.30 | |
| }) | |
| plot.update_xaxes(title_standoff = 0) | |
| return plot | |
| def show_scores(sentiment, desirability, input_text): | |
| with st.container(): | |
| num_steps = 10 | |
| colorscale = plt.get_cmap('viridis', num_steps) | |
| steps_sentiment = [{ | |
| 'range': [i/num_steps*2-1, (i+1)/num_steps*2-1], | |
| 'color': matplotlib.colors.rgb2hex(colorscale(i)[:3]) | |
| } for i in range(num_steps)] | |
| steps_desirability = [{ | |
| 'range': [i/num_steps*8-4, (i+1)/num_steps*8-4], | |
| 'color': matplotlib.colors.rgb2hex(colorscale(i)[:3]) | |
| } for i in range(num_steps)] | |
| plot1 = go.Indicator( | |
| mode = 'number+gauge', | |
| value = sentiment, | |
| domain = {'x': [0.25, 1], 'y': [0.45, 0.65]}, | |
| title = {'text': 'Sentiment', 'font': {'color': 'black', 'size': 22}}, | |
| number={'font': {'color': 'black', 'size': 26}, 'valueformat': '.2f'}, | |
| gauge = { | |
| 'shape': 'bullet', | |
| 'axis': {'range': [-1, 1]}, | |
| 'threshold': { | |
| 'line': {'color': '#36def1', 'width': 10}, | |
| 'thickness': 0.75, | |
| 'value': sentiment}, | |
| 'steps': steps_sentiment, | |
| 'bar': {'color': 'rgba(0,0,0,0)'} | |
| } | |
| ) | |
| plot2 = go.Indicator( | |
| mode = 'number+gauge', | |
| value = desirability, | |
| domain = {'x': [0.25, 1], 'y': [0.15, 0.35]}, | |
| title = {'text': 'Desirability', 'font': {'color': 'black', 'size': 22}}, | |
| number={'font': {'color': 'black', 'size': 26}, 'valueformat': '.2f'}, | |
| gauge = { | |
| 'shape': 'bullet', | |
| 'axis': {'range': [-4, 4]}, | |
| 'threshold': { | |
| 'line': {'color': '#36def1', 'width': 10}, | |
| 'thickness': 0.75, | |
| 'value': desirability}, | |
| 'steps': steps_desirability, | |
| 'bar': {'color': 'rgba(0,0,0,0)'} | |
| } | |
| ) | |
| fig = go.Figure() | |
| fig.add_trace(plot1) | |
| fig.add_trace(plot2) | |
| plot_title = f'Estimated Sentiment and Desirability for <br><i>"{input_text}</i>"' | |
| fig.update_layout( | |
| annotations=[ | |
| go.layout.Annotation( | |
| text=plot_title, | |
| align='center', | |
| showarrow=False, | |
| xref='paper', | |
| yref='paper', | |
| x=0.5, | |
| y=.85, | |
| xanchor='center', | |
| yanchor='bottom', | |
| font=dict(size=22) | |
| ), | |
| go.layout.Annotation( | |
| text="Negative", | |
| showarrow=False, | |
| xref='paper', | |
| yref='paper', | |
| x=0.25, | |
| y=0.8, | |
| font=dict(size=18) | |
| ), | |
| go.layout.Annotation( | |
| text="Positive", | |
| showarrow=False, | |
| xref='paper', | |
| yref='paper', | |
| x=.81, | |
| y=.8, | |
| font=dict(size=18) | |
| ), | |
| ], | |
| font = {'color': 'black', 'family': 'Arial'}, | |
| height=300, | |
| margin={'t': 50, 'b': 0, 'l': 0} | |
| ) | |
| st.plotly_chart(fig, theme=None, use_container_width=True) | |
| st.markdown(""" | |
| Item sentiment: Absolute differences between positive and negative sentiment. | |
| Item desirability: z-transformed values, 0 indicated "neutral". | |
| """) |