replaced gauge by bullet plot

plots.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import matplotlib
+import matplotlib.pyplot as plt
 import plotly.graph_objects as go
 import plotly.express as px
 
@@ -74,29 +76,101 @@ def scatter_plot(df, group_var):
 
 def show_scores(sentiment, desirability, input_text):
     with st.container():
-        p1 = indicator_plot(
-            value=sentiment,
-            title=f'Item Sentiment',
-            value_range=[-1, 1],
-            domain={'x': [0, .45], 'y': [0, .5]},
-        )
 
-        p2 = indicator_plot(
-            value=desirability,
-            title=f'Item Desirability',
-            value_range=[-4, 4],
-            domain={'x': [.55, 1], 'y': [0, .5]}
+        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(p1)
-        fig.add_trace(p2)
-
-        fig.update_layout(
-            title=dict(text=f'"{input_text}"', font=dict(size=36),yref='paper'),
-            paper_bgcolor = 'white', 
-            font = {'color': 'black', 'family': 'Arial'})
-            
+        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("""