Update find_similar_news.py

find_similar_news.py

@@ -43,17 +43,16 @@ def get_milvus_collection():
     logger.warning('Exiting get_milvus_collection()')
     return collection
 
-def find_similar_news(text, search_vec, collection, vectorizer, sent_model, ce_model, top_n: int=5):
+def find_similar_news(text, search_vec, collection, vectorizer, sent_model, ce_model, top_n: int=10):
     logger.warning('Entering find_similar_news')
     search_params = {"metric_type": "IP"}
-    # search_vec = vectorizer.vectorize_(text)
     logger.warning('Querying Milvus for most similar results')
     results = collection.search([search_vec],
                                 anns_field='article_embed', # annotations field specified in the schema definition
                                 param=search_params,
                                 limit=top_n,
                                 guarantee_timestamp=1, 
-                                output_fields=['article_title', 'article_src', 'article_url', 'article_date'])[0] # which fields to return in output
+                                output_fields=['article_title', 'article_url'])[0] # which fields to return in output
     
     logger.warning('retrieved search results from Milvus')
     logger.warning('Computing cross encoder similarity scores')
@@ -64,11 +63,19 @@ def find_similar_news(text, search_vec, collection, vectorizer, sent_model, ce_m
 
     logger.warning('Generating HTML output')
     html_output = ""
+    article_count = 0
     for n, i in enumerate(similarity_idxs):
         title_ = results[i].entity.get('article_title')
         url_ = results[i].entity.get('article_url')
-        html_output += f'''<a style="font-weight: bold; font-size:14px; color: black;" href="{url_}" target="_blank">{title_}</a><br>
-        '''
+        if title_ != text:
+            html_output += f'''<a class="similar-news-item" href="{url_}" target="_blank">{title_}</a><br>
+            '''
+            article_count += 1
+
+        if article_count == 5 :
+            break
+
+            
     logger.warning('Successfully generated HTML output')
     logger.warning('Exiting find_similar_news')
     return html_output