Update highlighter.py

highlighter.py

@@ -6,11 +6,13 @@ from predictors import predict_proba_quillbot
 def explainer(text):
     class_names = ['negative', 'positive']
     explainer = LimeTextExplainer(class_names=class_names, split_expression=sent_tokenize)
-    exp = explainer.explain_instance(text, predict_proba_quillbot, num_features=20, num_samples=300)
-    sentences = [t[0] for t in exp.as_list()]
-    attributions = [t[1] for t in exp.as_list()]
-    l, weights = zip(*exp.local_exp[exp.available_labels()[0]])
-    sentences_weights = {sentences[i]: attributions[i] for i in l}
+    exp = explainer.explain_instance(text, predictor, num_features=20, num_samples=300)
+    sentences = [sent for sent in sent_tokenize(text)]
+    weights_mapping = exp.as_map()[1]
+    sentences_weights = {sentence: 0 for sentence in sentences}
+    for idx, weight in weights_mapping:
+        if 0 <= idx < len(sentences):
+            sentences_weights[sentences[idx]] = weight
     print(sentences_weights)
     return sentences_weights
 
@@ -32,7 +34,7 @@ def analyze_and_highlight(text):
         if not sentence:
             continue
     
-        highlighted_sentence = f'<span style="background-color: {color}; color: black;">{sentence}.</span> '
+        highlighted_sentence = f'<span style="background-color: {color}; color: black;">{sentence}</span> '
         highlighted_text += highlighted_sentence
     
     return highlighted_text