Update app.py

app.py

@@ -43,16 +43,20 @@ topics, probs = fit_transform(topic_model, tiktok)
 placeholder = st.empty()
 text_input = placeholder.text_area("Enter product topic here", height=300)
 
-similar_topics, similarity = topic_model.find_topics("motor", top_n=20)
-
-most_similar = similar_topics[0]
-print(similar_topics[0])
-print("Most Similar Topic Info: \n{}".format(topic_model.get_topic(most_similar)))
-print("Similarity Score: {}".format(similarity[0]))
-
-answer_as_string = topic_model.get_topic(most_similar)
-
-st.text_area("Most Similar Topic List is Here",answer_as_string,key="topic_list")
-st.image('https://freepngimg.com/download/keyboard/6-2-keyboard-png-file.png',use_column_width=True)
-st.markdown("<h6 style='text-align: center; color: #808080;'>Created By LiHE</a></h6>", unsafe_allow_html=True)
+top_n = st.sidebar.slider("Select a number of keywords", 1, 10, 5, 1)
+
+similar_topics, similarity = topic_model.find_topics(text_input, top_n=top_n)
+
+if similar_topics != []:
+    most_similar = similar_topics[0]
+    print(similar_topics[0])
+    print("Most Similar Topic Info: \n{}".format(topic_model.get_topic(most_similar)))
+    print("Similarity Score: {}".format(similarity[0]))
+
+    answer_as_string = topic_model.get_topic(most_similar)
+    
+    st.info("Extracted Topic")
+    st.text_area("Most Similar Topic List is Here",answer_as_string,key="topic_list")
+    st.image('https://freepngimg.com/download/keyboard/6-2-keyboard-png-file.png',use_column_width=True)
+    st.markdown("<h6 style='text-align: center; color: #808080;'>Created By LiHE</a></h6>", unsafe_allow_html=True)