Update app.py

app.py

@@ -7,6 +7,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification, pipeline
 import os
 import plotly.graph_objects as go
+import colorsys
 
 # Utility functions for color conversions and brightness adjustment
 def hex_to_rgb(hex_color: str) -> tuple[int, int, int]:
@@ -86,9 +87,7 @@ def process_classification(text, model1, model2, tokenizer1) -> Tuple[float, flo
 def generate_pie_chart(values: list, labels: list, title: str):
     fig = go.Figure(data=[go.Pie(labels=labels, values=values, hole=.3)])
     fig.update_layout(title_text=title)
-    file_path = "/tmp/pie_chart.html"
-    fig.write_html(file_path)
-    return file_path
+    return fig.to_html(full_html=False)
 
 @spaces.GPU
 def all(text):
@@ -97,12 +96,12 @@ def all(text):
     int_count, ext_count, int_ext_ratio = process_classification(text, model1, model2, tokenizer1)
     
     # Create pie charts
-    pie_chart_path_int_ext = generate_pie_chart([int_count, ext_count], ['Internal', 'External'], "Internal vs External Details")
+    pie_chart_html_int_ext = generate_pie_chart([int_count, ext_count], ['Internal', 'External'], "Internal vs External Details")
     subclass_labels = [entity['entity'] for entity in ner_ext['entities']]
     subclass_values = [1] * len(subclass_labels)  # Each entity is counted once; adjust as needed for actual counts
-    pie_chart_path_subclass = generate_pie_chart(subclass_values, subclass_labels, "Detail Subclasses")
+    pie_chart_html_subclass = generate_pie_chart(subclass_values, subclass_labels, "Detail Subclasses")
 
-    return ner_bin, ner_ext, f"{round(int_count, 1)}", f"{round(ext_count, 1)}", f"{round(int_ext_ratio, 2)}", pie_chart_path_int_ext, pie_chart_path_subclass
+    return ner_bin, ner_ext, f"{round(int_count, 1)}", f"{round(ext_count, 1)}", f"{round(int_ext_ratio, 2)}", pie_chart_html_int_ext, pie_chart_html_subclass
 
 examples = [
     ['Bevor ich meinen Hund kaufte bin ich immer alleine durch den Park gelaufen. Gestern war ich aber mit dem Hund losgelaufen. Das Wetter war sehr schön, nicht wie sonst im Winter. Ich weiß nicht genau. Mir fällt sonst nichts dazu ein. Wir trafen auf mehrere Spaziergänger. Ein Mann mit seinem Kind. Das Kind hat ein Eis gegessen.'],
@@ -130,4 +129,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch(debug=True)
+iface.launch(debug=True)