Add application file 8

app.py

@@ -4,14 +4,14 @@ import os
 @tool
 def suggest_menu(occasion: str) -> str:
     """
-    "Предлагает меню на основе случая."
-    Аргументы:
-        случай: "Тип случая для вечеринки."
+    Предлагает меню на основе случая.
+    Args:
+        occasion: Тип случая для вечеринки.
     """
     if occasion == "casual":
         return "Пицца, закуски и напитки."
     elif occasion == "formal":
-        return "3-х курсный ужин с вином и десертом."
+        return "3-ходовой ужин с вином и десертом."
     elif occasion == "superhero":
         return "Буфет с высокоэнергетической и здоровой едой."
     else:
@@ -20,44 +20,46 @@ def suggest_menu(occasion: str) -> str:
 @tool
 def catering_service_tool(query: str) -> str:
     """
-    "Этот инструмент возвращает самый высоко оцененный сервис кейтеринга в Готэм-Сити."
+    Этот инструмент возвращает самый рейтинговый сервис кейтеринга в Готэм-Сити.
     
-    Аргументы:
-        запрос: "Поисковый запрос для поиска сервисов кейтеринга."
+    Args:
+        query: Поисковый запрос для поиска сервисов кейтеринга.
     """
     # Примерный список сервисов кейтеринга и их рейтингов
     services = {
-        "Готэм Кейтеринг Ко.": 4.9,
-        "Уэйн Мэнор Кейтеринг": 4.8,
-        "Готэм-Сити Ивенты": 4.7,
+        "Gotham Catering Co.": 4.9,
+        "Wayne Manor Catering": 4.8,
+        "Gotham City Events": 4.7,
     }
     
-    # Найти самый высоко оцененный сервис кейтеринга (симуляция фильтрации поискового запроса)
-    return max(services, key=services.get)
+    # Найти самый рейтинговый сервис кейтеринга (симуляция фильтрации поискового запроса)
+    best_service = max(services, key=services.get)
+    
+    return best_service
 
 class SuperheroPartyThemeTool(Tool):
-    name = "генератор_темы_вечеринки_супергероев"
+    name = "superhero_party_theme_generator"
     description = """
-    "Этот инструмент предлагает творческие темы вечеринок супергероев на основе категории."
-    """
+    Этот инструмент предлагает творческие идеи вечеринок на тему супергероев на основе категории.
+    Он возвращает уникальную идею вечеринки."""
     
     inputs = {
-        "категория": {
-            "тип": "строка",
-            "описание": "Тип вечеринки супергероев (например, 'классические герои', 'маскарад злодеев', 'футуристический Готэм').",
+        "category": {
+            "type": "string",
+            "description": "Тип вечеринки супергероев (например, 'классические герои', 'маскарад злодеев', 'футуристический Готэм').",
         }
     }
     
-    output_type = "строка"
+    output_type = "string"
 
     def forward(self, category: str):
         themes = {
-            "классические герои": "Гала Лиги Справедливости: Гости приходят одетыми как их любимые герои DC с тематическими коктейлями, такими как 'Панч Криптонита'.",
-            "маскарад злодеев": "Бал злодеев Готэма: Таинственный маскарад, где гости одеваются как классические злодеи Бэтмена.",
-            "футуристический Готэм": "Ночь Нео-Готэма: Вечеринка в стиле киберпанка, вдохновленная Бэтменом из будущего, с неоновыми украшениями и футуристическими гаджетами."
+            "classic heroes": "Гала Лиги Справедливости: Гости приходят одетыми в своих любимых героев DC с тематическими коктейлями, такими как 'Панч Криптонита'.",
+            "villain masquerade": "Бал злодеев Готэма: Таинственный маскарад, где гости одеваются как классические злодеи Бэтмена.",
+            "futuristic Gotham": "Ночь Нео-Готэма: Вечеринка в стиле киберпанка, вдохновленная Бэтменом из будущего, с неоновыми украшениями и футуристическими гаджетами."
         }
         
-        return "Тема вечеринки не найдена. Попробуйте 'классические герои', 'маскарад злодеев' или 'футуристический Готэм'."
+        return themes.get(category.lower(), "Идея тематической вечеринки не найдена. Попробуйте 'классические герои', 'маскарад злодеев' или 'футуристический Готэм'.")
 
 
 # Альфред, дворецкий, готовит меню для вечеринки
@@ -74,4 +76,4 @@ agent = CodeAgent(
     verbosity_level=2
 )
 
-agent.run("Дайте мне лучший плейлист для вечеринки в поместье Уэйна. Идея вечеринки - 'маскарад злодеев'")
+agent.run("Дайте мне лучший плейлист для вечеринки в особняке Уэйна. Идея вечеринки - тема 'маскарад злодеев'")