making it prettier

app.py

@@ -105,25 +105,39 @@ def plot_sunburst_chart(heatmap_data):
     for item in heatmap_data:
         sentence = item['sentence']
         emotions = item['emotions']
+
+        # Wrap long sentences for better display
+        sentence_wrapped = "\n".join([sentence[i:i + 50] for i in range(0, len(sentence), 50)])
+
         for i, score in enumerate(emotions):
             data.append({
-                'root': 'All Sentences',
-                'sentence': sentence,
+                'sentence': sentence_wrapped,
                 'emotion': id2label[i],
                 'score': float(score)
             })
 
     df = pd.DataFrame(data)
 
-    # Plot sunburst
+    # Create sunburst chart (no "root" level)
     fig = px.sunburst(
         df,
-        path=['root', 'sentence', 'emotion'],
+        path=['sentence', 'emotion'],
         values='score',
         color='emotion',
-        title='Sentence-level Emotion Confidences'
+        hover_data={'score': ':.3f'},
+        title='Sentence-Level Emotion Confidence Sunburst'
     )
 
+    # Make the chart bigger
+    fig.update_layout(
+        width=800,
+        height=800,
+        margin=dict(t=50, l=0, r=0, b=0)
+    )
+
+    # Display the chart in Streamlit
+    st.plotly_chart(fig, use_container_width=True)
+
     return fig
 
 def plot_emotion_barplot(heatmap_data):