Update app.py

app.py

@@ -165,6 +165,24 @@ def file(input_file, input_checks):
     else:
         return gr.update(value=output, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)  # next_button_freq becomes available
 
+def freq(output_file, input_checks):
+    preds_dict = {"neutral": 0, "anger": 0, "fear": 0, "joy": 0, "love": 0, "sadness": 0}
+	for pred in preds:
+		preds_dict[pred] = preds_dict[pred] + 1
+	bars = list(preds_dict.keys())
+	height = list(preds_dict.values())
+
+	x_pos = np.arange(len(bars))
+	plt.bar(x_pos, height, color=['lightgrey', 'firebrick', 'rebeccapurple', 'orange', 'palevioletred', 'cornflowerblue'])
+	plt.xticks(x_pos, bars)
+    if "emotion distribution over time" in input_checks:
+        return gr.update(value=plt, visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)  # next_button_dist becomes available
+    elif "peaks" in input_checks:
+        return gr.update(value=plt, visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)  # next_button_peaks becomes available
+    elif "topics" in input_checks:
+        return gr.update(value=plt, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)  # next_button_peaks becomes available
+    else:
+        return gr.update(value=plt, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)  # next_button_freq becomes available
 
 
 with gr.Blocks() as demo:
@@ -211,6 +229,7 @@ with gr.Blocks() as demo:
         
         #send_btn.click(fn=inference_dataset1, inputs=input_file, outputs=[output_file,next_button_freq]) # next_button_freq becomes visible
         send_btn.click(fn=file, inputs=[input_file,input_checks], outputs=[output_file,next_button_freq,next_button_dist,next_button_peaks,next_button_topics])
+        next_button_freq.click(fn=freq, inputs=[output_file,input_checks], outputs=[output_plot,next_button_dist,next_button_peaks,next_button_topics])
 
         """"
         if "emotion frequencies" in input_checks: