Update app.py

app.py

@@ -203,11 +203,11 @@ def dist_colors(output_file, input_checks):
     plot = (points + lines).properties(width=600, height=350).interactive()
     
     if "peaks" in input_checks:
-        return gr.update(value=plot, visible=True), gr.update(visible=True), gr.update(visible=False)  # next_button_peaks becomes available
+        return ggr.Plot.update(value=plot, visible=True), gr.update(visible=True), gr.update(visible=False)  # next_button_peaks becomes available
     elif "topics" in input_checks:
-        return gr.update(value=plot, visible=True), gr.update(visible=False), gr.update(visible=True)  # next_button_topics becomes available
+        return gr.Plot.update(value=plot, visible=True), gr.update(visible=False), gr.update(visible=True)  # next_button_topics becomes available
     else:
-        return gr.update(value=plot, visible=True), gr.update(visible=False), gr.update(visible=False)  # no next_button becomes available
+        return gr.Plot.update(value=plot, visible=True), gr.update(visible=False), gr.update(visible=False)  # no next_button becomes available
 
 
 def peaks(output_file, input_checks):
@@ -255,7 +255,7 @@ with gr.Blocks() as demo:
                 output_plot = gr.Plot(label="Emotion frequencies", visible=False).style(container=True)
                 next_button_dist = gr.Button("Show emotion distribution over time", visible=False)
                 
-                output_dist = gr.Plot(label="Emotion distribution over time", visible=False).style(container=True)
+                output_dist = gr.Plot(label="Emotion distribution over time", visible=False)
                 next_button_peaks = gr.Button("Show peaks", visible=False)
                 
                 output_peaks = gr.Textbox(label="Peaks", visible=False)
@@ -265,7 +265,7 @@ with gr.Blocks() as demo:
         
         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])
-        next_button_dist.click(fn=dist, inputs=[output_file,input_checks], outputs=[output_dist,next_button_peaks,next_button_topics])
+        next_button_dist.click(fn=dist_colors, inputs=[output_file,input_checks], outputs=[output_dist,next_button_peaks,next_button_topics])
         next_button_peaks.click(fn=peaks, inputs=[output_file,input_checks], outputs=[output_peaks,next_button_topics])
         next_button_topics.click(fn=topics, inputs=[output_file,input_checks], outputs=output_topics)