Update app.py

app.py

@@ -299,7 +299,6 @@ def process_all_analyses(text, _analyzer):
     return segments, segment_labels, sentiment_scores, basic_emotions, moral_trajectories
 
             
-    return segments, segment_labels, sentiment_scores, basic_emotions, moral_trajectories
 def main():
     st.title("🗣️ Political Text Analysis Toolkit")
     
@@ -399,40 +398,34 @@ def main():
                         line=dict(color=colors[idx], width=2),
                         marker=dict(size=6)
                     ))
-            
             if "Basic Emotions Flow" in viz_options:
                 emotions_df = pd.DataFrame({
                     'Segment': segment_labels,
                     'Emotion': basic_emotions
                 })
+                
+                # Create color mapping for emotions
+                emotion_colors = {
+                    'joy': '#FFD700',      # Gold
+                    'sadness': '#4169E1',  # Royal Blue
+                    'anger': '#FF4500',    # Red Orange
+                    'fear': '#800080',     # Purple
+                    'disgust': '#006400',  # Dark Green
+                    'surprise': '#FFA500'  # Orange
+                }
+                
                 unified_fig.add_trace(go.Bar(
                     x=segment_labels,
-                    y=[1 if e == 'joy' else 0.5 if e == 'neutral' else -0.5 for e in basic_emotions],
-                    name=f'Emotions ({", ".join(set(basic_emotions))})',
-                    marker_color='#2ecc71',
-                    opacity=0.6,
-                    hovertemplate="Segment: %{x}<br>Emotion: " + 
-                                  "<br>".join([f"{e}" for e in basic_emotions]) +
-                                  "<extra></extra>"
+                    y=[1] * len(basic_emotions),  # Full height bars
+                    name='Basic Emotions',
+                    marker=dict(
+                        color=[emotion_colors.get(e.lower(), '#808080') for e in basic_emotions]
+                    ),
+                    opacity=0.8,
+                    hovertemplate="Segment: %{x}<br>Emotion: %{text}<extra></extra>",
+                    text=basic_emotions
                 ))
-            
-            unified_fig.update_layout(
-                title='Speech Analysis Flow',
-                xaxis_title='Speech Segments',
-                yaxis_title='Analysis Values',
-                plot_bgcolor='white',
-                height=600,
-                showlegend=True,
-                legend=dict(
-                    orientation="h",
-                    yanchor="bottom",
-                    y=1.02,
-                    xanchor="right",
-                    x=1
-                ),
-                hovermode='x unified'
-            )
-            
+
             st.plotly_chart(unified_fig, use_container_width=True)
 
         with tab3: