Update app.py

app.py

@@ -27,13 +27,21 @@ def main():
     # 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)
 
+    # Get user choice for keyword diversity (MMR only)
+    diversity = st.slider("Keyword Diversity (MMR)", min_value=0.0, max_value=1.0, value=0.2, step=0.1, format="%.1f")
+
     # 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)
+        if apply_mmr:
+            keywords = kw_model.extract_keywords(doc,
+                                                 keyphrase_ngram_range=(min_ngram, max_ngram),
+                                                 stop_words='english' if remove_stopwords else None,
+                                                 use_mmr=True,
+                                                 diversity=diversity)
+        else:
+            keywords = kw_model.extract_keywords(doc,
+                                                 keyphrase_ngram_range=(min_ngram, max_ngram),
+                                                 stop_words='english' if remove_stopwords else None)
 
         selected_keywords = keywords[:num_results]