Update app.py

app.py

@@ -22,7 +22,7 @@ def calculate_bill(units_consumed, tariff_data):
                     threshold = int(unit_range.strip(">").strip())
                     if units_consumed > threshold:
                         return units_consumed * rate
-            except:
+            except Exception:
                 pass
     return 0.0  # Default to 0 if no matching rate is found
 
@@ -30,24 +30,23 @@ def main():
     st.set_page_config(page_title="Electricity Bill Calculator", layout="wide")
     st.title("Electricity Bill Calculator")
 
-    # Sidebar for company and appliance selection
+    # Sidebar for settings
     with st.sidebar:
         st.header("Settings")
-        
-        # Select company
+
+        # Company selection
         st.subheader("Select Company")
         selected_company = st.selectbox("Company", options=list(TARIFF_URLS.keys()))
+
         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.session_state['tariff_data'] = scrape_tariff_data(url)
+            if st.session_state['tariff_data']:
                 st.success(f"Tariff data fetched for {selected_company}!")
             else:
-                st.error("Failed to fetch tariff data.")
+                st.error("Failed to fetch tariff data. Please check the URL or try again.")
 
-        # Appliance selection
+        # Appliances selection
         st.subheader("Add Appliances")
         appliances = {
             "Fan": 75,  # Power in watts
@@ -58,18 +57,19 @@ def main():
             "Microwave": 1200
         }
 
-        # Allow users to add predefined or custom appliances
+        # Allow users to select predefined or add 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)
 
+        # Add custom appliances
         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
+    # Main application for usage inputs and bill calculation
     st.header("Appliance Usage")
-    st.write("Add usage details for the selected appliances:")
+    st.write("Enter usage details for the selected appliances:")
 
     total_units = 0
     for appliance in selected_appliances:
@@ -84,14 +84,17 @@ def main():
 
     # Calculate the bill
     if st.button("Calculate Bill"):
-        if 'tariff_data' in locals() and tariff_data:
-            bill = calculate_bill(total_units, tariff_data)
+        if 'tariff_data' in st.session_state and st.session_state['tariff_data']:
+            bill = calculate_bill(total_units, st.session_state['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.")
+                st.error("Could not calculate the bill. Please check the tariff data.")
         else:
             st.error("Please fetch the tariff rates first.")
 
 if __name__ == "__main__":
+    # Initialize session state to store tariff data
+    if 'tariff_data' not in st.session_state:
+        st.session_state['tariff_data'] = None
     main()