Update app.py

app.py

@@ -24,6 +24,17 @@ import plotly.graph_objects as go
 from wordcloud import WordCloud
 from PIL import Image
 
+ext_color_map = {
+        "INTemothou": "#FF7F50",
+        "INTpercept": "#FF4500",
+        "INTtime": "#FF6347",
+        "INTplace": "#FFD700",
+        "INTevent": "#FFA500",
+        "EXTsemantic": "#4682B4",
+        "EXTrepetition": "#5F9EA0",
+        "EXTother": "#00CED1",
+    }
+
 
 def hex_to_rgb(hex_color: str) -> tuple[int, int, int]:
     hex_color = hex_color.lstrip('#')
@@ -79,16 +90,6 @@ def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
     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())
-    ext_color_map = {
-        "INTemothou": "#FF7F50",
-        "INTpercept": "#FF4500",
-        "INTtime": "#FF6347",
-        "INTplace": "#FFD700",
-        "INTevent": "#FFA500",
-        "EXTsemantic": "#4682B4",
-        "EXTrepetition": "#5F9EA0",
-        "EXTother": "#00CED1",
-    }
     ext_colors = [ext_color_map.get(label, "#FFFFFF") for label in ext_labels]
     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(
@@ -173,16 +174,7 @@ iface = gr.Interface(
     inputs=gr.Textbox(lines=5, label="Input Text", placeholder="Write about how your breakfast went or anything else that happened or might happen to you ..."),
     outputs=[
         gr.HighlightedText(label="Extended Sequence Classification",
-                           ext_color_map = {
-                              "INTemothou": "#FF7F50",
-                              "INTpercept": "#FF4500",
-                              "INTtime": "#FF6347",
-                              "INTplace": "#FFD700",
-                              "INTevent": "#FFA500",
-                              "EXTsemantic": "#4682B4",
-                              "EXTrepetition": "#5F9EA0",
-                              "EXTother": "#00CED1",
-                            }
+                           ext_color_map = ext_color_map
                           ),
         gr.Plot(label="Extended SeqClass Entity Distribution Pie Chart"),
         gr.Image(label="Entity Word Cloud")