Update app.py

app.py

@@ -24,6 +24,11 @@ import plotly.graph_objects as go
 from wordcloud import WordCloud
 from PIL import Image
 
+bin_color_map = {
+    "External": "#00008B",
+    "Internal": "#8B0000"
+}
+
 def hex_to_rgb(hex_color: str) -> tuple[int, int, int]:
     hex_color = hex_color.lstrip('#')
     return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
@@ -85,11 +90,6 @@ def generate_charts(ner_output_bin: dict) -> Tuple[go.Figure, np.ndarray]:
     bin_labels = list(entity_counts_bin.keys())
     bin_sizes = list(entity_counts_bin.values())
 
-    bin_color_map = {
-      "External": "#00008B",
-      "Internal": "#8B0000"
-    }
-
     bin_colors = [bin_color_map.get(label, "#FFFFFF") for label in bin_labels]
     
     # Create bar chart for binary classification
@@ -182,10 +182,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="Binary Sequence Classification",
-                           bin_color_map = {
-                              "External": "#00008B",
-                              "Internal": "#8B0000"
-                            }
+                           color_map = bin_color_map
                           ),
         gr.Plot(label="Binary SeqClass Entity Count Bar Chart"),
         gr.Image(label="Entity Word Cloud")