Update app.py

app.py

@@ -177,13 +177,14 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
             analyze_button = gr.Button("Analyze Tone", variant="primary")
             
         with gr.Column(scale=2):
+            # Example Tones Dropdown
             tone_dropdown = gr.Dropdown(
                 choices=sorted(df['label'].unique().tolist()),
                 label="Select a tone to view an example below."
             )
 
     gr.Markdown("<br>", elem_id="line-break-1")
-
+    
     with gr.Row():
         with gr.Column(scale=1):
             result_message = gr.Markdown()
@@ -191,22 +192,22 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
     with gr.Row():
         with gr.Column(scale=2):
             plot_output = gr.Plot(label="Tone Probabilities")
-
+        with gr.Column(scale=1):
+            all_probs_output = gr.JSON(label="All Probabilities")
+            
     with gr.Row():
         examples_output = gr.Dataframe(
             headers=["Examples of similar texts"],
             datatype=["str"],
             label="Example texts with similar tone"
         )
-
-    # Button callback
+    
     analyze_button.click(
         fn=analyze_tone,
         inputs=[text_input, tone_dropdown],  
-        outputs=[result_message, plot_output, examples_output]
+        outputs=[result_message, all_probs_output, plot_output, examples_output]
     )
     
-    # Dropdown change
     tone_dropdown.change(
         fn=get_tone_examples,
         inputs=tone_dropdown,