Update app.py

app.py

@@ -52,14 +52,17 @@ if button and (uploaded_file is not None or input_text != ""):
 
     # Display top N most representative topics and their documents
     num_topics = st.sidebar.slider("Select number of topics to display", 1, 20, 5, 1)
-    topic_words, topic_docs = model.get_topics(with_documents=True)
-    for i, topic in enumerate(topic_words.items()):
-        if i >= num_topics:
-            break
-        st.write(f"## Topic {topic[0]}")
-        st.write("Keywords:", ", ".join(topic[1]))
+    topic_words = model.get_topics()
+    topic_freq = model.get_topic_freq().head(num_topics + 1)  # Add 1 to exclude -1 (outliers topic)
+    for _, row in topic_freq.iterrows():
+        topic_id = row["Topic"]
+        if topic_id == -1:
+            continue  # Skip the outliers topic
+        st.write(f"## Topic {topic_id}")
+        st.write("Keywords:", ", ".join(topic_words[topic_id]))
         st.write("Documents:")
-        for doc in topic_docs[topic[0]][:5]:
+        doc_ids = [idx for idx, topic in enumerate(topics) if topic == topic_id][:5]
+        for doc in doc_ids:
             st.write("-", texts[doc])
 
     # Display topic clusters