zasharepw77 commited on
Commit
48b2b94
·
1 Parent(s): 704c6ae

Add application file 7

Browse files
Files changed (1) hide show
  1. app.py +29 -31
app.py CHANGED
@@ -4,65 +4,63 @@ import os
4
  @tool
5
  def suggest_menu(occasion: str) -> str:
6
  """
7
- Suggests a menu based on the occasion.
8
- Args:
9
- occasion: The type of occasion for the party.
10
  """
11
  if occasion == "casual":
12
- return "Pizza, snacks, and drinks."
13
  elif occasion == "formal":
14
- return "3-course dinner with wine and dessert."
15
  elif occasion == "superhero":
16
- return "Buffet with high-energy and healthy food."
17
  else:
18
- return "Custom menu for the butler."
19
 
20
  @tool
21
  def catering_service_tool(query: str) -> str:
22
  """
23
- This tool returns the highest-rated catering service in Gotham City.
24
 
25
- Args:
26
- query: A search term for finding catering services.
27
  """
28
- # Example list of catering services and their ratings
29
  services = {
30
- "Gotham Catering Co.": 4.9,
31
- "Wayne Manor Catering": 4.8,
32
- "Gotham City Events": 4.7,
33
  }
34
 
35
- # Find the highest rated catering service (simulating search query filtering)
36
- best_service = max(services, key=services.get)
37
-
38
- return best_service
39
 
40
  class SuperheroPartyThemeTool(Tool):
41
- name = "superhero_party_theme_generator"
42
  description = """
43
- This tool suggests creative superhero-themed party ideas based on a category.
44
- It returns a unique party theme idea."""
45
 
46
  inputs = {
47
- "category": {
48
- "type": "string",
49
- "description": "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic Gotham').",
50
  }
51
  }
52
 
53
- output_type = "string"
54
 
55
  def forward(self, category: str):
56
  themes = {
57
- "classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
58
- "villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
59
- "futuristic Gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
60
  }
61
 
62
- return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
63
 
64
 
65
- # Alfred, the butler, preparing the menu for the party
66
  agent = CodeAgent(
67
  tools=[
68
  DuckDuckGoSearchTool(),
@@ -76,4 +74,4 @@ agent = CodeAgent(
76
  verbosity_level=2
77
  )
78
 
79
- agent.run("Give me the best playlist for a party at the Wayne's mansion. The party idea is a 'villain masquerade' theme")
 
4
  @tool
5
  def suggest_menu(occasion: str) -> str:
6
  """
7
+ "Предлагает меню на основе случая."
8
+ Аргументы:
9
+ случай: "Тип случая для вечеринки."
10
  """
11
  if occasion == "casual":
12
+ return "Пицца, закуски и напитки."
13
  elif occasion == "formal":
14
+ return "3 курсный ужин с вином и десертом."
15
  elif occasion == "superhero":
16
+ return "Буфет с высокоэнергетической и здоровой едой."
17
  else:
18
+ return "Пользовательское меню для дворецкого."
19
 
20
  @tool
21
  def catering_service_tool(query: str) -> str:
22
  """
23
+ "Этот инструмент возвращает самый высоко оцененный сервис кейтеринга в Готэм-Сити."
24
 
25
+ Аргументы:
26
+ запрос: "Поисковый запрос для поиска сервисов кейтеринга."
27
  """
28
+ # Примерный список сервисов кейтеринга и их рейтингов
29
  services = {
30
+ "Готэм Кейтеринг Ко.": 4.9,
31
+ "Уэйн Мэнор Кейтеринг": 4.8,
32
+ "Готэм-Сити Ивенты": 4.7,
33
  }
34
 
35
+ # Найти самый высоко оцененный сервис кейтеринга (симуляция фильтрации поискового запроса)
36
+ return max(services, key=services.get)
 
 
37
 
38
  class SuperheroPartyThemeTool(Tool):
39
+ name = "генератор_темы_вечеринки_супергероев"
40
  description = """
41
+ "Этот инструмент предлагает творческие темы вечеринок супергероев на основе категории."
42
+ """
43
 
44
  inputs = {
45
+ "категория": {
46
+ "тип": "строка",
47
+ "описание": "Тип вечеринки супергероев (например, 'классические герои', 'маскарад злодеев', 'футуристический Готэм').",
48
  }
49
  }
50
 
51
+ output_type = "строка"
52
 
53
  def forward(self, category: str):
54
  themes = {
55
+ "классические герои": "Гала Лиги Справедливости: Гости приходят одетыми как их любимые герои DC с тематическими коктейлями, такими как 'Панч Криптонита'.",
56
+ "маскарад злодеев": "Бал злодеев Готэма: Таинственный маскарад, где гости одеваются как классические злодеи Бэтмена.",
57
+ "футуристический Готэм": "Ночь Нео-Готэма: Вечеринка в стиле киберпанка, вдохновленная Бэтменом из будущего, с неоновыми украшениями и футуристическими гаджетами."
58
  }
59
 
60
+ return "Тема вечеринки не найдена. Попробуйте 'классические герои', 'маскарад злодеев' или 'футуристический Готэм'."
61
 
62
 
63
+ # Альфред, дворецкий, готовит меню для вечеринки
64
  agent = CodeAgent(
65
  tools=[
66
  DuckDuckGoSearchTool(),
 
74
  verbosity_level=2
75
  )
76
 
77
+ agent.run("Дайте мне лучший плейлист для вечеринки в поместье Уэйна. Идея вечеринки - 'маскарад злодеев'")