Update app.py

app.py

@@ -44,12 +44,11 @@ def calculate_monthly_bill(tariff_data, load, daily_usage):
     
     total_bill = 0
     for slab in tariff_data:
-        slab_range = slab['slab'].split('-')
-        lower_limit = float(slab_range[0])
-        upper_limit = float(slab_range[1]) if len(slab_range) > 1 else float('inf')
+        # Default to infinity for upper limit if not defined
+        upper_limit = slab['upper_limit'] if slab['upper_limit'] != float('inf') else monthly_consumption
         
-        if monthly_consumption > lower_limit:
-            consumption_in_slab = min(monthly_consumption, upper_limit) - lower_limit
+        if monthly_consumption > slab['lower_limit']:
+            consumption_in_slab = min(monthly_consumption, upper_limit) - slab['lower_limit']
             total_bill += consumption_in_slab * slab['rate']
     
     return round(total_bill, 2)