Update app.py

app.py

@@ -7,56 +7,73 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def keyword_research_tool(topic: str, max_results: int) -> str:
+def generate_keywords(topic: str, max_keywords: int = 30) -> str:
     """Генерирует релевантные ключевые слова для заданной темы
     Args:
-        topic: Основная тема для генерации ключевых слов
-        max_results: Максимальное количество возвращаемых ключевых слов
+        topic: Тема для генерации (на русском языке)
+        max_keywords: Максимальное количество возвращаемых ключевых слов
     """
     import random
-    from typing import List
+    from collections import deque
     
-    # Базовые модификаторы для расширения темы
-    modifiers = [
-        "лучшие", "топ", "как", "советы", "рецепты",
-        "польза", "вред", "отзывы", "купить", "сделать",
-        "обзор", "курс", "уроки", "для начинающих", 
-        "2023", "идеи", "примеры", "методы", "технологии",
-        "тренды", "планы", "стоимость", "бесплатно", "преимущества"
-    ]
-    
-    # Разбиваем тему на основные слова
-    seed_words = [word.strip() for word in topic.split() if len(word) > 3]
+    # Инициализация компонентов генерации
+    base_words = [w.strip().lower() for w in topic.split() if len(w) > 2]
+    if not base_words:
+        return "Необходимо указать более конкретную тему"
     
-    # Генерируем комбинации
-    candidates = set()
-    
-    # Добавляем базовые варианты
-    candidates.add(topic)
-    candidates.add(f"лучшие {topic}")
-    candidates.add(f"{topic} 2023")
+    # Паттерны для генерации
+    patterns = [
+        ["вопросы", "что такое", "как сделать", "почему", "когда", "где найти"],
+        ["топ", "лучшие", "рейтинг", "самые популярные"],
+        ["методы", "способы", "технологии", "решения"],
+        ["обзор", "отзывы", "сравнение", "виды"],
+        ["курс", "урок", "обучение", "самоучитель"],
+        ["2025", "новинки", "тренды", "будущее"],
+        ["для начинающих", "профессиональные", "премиум"],
+        ["бесплатно", "купить", "заказать", "стоимость"],
+        ["преимущества", "недостатки", "польза", "вред"],
+        ["идеи", "примеры", "кейсы", "истории успеха"]
+    ]
     
-    # Генерируем комбинации с модификаторами
-    for modifier in modifiers:
-        candidates.add(f"{modifier} {topic}")
-        candidates.add(f"{topic} {modifier}")
-        for word in seed_words:
-            candidates.add(f"{modifier} {word}")
-            candidates.add(f"{word} {modifier}")
+    # Генерация вариантов
+    keywords = set()
+    queue = deque([(0, [], base_words)])
     
-    # Добавляем вопросы
-    question_words = ["как", "что", "где", "когда", "почему"]
-    for qw in question_words:
-        candidates.add(f"{qw} {topic}")
-        for word in seed_words:
-            candidates.add(f"{qw} {word}")
+    while queue:
+        level, current, remaining = queue.popleft()
+        
+        if level >= 3 or not remaining:
+            if current:
+                keywords.add(" ".join(current))
+            continue
+            
+        # Добавляем паттерны
+        if level == 0:
+            for pattern_group in patterns:
+                for p in pattern_group:
+                    new_phrase = [p] + current
+                    queue.append((level+1, new_phrase, remaining))
+        
+        # Комбинируем слова
+        for i in range(len(remaining)):
+            next_word = remaining[i]
+            new_phrase = current + [next_word]
+            new_remaining = remaining[:i] + remaining[i+1:]
+            queue.append((level+1, new_phrase, new_remaining))
+            
+            # Добавляем модификаторы
+            modifiers = ["", "2025", "курс", "обзор", "топ", "способы"]
+            for mod in modifiers:
+                if mod:
+                    modified = current + [f"{next_word}-{mod}"]
+                    queue.append((level+2, modified, new_remaining))
     
-    # Преобразуем в список и перемешиваем
-    results = list(candidates)
-    random.shuffle(results)
+    # Ранжирование и выборка
+    result = sorted(keywords, 
+                   key=lambda x: (-len(x.split()), x))
+    random.shuffle(result[:len(result)//2])
     
-    # Выбираем запрошенное количество
-    return ", ".join(results[:max(max_results, 1)])
+    return ", ".join(result[:max(max_keywords, 1)])
 
 # Initialize components
 final_answer = FinalAnswerTool()
@@ -80,7 +97,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     tools=[
         image_generation_tool,
-        keyword_research_tool,
+        generate_keywords,
         final_answer,
         DuckDuckGoSearchTool()
     ],