Update app.py

app.py

@@ -277,6 +277,7 @@ def button_pressed(text_box):
 
     tweets = 0
 
+    user_data = {}
     for user in tweet_data:
         data = tweet_data[user]
 
@@ -300,6 +301,7 @@ def button_pressed(text_box):
                 total_types_count[type] = total_types_count[type] + 1
 
             total_average_toxicity.append(data["average_toxicity"])
+            user_data[user.name] = data["average_toxicity"]
             total_average_severe_toxicity.append(data["average_severe_toxicity"])
             total_average_obscene.append(data["average_obscene"])
             total_average_identity_attack.append(data["average_identity_attack"])
@@ -349,6 +351,22 @@ def button_pressed(text_box):
 
     sentiment_plot_fig = go.Figure(sentiment_plot)
 
+    # User distrabution plot
+    user_plot = dict({
+        "data": [{"type": "bar",
+                  "x": list(user_data.keys()),
+                  "y": list(user_data.values())}],
+        "layout": {"title": {"text": "Hate Speech By Observed User"}}
+    })
+
+    user_plot_fig = go.Figure(user_plot)
+
+    # Distrabution Pie
+    labels = ['Timeline Users Observed', 'Observed Users Mentioning @{}'.format(text_box)]
+    values = [following,mentions]
+
+    distrabution_fig = go.Figure(data=[go.Pie(labels=labels, values=values, title="Distribution Of Observed Users")])
+
     # Comprise elements for 'type' plot
     colours = []
     keys = list(total_types_count.keys())
@@ -374,10 +392,10 @@ def button_pressed(text_box):
 
     # Comprise text for summary label
     original_text = "A total number of {} recent tweets in @{}'s mentions and timeline were reviewed, of which @{} was exposed to {} users via mentions and " \
-           "{} directly via following them.".format(tweets, text_box, text_box, mentions, following)
+                    "{} directly via following them.".format(tweets, text_box, text_box, mentions, following)
 
     text = original_text
-    
+
     high_identifiers = []
     extreme_identifiers = []
 
@@ -430,15 +448,16 @@ def button_pressed(text_box):
 
         text = text[:len(text - 1)] + " language."
 
-    if total_average_neg_sentiment > 0.7  and total_average_neg_sentiment > total_average_pos_sentiment:
+    if total_average_neg_sentiment > 0.7 and total_average_neg_sentiment > total_average_pos_sentiment:
         text = text + " @{} is experiencing a high amount of negative sentiment content.".format(text_box)
-    elif total_average_neg_sentiment > 0.9  and total_average_neg_sentiment > total_average_pos_sentiment:
+    elif total_average_neg_sentiment > 0.9 and total_average_neg_sentiment > total_average_pos_sentiment:
         text = text + " '{} is experiencing a significantly high amount of negative sentiment content.".format(text_box)
 
     if len(text) == len(original_text):
-        text = text + " No excessive hate speech or low sentiment was observed in @{}'s mentions or timeline.".format(text_box)
-    
-    return [toxicity_plot_fig, sentiment_plot_fig, fig, text]
+        text = text + " No excessive hate speech or low sentiment was observed in @{}'s mentions or timeline.".format(
+            text_box)
+
+    return [toxicity_plot_fig, sentiment_plot_fig, fig, text,user_plot_fig,distrabution_fig]
 
 
 # The main chunk of code that uses Gradio blocks to create the UI
@@ -465,7 +484,11 @@ with block:
                     sentiment_plot = gr.Plot(label="Sentiment Graph")
                 gr.HTML(value="<br>")
                 type_plot = gr.Plot(label="Content Type Graph")
-    btn.click(fn=button_pressed, inputs=[text_input], outputs=[toxicity_plot, sentiment_plot, type_plot, output_label])
+                gr.HTML(value="<br>")
+                with gr.Row().style(equal_height=True):
+                    user_plot = gr.Plot(label="Observed Users")
+                    format_type_plot = gr.Plot(label="Distribution")
+    btn.click(fn=button_pressed, inputs=[text_input], outputs=[toxicity_plot, sentiment_plot, type_plot, output_label,user_plot,format_type_plot])
     gr.Markdown(
         """___
    <p style='text-align: center'>
@@ -477,4 +500,4 @@ with block:
 # block.attach_load_events()
 
 # Launcg the page
-block.launch(enable_queue=True)
+block.launch(enable_queue=True,show_api=False)