Update app.py

app.py

@@ -21,21 +21,18 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
 
 @tool
 def calculate_bandwidth(users: int, usage: Dict[str, int]) -> float:
-    """
-    Calculate the recommended internet speed based on user inputs.
-
-    :param users: The total number of users requiring internet access.
-    :type users: int
-    :param usage: A dictionary with usage categories as keys. Expected keys are:
-                  "browsing" - Number of users browsing the web.
-                  "video_call" - Number of users on video calls.
-                  "hd_streaming" - Number of users streaming in HD.
-                  "4k_streaming" - Number of users streaming in 4K.
-                  "gaming" - Number of users gaming online.
-                  "remote_work" - Number of users working remotely.
-    :type usage: Dict[str, int]
-    :return: Recommended bandwidth in Mbps to ensure smooth performance.
-    :rtype: float
+    """Calculate the recommended internet speed based on user inputs.
+    
+    Args:
+        users: The total number of users requiring internet access.
+        usage: A dictionary with usage categories as keys and the number of users per category as values.
+               Expected keys are:
+                   - "browsing": Number of users browsing the web.
+                   - "video_call": Number of users on video calls.
+                   - "hd_streaming": Number of users streaming in HD.
+                   - "4k_streaming": Number of users streaming in 4K.
+                   - "gaming": Number of users gaming online.
+                   - "remote_work": Number of users working remotely.
     """
     usage_requirements = {
         "browsing": 1,        # Mbps per user
@@ -46,8 +43,7 @@ def calculate_bandwidth(users: int, usage: Dict[str, int]) -> float:
         "remote_work": 3      # Mbps per user
     }
     
-    total_bandwidth = sum(usage_requirements[activity] * count for activity, count in usage.items())
-    
+    total_bandwidth = sum(usage_requirements[activity] * usage.get(activity, 0) for activity in usage_requirements)
     overhead = 1.2  # 20% overhead for seamless experience
     return round(total_bandwidth * overhead, 2)