Update app.py

app.py

@@ -30,41 +30,47 @@ def main():
     st.set_page_config(page_title="Electricity Bill Calculator", layout="wide")
     st.title("Electricity Bill Calculator")
 
-    # Sidebar for company selection
+    # Sidebar for company and appliance selection
     with st.sidebar:
-        st.header("Select Company")
+        st.header("Settings")
+        
+        # Select company
+        st.subheader("Select Company")
         selected_company = st.selectbox("Company", options=list(TARIFF_URLS.keys()))
-        st.write("Selected Company:", selected_company)
+        if st.button("Fetch Tariff Rates", key="fetch_tariff"):
+            url = TARIFF_URLS[selected_company]
+            st.write(f"Fetching data for {selected_company}...")
+            with st.spinner("Fetching tariff data..."):
+                tariff_data = scrape_tariff_data(url)
+            if tariff_data:
+                st.success(f"Tariff data fetched for {selected_company}!")
+            else:
+                st.error("Failed to fetch tariff data.")
 
-        # Fetch tariff data from the selected company
-        url = TARIFF_URLS[selected_company]
-        with st.spinner("Fetching tariff data..."):
-            tariff_data = scrape_tariff_data(url)
+        # Appliance selection
+        st.subheader("Add Appliances")
+        appliances = {
+            "Fan": 75,  # Power in watts
+            "Bulb": 40,
+            "Refrigerator": 150,
+            "Air Conditioner": 1500,
+            "Washing Machine": 500,
+            "Microwave": 1200
+        }
 
-    # Main app interface for bill calculation
-    st.header("Appliances and Usage")
-    st.write("Add your appliances and usage details below:")
-
-    # Predefined appliances
-    appliances = {
-        "Fan": 75,  # Power in watts
-        "Bulb": 40,
-        "Refrigerator": 150,
-        "Air Conditioner": 1500,
-        "Washing Machine": 500,
-        "Microwave": 1200
-    }
+        # Allow users to add predefined or custom appliances
+        selected_appliances = st.multiselect("Select Appliances", options=list(appliances.keys()))
+        custom_appliance_name = st.text_input("Custom Appliance Name")
+        custom_appliance_power = st.number_input("Custom Appliance Power (Watts)", min_value=1, step=1)
 
-    # Allow users to add appliances and customize
-    selected_appliances = st.multiselect("Select Appliances", options=list(appliances.keys()))
-    custom_appliance_name = st.text_input("Add Custom Appliance Name")
-    custom_appliance_power = st.number_input("Custom Appliance Power (Watts)", min_value=1, step=1)
+        if custom_appliance_name and custom_appliance_power:
+            appliances[custom_appliance_name] = custom_appliance_power
+            st.success(f"Added custom appliance: {custom_appliance_name} ({custom_appliance_power}W)")
 
-    if custom_appliance_name and custom_appliance_power:
-        appliances[custom_appliance_name] = custom_appliance_power
-        st.success(f"Added custom appliance: {custom_appliance_name} ({custom_appliance_power}W)")
+    # Main app interface for bill calculation
+    st.header("Appliance Usage")
+    st.write("Add usage details for the selected appliances:")
 
-    # Input fields for usage details
     total_units = 0
     for appliance in selected_appliances:
         st.subheader(f"{appliance} ({appliances[appliance]}W)")
@@ -78,11 +84,14 @@ def main():
 
     # Calculate the bill
     if st.button("Calculate Bill"):
-        bill = calculate_bill(total_units, tariff_data)
-        if bill > 0:
-            st.success(f"Estimated Monthly Bill: Rs. {bill:.2f}")
+        if 'tariff_data' in locals() and tariff_data:
+            bill = calculate_bill(total_units, tariff_data)
+            if bill > 0:
+                st.success(f"Estimated Monthly Bill: Rs. {bill:.2f}")
+            else:
+                st.error("Could not calculate the bill. Please check the tariff data or try again.")
         else:
-            st.error("Could not calculate the bill. Please check the tariff data or try again.")
+            st.error("Please fetch the tariff rates first.")
 
 if __name__ == "__main__":
     main()