Update app.py

app.py

@@ -5,11 +5,11 @@ from appStore.prep_utils import create_documents, get_client
 from appStore.embed import hybrid_embed_chunks
 from appStore.search import hybrid_search
 from appStore.region_utils import load_region_data, get_country_name
+from appStore.tfidf_extraction import extract_top_keywords 
 from torch import cuda
 import json
 from datetime import datetime
 
-
 # get the device to be used eithe gpu or cpu
 device = 'cuda' if cuda.is_available() else 'cpu'
 
@@ -133,6 +133,12 @@ if button:
             remainder_text = " ".join(words[preview_word_count:])
             # Always display the preview_text
             st.write(preview_text + ("..." if remainder_text else ""))
+            # ------ Extract top 5 keywords and display ------
+            top_keywords = extract_top_keywords(full_text, top_n=5)
+            # Join them with " · " and make them italic
+            if top_keywords:
+                st.write("")
+                st.markdown(f"_{' · '.join(top_keywords)}_")  # e.g. _keyword1 · keyword2 · keyword3_
             # ------- Additional info below the text -------
             metadata = res.payload.get('metadata', {})
             countries = metadata.get('countries', "[]")
@@ -178,6 +184,13 @@ if button:
             remainder_text = " ".join(words[preview_word_count:])
             # Always display the preview_text
             st.write(preview_text + ("..." if remainder_text else ""))
+            # ------ Extract top 5 keywords and display ------
+            top_keywords = extract_top_keywords(full_text, top_n=5)
+            # Join them with " · " and make them italic
+            if top_keywords:
+                st.write("")  # line break
+                st.markdown(f"_{' · '.join(top_keywords)}_")
+            
             # Additional text below the content
             metadata = res.payload.get('metadata', {})
             countries = metadata.get('countries', "[]")