Update app.py

app.py

@@ -8,15 +8,32 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity!
+
 @tool
-def keyword_generator(arg1:str)-> str: # it's important to specify the return type
-    # Keep this format for the tool description / args description but feel free to modify the tool
-    """A tool that generate keywords by the topic 
+def generate_keywords(topic: str, max_keywords: int) -> str:
+    """A tool that generates the most relevant keywords based on a given topic.
     Args:
-        arg1: generate 10 most relevant keywords
-        
+        topic: the subject or theme for which keywords need to be generated
+        max_keywords: the maximum number of keywords to generate
     """
-    return "keywords"
+    # Basic preprocessing to remove common stopwords (without external libraries)
+    stopwords = set(["и", "в", "на", "с", "по", "для", "как", "что", "это", "а", "но", "я", "у", "за", "из", "о", "об", "от"])
+    
+    # Splitting the topic into words and filtering out short words and stopwords
+    words = topic.lower().split()
+    filtered_words = [word for word in words if len(word) > 2 and word not in stopwords]
+    
+    # Generating unique keywords by frequency and limiting to max_keywords
+    keyword_frequency = {}
+    for word in filtered_words:
+        keyword_frequency[word] = keyword_frequency.get(word, 0) + 1
+    
+    # Sorting keywords by frequency (descending) and selecting top ones
+    sorted_keywords = sorted(keyword_frequency.keys(), key=lambda x: keyword_frequency[x], reverse=True)
+    top_keywords = sorted_keywords[:max_keywords]
+    
+    return ", ".join(top_keywords) if top_keywords else "Нет релевантных ключевых слов"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -51,7 +68,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, keyword_generator], # add your tools here (don't remove final_answer)
+    tools=[final_answer, generate_keywords], # add your tools here (don't remove final_answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,