span highlight for web search

handler.py

@@ -54,13 +54,22 @@ class EndpointHandler():
         else:
             assert len(data['inputs']['claims']) == 1, "Only one claim is allowed for web retrieval for the current version."
 
-            
             ranked_docs, scores, ranked_urls = self.search_relevant_docs(claim)
 
+            span_to_highlight = []
+            for doc_chunk, score in zip(ranked_docs, scores):
+                # If the chunk can support the claim, find the sentence with the highest rouge score
+                if score > 0.5:
+                    highest_score_sent, _ = self.chunk_and_highest_rouge_score(doc_chunk, claim)
+                    span_to_highlight.append(highest_score_sent)
+                else:
+                    span_to_highlight.append("")
+
             outputs = {
                 'ranked_docs': ranked_docs,
                 'scores': scores,
-                'ranked_urls': ranked_urls
+                'ranked_urls': ranked_urls,
+                'span_to_highlight': span_to_highlight
             }
             
         return outputs