Merge pull request #41 from TheDataGuild/feature/parse-keywords-2

display top keywords in welcome message
TheDataGuild · Sep 28, 2023 · 0dcbaa9 · 0dcbaa9
2 parents 80a1a88 + 40b2a20
commit 0dcbaa9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/mind_palace/app.py b/mind_palace/app.py
@@ -30,10 +30,14 @@ def load_nodes_and_index(xml_dir, model):
 query_engine = CitationQueryEngine.from_args(index=vector_index, verbose=True)
 
 
-# TODO: pass in nodes instead of abstracts
 @st.cache_data(show_spinner="Summarizing papers... just a few more seconds.")
 def get_welcome_message(abstracts):
-    return welcome.summarize(gpt_model, abstracts)
+    return (
+        welcome.summarize(gpt_model, abstracts)
+        + "\n\n"
+        + "Top keywords: "
+        + ", ".join(welcome.extract_keywords(gpt_model, abstracts))
+    )
 
 
 if "messages" not in st.session_state.keys():  # Initialize the chat messages history