Update app.py

app.py

@@ -7,51 +7,56 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def generate_keywords(topic: str, num_keywords: int) -> str:
-    """A tool that generates relevant keywords for a given topic
-    
+def keyword_research_tool(topic: str, max_results: int) -> str:
+    """Генерирует релевантные ключевые слова для заданной темы
     Args:
-        topic: The main topic or subject area to generate keywords for
-        num_keywords: Number of keywords to generate (1-20)
+        topic: Основная тема для генерации ключевых слов
+        max_results: Максимальное количество возвращаемых ключевых слов
     """
-    # Common keyword patterns and modifiers
+    import random
+    from typing import List
+    
+    # Базовые модификаторы для расширения темы
     modifiers = [
-        "как", "почему", "что такое", "топ", "лучший",
-        "обзор", "руководство", "советы", "примеры",
-        "методы", "способы", "виды", "типы"
+        "лучшие", "топ", "как", "советы", "рецепты",
+        "польза", "вред", "отзывы", "купить", "сделать",
+        "обзор", "курс", "уроки", "для начинающих", 
+        "2023", "идеи", "примеры", "методы", "технологии",
+        "тренды", "планы", "стоимость", "бесплатно", "преимущества"
     ]
     
-    suffixes = [
-        "для начинающих", "пошагово", "с нуля",
-        "быстро", "эффективно", "просто",
-        "в 2025 году", "онлайн", "на практике"
-    ]
+    # Разбиваем тему на основные слова
+    seed_words = [word.strip() for word in topic.split() if len(word) > 3]
     
-    import random
+    # Генерируем комбинации
+    candidates = set()
+    
+    # Добавляем базовые варианты
+    candidates.add(topic)
+    candidates.add(f"лучшие {topic}")
+    candidates.add(f"{topic} 2023")
+    
+    # Генерируем комбинации с модификаторами
+    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}")
     
-    # Ensure number of keywords is within reasonable range
-    num_keywords = max(1, min(20, num_keywords))
+    # Добавляем вопросы
+    question_words = ["как", "что", "где", "когда", "почему"]
+    for qw in question_words:
+        candidates.add(f"{qw} {topic}")
+        for word in seed_words:
+            candidates.add(f"{qw} {word}")
     
-    # Generate keyword combinations
-    keywords = set()
-    while len(keywords) < num_keywords:
-        keyword_type = random.randint(1, 3)
-        
-        if keyword_type == 1:
-            keyword = f"{random.choice(modifiers)} {topic}"
-        elif keyword_type == 2:
-            keyword = f"{topic} {random.choice(suffixes)}"
-        else:
-            keyword = f"{random.choice(modifiers)} {topic} {random.choice(suffixes)}"
-            
-        keywords.add(keyword)
+    # Преобразуем в список и перемешиваем
+    results = list(candidates)
+    random.shuffle(results)
     
-    # Format the response
-    response = f"""Ключевые слова по теме "{topic}":\n\n"""
-    for i, keyword in enumerate(keywords, 1):
-        response += f"{i}. {keyword}\n"
-        
-    return response
+    # Выбираем запрошенное количество
+    return ", ".join(results[:max(max_results, 1)])
 
 # Initialize components
 final_answer = FinalAnswerTool()
@@ -75,7 +80,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     tools=[
         image_generation_tool,
-        generate_keywords,
+        keyword_research_tool,
         final_answer,
         DuckDuckGoSearchTool()
     ],