Update app.py

app.py

@@ -25,7 +25,29 @@ def my_party_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg1: A word that describes the group of people: family, friends, coworkers
         arg2: The total number of people
     """
-    return "Outdoor party at backyard with grilled meat and wine"
+    party = ""
+    if "friends" in arg1:
+        if arg2 < 6:
+            party = "Outdoor party at backyard with grilled meat and wine"
+        else:
+            party = "Book a cabin near the lake. Pizza and drinks"
+
+    elif "family" in arg1:
+        if arg2 < 4:
+            party = "Cozy party at home, pirogi, meat and wine"
+        else:
+            party = "Book a table in restaurant"
+
+    elif "coworkers" in arg1:
+        if arg2 < 6:
+            party = "Outdoor party at backyard with grilled meat and drinks"
+        else:
+            party = "Book a party room. Pizza and drinks"
+
+    else:
+        party = "Please give more information"
+        
+    return party
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: