Update app.py

app.py

@@ -19,9 +19,11 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def calculate_bandwidth():
+def calculate_bandwidth(users, usage):
     """
     Calculate the recommended internet speed based on user inputs.
+    :param users: Number of users
+    :param usage: Dictionary with usage categories and number of users per category
     :return: Recommended bandwidth in Mbps
     """
     usage_requirements = {
@@ -33,16 +35,6 @@ def calculate_bandwidth():
         "remote_work": 3      # Mbps per user
     }
     
-    users = int(input("Enter the number of users: "))
-    usage = {
-        "browsing": int(input("Users browsing the web: ")),
-        "video_call": int(input("Users on video calls: ")),
-        "hd_streaming": int(input("Users streaming in HD: ")),
-        "4k_streaming": int(input("Users streaming in 4K: ")),
-        "gaming": int(input("Users gaming online: ")),
-        "remote_work": int(input("Users working remotely: "))
-    }
-    
     total_bandwidth = sum(usage_requirements[activity] * count for activity, count in usage.items())
     
     overhead = 1.2  # 20% overhead for seamless experience