Update app.py

app.py

@@ -2,32 +2,68 @@ import streamlit as st
 from tariff_scraper import scrape_tariff_data
 
 def main():
-    """
-    Main application for scraping and previewing tariff data.
-    """
-    st.title("Electricity Tariff Data Scraper")
-    st.sidebar.write("Enter the URL of the electricity tariff page:")
-
-    # Input for URL
-    url = st.sidebar.text_input(
-        "Tariff URL",
-        "https://iesco.com.pk/index.php/customer-services/tariff-guide"
-    )
+    st.title("Electricity Bill Calculator with Tariff Scraper")
     
-    if st.sidebar.button("Scrape Tariff Data"):
-        if url:
-            with st.spinner("Scraping data..."):
-                data = scrape_tariff_data(url)
-                if isinstance(data, list):
-                    st.success("Tariff data fetched successfully!")
-                    st.write("### Scraped Tariff Data")
-                    # Display scraped data in a table format
-                    for row in data:
-                        st.write(row)
-                else:
-                    st.error(data)
-        else:
-            st.sidebar.error("Please provide a valid URL.")
+    # Fetch tariff rates
+    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
+                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
+    tariff_categories = st.session_state.get("tariff_rates", {})
+    if tariff_categories:
+        load_type = st.sidebar.selectbox("Select Load Type", options=tariff_categories.keys())
+        tariff_rate = tariff_categories.get(load_type, 0.0)
+    else:
+        st.sidebar.warning("Using default rate. Fetch tariff rates to use live data.")
+        load_type = "Default"
+        tariff_rate = st.sidebar.number_input("Default Tariff Rate (PKR/unit)", min_value=0.0, value=20.0)
+
+    # Electricity bill calculation
+    st.sidebar.subheader("Appliance Usage")
+    appliances = {
+        "Fan": 75, "Air Conditioner (1 Ton)": 1500, "Refrigerator": 150,
+        "LED Bulb (20W)": 20, "Iron": 1000, "Microwave Oven": 1200
+    }
+    appliance = st.sidebar.selectbox("Select Appliance", options=appliances.keys())
+    quantity = st.sidebar.number_input("Quantity", min_value=1, value=1)
+    usage_hours = st.sidebar.number_input("Usage Hours/Day", min_value=1.0, value=6.0)
+
+    if st.sidebar.button("Add Appliance"):
+        st.session_state.setdefault("appliances", []).append({
+            "appliance": appliance,
+            "load": appliances[appliance],
+            "quantity": quantity,
+            "usage_hours": usage_hours
+        })
+        st.sidebar.success(f"{quantity} x {appliance} added.")
+
+    # Display appliances and calculate bill
+    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)")
+        
+        if st.button("Calculate Bill"):
+            total_units = sum(
+                app["quantity"] * app["load"] * app["usage_hours"] / 1000 for app in st.session_state["appliances"]
+            )
+            bill = total_units * 30 * tariff_rate
+            st.success(f"Estimated Monthly Bill for {load_type}: PKR {bill:.2f}")
+    else:
+        st.write("No appliances added yet.")
 
 if __name__ == "__main__":
     main()
+