Update app.py

app.py

@@ -103,17 +103,38 @@ def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figu
     entity_counts_ext = {entity: entities_ext.count(entity) for entity in set(entities_ext)}
     ext_labels = list(entity_counts_ext.keys())
     ext_sizes = list(entity_counts_ext.values())
+
+    # Define color mapping
+    bin_color_map = {
+        "External": "#6ad5bcff",
+        "Internal": "#ee8bacff"
+    }
+
+    ext_color_map = {
+        "INTemothou": "#FF7F50",  # Coral
+        "INTpercept": "#FF4500",  # OrangeRed
+        "INTtime": "#FF6347",     # Tomato
+        "INTplace": "#FFD700",    # Gold
+        "INTevent": "#FFA500",    # Orange
+        "EXTsemantic": "#4682B4", # SteelBlue
+        "EXTrepetition": "#5F9EA0", # CadetBlue
+        "EXTother": "#00CED1",    # DarkTurquoise
+    }
+
+    bin_colors = [bin_color_map[label] for label in bin_labels]
+    ext_colors = [ext_color_map[label] for label in ext_labels]
     
     # Create pie chart for extended classification
-    fig1 = go.Figure(data=[go.Pie(labels=ext_labels, values=ext_sizes, textinfo='label+percent', hole=.3)])
-    fig1.update_layout(title_text='Extended Sequence Classification Subclasses')
+    fig1 = go.Figure(data=[go.Pie(labels=ext_labels, values=ext_sizes, textinfo='label+percent', hole=.3, marker=dict(colors=ext_colors))])
+    fig1.update_layout(title_text='Extended Sequence Classification Subclasses', template='plotly_dark')
     
     # Create bar chart for binary classification
-    fig2 = go.Figure(data=[go.Bar(x=bin_labels, y=bin_sizes)])
+    fig2 = go.Figure(data=[go.Bar(x=bin_labels, y=bin_sizes, marker=dict(color=bin_colors))])
     fig2.update_layout(
         title='Binary Sequence Classification Classes',
         xaxis_title='Entity Type',
-        yaxis_title='Count'
+        yaxis_title='Count',
+        template='plotly_dark'
     )
     
     return fig1, fig2