Update app.py

app.py

@@ -1,23 +1,49 @@
 import gradio as gr
 from keybert import KeyBERT
+from nltk.tokenize import sent_tokenize
 
-# Инициализируем KeyBERT
-kw_model = KeyBERT()
+# Инициализируем KeyBERT с заданной моделью
+from sentence_transformers import SentenceTransformer
+model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
+kw_model = KeyBERT(model)
 
-def extract_keywords(text):
+# Функция извлечения ключевых слов
+def extract_keywords(text, top_n=10, diversity=0.5, use_maxsum=False, max_ngram=3):
     if not text or not text.strip():
         return []
-    # Извлекаем топ-5 ключевых слов
-    keywords = kw_model.extract_keywords(text, top_n=5)
-    return keywords  # список кортежей (keyword, score)
+    
+    # Разбиваем текст на абзацы
+    segments = sent_tokenize(text)
+    
+    all_keywords = []
+    for segment in segments:
+        keywords = kw_model.extract_keywords(
+            segment,
+            keyphrase_ngram_range=(1, max_ngram),
+            top_n=top_n,
+            diversity=diversity,
+            use_maxsum=use_maxsum,
+            stop_words="english"
+        )
+        all_keywords.extend(keywords)
+    
+    # Убираем дубликаты
+    unique_keywords = list(set(all_keywords))
+    return unique_keywords  # возвращаем список кортежей (ключевое слово, оценка)
 
-# Готовим Gradio-интерфейс
+# Создаем Gradio интерфейс
 iface = gr.Interface(
     fn=extract_keywords,
-    inputs=gr.Textbox(label="Enter text"),
-    outputs="json", 
+    inputs=[
+        gr.Textbox(label="Enter text", lines=10, placeholder="Введите текст здесь"),
+        gr.Slider(5, 50, value=10, step=1, label="Количество ключевых слов (top_n)"),
+        gr.Slider(0.0, 1.0, value=0.5, step=0.1, label="Разнообразие ключевых слов (diversity)"),
+        gr.Checkbox(label="Использовать MaxSum", value=False),
+        gr.Slider(1, 3, value=3, step=1, label="Максимальная длина фраз (n-gram)")
+    ],
+    outputs="json",
     title="KeyBERT Extractor",
-    description="Введите текст, чтобы получить ключевые слова"
+    description="Введите текст, настройте параметры и извлеките ключевые слова"
 )
 
 if __name__ == "__main__":