Update app.py

app.py

@@ -5,13 +5,7 @@ import scraper
 # URLs for different companies
 tariff_urls = {
     "IESCO": "https://iesco.com.pk/index.php/customer-services/tariff-guide",
-    "FESCO": "https://fesco.com.pk/tariff",
-    "HESCO": "http://www.hesco.gov.pk/htmls/tariffs.htm",
-    "KE": "https://www.ke.com.pk/customer-services/tariff-structure/",
-    "LESCO": "https://www.lesco.gov.pk/ElectricityTariffs",
-    "PESCO": "https://pesconlinebill.pk/pesco-tariff/",
-    "QESCO": "http://qesco.com.pk/Tariffs.aspx",
-    "TESCO": "https://tesco.gov.pk/index.php/electricity-traiff"
+    # Add other companies as needed
 }
 
 def main():
@@ -36,7 +30,7 @@ def main():
             st.write(f"Monthly Bill for {appliance_name}: Rs. {monthly_bill}")
             st.write(f"Carbon Footprint: {carbon_footprint} kg CO2")
         else:
-            st.write("Error fetching tariff data. Please try again.")
+            st.error("Error fetching tariff data. Please try again.")
 
 def calculate_monthly_bill(tariff_data, load, daily_usage):
     daily_consumption = load * daily_usage  # kWh per day
@@ -44,11 +38,8 @@ def calculate_monthly_bill(tariff_data, load, daily_usage):
     
     total_bill = 0
     for slab in tariff_data:
-        # 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 > slab['lower_limit']:
-            consumption_in_slab = min(monthly_consumption, upper_limit) - slab['lower_limit']
+            consumption_in_slab = min(monthly_consumption, slab['upper_limit']) - slab['lower_limit']
             total_bill += consumption_in_slab * slab['rate']
     
     return round(total_bill, 2)
@@ -57,8 +48,7 @@ def calculate_carbon_footprint(load, daily_usage):
     daily_consumption = load * daily_usage  # kWh per day
     monthly_consumption = daily_consumption * 30  # kWh per month
     
-    # Assuming 0.92 kg CO2 per kWh
-    emissions_factor = 0.92
+    emissions_factor = 0.92  # kg CO2 per kWh
     carbon_footprint = monthly_consumption * emissions_factor
     return round(carbon_footprint, 2)