Update app.py

app.py

@@ -4,23 +4,20 @@ from tariff_scraper import scrape_tariff_data
 def main():
     st.title("Electricity Bill Calculator with Tariff Scraper")
     
-    # Fetch tariff rates
+    # Tariff Rates Section
     st.sidebar.subheader("Tariff Rates")
     url = st.sidebar.text_input("Tariff URL", "https://iesco.com.pk/index.php/customer-services/tariff-guide")
 
     if st.sidebar.button("Fetch Tariff Rates"):
         with st.spinner("Fetching tariff rates..."):
             data = scrape_tariff_data(url)
-            if isinstance(data, dict):  # Check if data is a valid dictionary
+            if isinstance(data, dict):  # Check if valid tariff data is returned
                 st.session_state["tariff_rates"] = data
                 st.sidebar.success("Tariff rates fetched successfully!")
-                st.write("### Tariff Rates")
-                for category, rate in data.items():
-                    st.write(f"{category}: {rate} PKR/unit")
             else:
                 st.sidebar.error(data)
 
-    # Use fetched tariff rates
+    # Use Fetched Tariff Rates
     tariff_categories = st.session_state.get("tariff_rates", {})
     if tariff_categories:
         load_type = st.sidebar.selectbox("Select Load Type", options=tariff_categories.keys())
@@ -30,7 +27,7 @@ def main():
         load_type = "Default"
         tariff_rate = st.sidebar.number_input("Default Tariff Rate (PKR/unit)", min_value=0.0, value=20.0)
 
-    # Electricity bill calculation
+    # Appliance Usage Section
     st.sidebar.subheader("Appliance Usage")
     appliances = {
         "Fan": 75, "Air Conditioner (1 Ton)": 1500, "Refrigerator": 150,
@@ -49,12 +46,13 @@ def main():
         })
         st.sidebar.success(f"{quantity} x {appliance} added.")
 
-    # Display appliances and calculate bill
+    # Display Appliances
     st.write("### Selected Appliances")
     if "appliances" in st.session_state:
         for app in st.session_state["appliances"]:
             st.write(f"{app['quantity']} x {app['appliance']} ({app['usage_hours']} hours/day)")
         
+        # Calculate Bill
         if st.button("Calculate Bill"):
             total_units = sum(
                 app["quantity"] * app["load"] * app["usage_hours"] / 1000 for app in st.session_state["appliances"]
@@ -66,4 +64,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-