Update app.py

app.py

@@ -21,29 +21,24 @@ def main():
     # Get user choice for number of results (slider)
     num_results = st.slider("Number of Results", min_value=1, max_value=30, value=5, step=1)
 
-    # Extract keywords
-    if st.button("Extract Keywords"):
-        keywords = kw_model.extract_keywords(doc, stop_words=None if remove_stopwords else "english")
-
-        if apply_mmr:
-            # Apply Maximal Marginal Relevance (MMR)
-            selected_keywords = []
-            selected_keywords.append(keywords[0])  # Select the top-scoring keyword
+    # Get user choice for minimum n-gram value (default textbox)
+    min_ngram = st.number_input("Minimum N-gram", value=1, min_value=1, max_value=10, step=1)
 
-            # Set the MMR hyperparameters
-            lambda_param = 0.7  # Weight for the trade-off between relevance and diversity
+    # Get user choice for maximum n-gram value (default textbox)
+    max_ngram = st.number_input("Maximum N-gram", value=3, min_value=1, max_value=10, step=1)
 
-            for i in range(1, num_results):
-                selected_keywords_scores = [kw[1] for kw in selected_keywords]
-                remaining_keywords = [kw for kw in keywords if kw[0] not in [kw[0] for kw in selected_keywords]]
-                mmr_scores = kw_model.maximal_marginal_relevance(doc, remaining_keywords, selected_keywords_scores, lambda_param)
-                max_mmr_index = mmr_scores.index(max(mmr_scores))
-                selected_keywords.append(remaining_keywords[max_mmr_index])
+    # Extract keywords
+    if st.button("Extract Keywords"):
+        keywords = kw_model.extract_keywords(doc,
+                                             keyphrase_ngram_range=(min_ngram, max_ngram),
+                                             stop_words='english' if remove_stopwords else None,
+                                             use_mmr=apply_mmr,
+                                             diversity=0.2)
 
-            keywords = selected_keywords  # Update keywords with MMR-selected keywords
+        selected_keywords = keywords[:num_results]
 
         st.write(f"Top {num_results} Keywords:")
-        for keyword, score in keywords:
+        for keyword, score in selected_keywords:
             st.write(f"- {keyword} (Score: {score})")
 
 # Run the app